From 1dd33f4a631ee0475fafca3c87ce39494872f31b Mon Sep 17 00:00:00 2001
From: go-to-k <24818752+go-to-k@users.noreply.github.com>
Date: Wed, 29 Jan 2025 22:24:49 +0900
Subject: [PATCH] defer

---
 internal/wrapper/s3_tables_wrapper.go      | 1 +
 internal/wrapper/s3_tables_wrapper_test.go | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/internal/wrapper/s3_tables_wrapper.go b/internal/wrapper/s3_tables_wrapper.go
index c9206a6..b75f570 100644
--- 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
index ce868e3..1ccfc08 100644
--- 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)