making wg to work with each proposal

hypermodeinc · Jan 5, 2021 · d1f39f4 · d1f39f4
1 parent f733120
commit d1f39f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/worker/online_restore_ee.go b/worker/online_restore_ee.go
@@ -93,11 +93,11 @@ func ProcessRestoreRequest(ctx context.Context, req *pb.RestoreRequest, wg *sync
 
 	// TODO: prevent partial restores when proposeRestoreOrSend only sends the restore
 	// request to a subset of groups.
-	wg.Add(1)
 	errCh := make(chan error, len(currentGroups))
 	for _, gid := range currentGroups {
 		reqCopy := proto.Clone(req).(*pb.RestoreRequest)
 		reqCopy.GroupId = gid
+		wg.Add(1)
 		go func() {
 			errCh <- tryRestoreProposal(ctx, reqCopy)
 		}()
@@ -108,8 +108,8 @@ func ProcessRestoreRequest(ctx context.Context, req *pb.RestoreRequest, wg *sync
 			if err := <-errCh; err != nil {
 				glog.Errorf("Error while restoring %v", err)
 			}
+			wg.Done()
 		}
-		wg.Done()
 	}()
 
 	return nil