fix: call wg.Close() in defer func (#9)

or-shachar · Nov 25, 2023 · cc47faa · cc47faa
1 parent f8a97c7
commit cc47faa
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/cacheproc/cacheproc.go b/cacheproc/cacheproc.go
@@ -69,6 +69,7 @@ func (p *Process) Run(ctx context.Context) error {
 	}
 	defer func() {
 		_ = p.close()
+		_ = wg.Wait()
 	}()
 	for {
 		var req wire.Request
@@ -102,11 +103,6 @@ func (p *Process) Run(ctx context.Context) error {
 			_ = bw.Flush()
 			return nil
 		})
-		select {
-		case <-ctx.Done():
-			return wg.Wait()
-		default:
-		}
 	}
 }