cli: move wg.Add(1)

edgelesssys · Mar 17, 2023 · afb7b1c · afb7b1c
1 parent 219b760
commit afb7b1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cli/internal/cmd/init.go b/cli/internal/cmd/init.go
@@ -250,7 +250,6 @@ func (d *initDoer) Do(ctx context.Context) error {
 
 	grpcStateLogCtx, grpcStateLogCancel := context.WithCancel(ctx)
 	defer grpcStateLogCancel()
-	wg.Add(1)
 	// Use same context as init here since they both run on the same connection
 	d.handleGRPCStateChanges(grpcStateLogCtx, &wg, conn)
 
@@ -265,6 +264,7 @@ func (d *initDoer) Do(ctx context.Context) error {
 }
 
 func (d *initDoer) handleGRPCStateChanges(ctx context.Context, wg *sync.WaitGroup, conn *grpc.ClientConn) {
+	wg.Add(1)
 	go func() {
 		defer wg.Done()
 		state := conn.GetState()