Merge pull request 'fix startall stopall bugs' (#305) from tzwang/pcm-coordinator:master into master

Former-commit-id: ee8207785f

internal/logic/inference/startallbydeploytaskidlogic.go

@@ -82,6 +82,7 @@ func (l *StartAllByDeployTaskIdLogic) startAll(list []*models.AiInferDeployInsta
 				errCh <- e
 				wg.Done()
 				<-buf
+				return
 			}
 			if checkStopStatus(in) {
 				success := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[strconv.FormatInt(ins.AdapterId, 10)][strconv.FormatInt(ins.ClusterId, 10)].StartInferDeployInstance(l.ctx, ins.InstanceId)
@@ -100,11 +101,15 @@ func (l *StartAllByDeployTaskIdLogic) startAll(list []*models.AiInferDeployInsta
 					errCh <- e
 					wg.Done()
 					<-buf
+					return
 				}
 			}
+			wg.Done()
+			<-buf
 		}()
-		<-buf
 	}
+	wg.Wait()
+	close(errCh)
 
 	for e := range errCh {
 		errs = append(errs, e)

internal/logic/inference/stopallbydeploytaskidlogic.go

@@ -83,6 +83,7 @@ func (l *StopAllByDeployTaskIdLogic) stopAll(list []*models.AiInferDeployInstanc
 				errCh <- e
 				wg.Done()
 				<-buf
+				return
 			}
 			if checkStatus(in) {
 				success := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[strconv.FormatInt(ins.AdapterId, 10)][strconv.FormatInt(ins.ClusterId, 10)].StopInferDeployInstance(l.ctx, ins.InstanceId)
@@ -101,11 +102,15 @@ func (l *StopAllByDeployTaskIdLogic) stopAll(list []*models.AiInferDeployInstanc
 					errCh <- e
 					wg.Done()
 					<-buf
+					return
 				}
 			}
+			wg.Done()
+			<-buf
 		}()
-		<-buf
 	}
+	wg.Wait()
+	close(errCh)
 
 	for e := range errCh {
 		errs = append(errs, e)