avoid race condition

Signed-off-by: Florian Lehner <[email protected]>
florianl · Mar 25, 2022 · 04d1b2d · 04d1b2d
1 parent 697db84
commit 04d1b2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/nfqueue.go b/nfqueue.go
@@ -151,10 +151,10 @@ func (nfqueue *Nfqueue) RegisterWithErrorFunc(ctx context.Context, fn HookFunc,
 		return err
 	}
 
+	nfqueue.wg.Add(1)
 	go func() {
-		nfqueue.wg.Add(1)
+		defer nfqueue.wg.Done()
 		nfqueue.socketCallback(ctx, fn, errfn, seq)
-		nfqueue.wg.Done()
 	}()
 
 	return nil

diff --git a/nfqueue_linux_integration_test.go b/nfqueue_linux_integration_test.go
@@ -98,5 +98,4 @@ func TestTimeout(t *testing.T) {
 
 	// Block till the context expires
 	<-ctx.Done()
-
 }
Original file line number	Diff line number	Diff line change
Expand Up		@@ -98,5 +98,4 @@ func TestTimeout(t *testing.T) {

		// Block till the context expires
		<-ctx.Done()

		}