Merge pull request #107 from timescale/vperez/fix-race-condition-on-w…

…orker-failure Fix race condition on worker failure
timescale · Jan 13, 2025 · 27507a2 · 27507a2
2 parents 5885d47 + d7eb615
commit 27507a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/csvcopy/csvcopy.go b/pkg/csvcopy/csvcopy.go
@@ -161,7 +161,9 @@ func (c *Copier) Copy(ctx context.Context, reader io.Reader) (Result, error) {
 	}
 
 	start := time.Now()
+	workerWg.Add(1)
 	go func() {
+		defer workerWg.Done()
 		if err := batch.Scan(ctx, reader, batchChan, opts); err != nil {
 			errCh <- fmt.Errorf("failed reading input: %w", err)
 			cancel()