feat(kitmw): fix race condition

DoNewsCode · Mar 10, 2021 · aad54e7 · aad54e7
1 parent 0392be1
commit aad54e7
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/kitmw/async_test.go b/kitmw/async_test.go
@@ -10,16 +10,17 @@ import (
 	"github.com/opentracing/opentracing-go"
 	"github.com/opentracing/opentracing-go/mocktracer"
 	"github.com/stretchr/testify/assert"
+	"go.uber.org/atomic"
 )
 
 func TestMakeAsyncMiddleware(t *testing.T) {
-	var c int
+	var c atomic.Int32
 	m := MakeAsyncMiddleware(log.NewNopLogger(), 5)
 	f := m(func(ctx context.Context, request interface{}) (response interface{}, err error) {
-		c++
-		assert.Less(t, c, 5)
+		c.Inc()
+		assert.Less(t, c.Load(), 5)
 		time.Sleep(time.Duration(rand.Float64()) * time.Second)
-		c--
+		c.Dec()
 		return nil, nil
 	})