defer

go-to-k · Jan 29, 2025 · 1dd33f4 · 1dd33f4
1 parent 6ae208d
commit 1dd33f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/internal/wrapper/s3_tables_wrapper.go b/internal/wrapper/s3_tables_wrapper.go
@@ -113,6 +113,7 @@ func (s *S3TablesWrapper) ClearBucket(
 	wg := sync.WaitGroup{}
 	wg.Add(1)
 	go func() {
+		defer wg.Done()
 		for range progressCh {
 			count := deletedTablesCount.Add(1)
 			if !input.QuietMode {

diff --git a/internal/wrapper/s3_tables_wrapper_test.go b/internal/wrapper/s3_tables_wrapper_test.go
@@ -663,10 +663,10 @@ func TestS3TablesWrapper_deleteNamespace(t *testing.T) {
 			wg := sync.WaitGroup{}
 			wg.Add(1)
 			go func() {
+				defer wg.Done()
 				for range progressCh {
 					deletedCount.Add(1)
 				}
-				wg.Done()
 			}()
 
 			err := s3Tables.deleteNamespace(tt.args.ctx, tt.args.bucketArn, tt.args.bucketName, tt.args.namespace, progressCh)