LingxinMeng 1 年間 前
コミット
a91da2cda1
1 ファイル変更13 行追加12 行削除
  1. 13 12
      amd64/dispatch_server/package/service/run_task.go

+ 13 - 12
amd64/dispatch_server/package/service/run_task.go

@@ -93,13 +93,20 @@ func RunWaitingCluster() {
 			}
 
 			// 取出但不移除
-			firstTaskCacheJson, err := infra.GlobalRedisClient.LIndex(global.KeyTaskQueueWaitingCluster, 0).Result()
-			if err != nil {
-				infra.GlobalLogger.Error("取出集群等待队列中的头元素报错,错误信息为:", err)
-				global.GpuNodeListMutex.Unlock()
-				continue
+			{
+				firstTaskCacheJson, err := infra.GlobalRedisClient.LIndex(global.KeyTaskQueueWaitingCluster, 0).Result()
+				if err != nil {
+					infra.GlobalLogger.Error("取出集群等待队列中的头元素报错,错误信息为:", err)
+					global.GpuNodeListMutex.Unlock()
+					continue
+				}
+				firstTaskCache, err = JsonToTaskCache(firstTaskCacheJson)
+				if err != nil {
+					infra.GlobalLogger.Error(err)
+					global.GpuNodeListMutex.Unlock()
+					continue
+				}
 			}
-
 			// --------------- 下载算法 ---------------
 			infra.GlobalLogger.Infof("开始下载算法 %v。", firstTaskCache.AlgorithmObjectKey)
 			algorithmTarName = filepath.Base(firstTaskCache.AlgorithmObjectKey)
@@ -114,12 +121,6 @@ func RunWaitingCluster() {
 			}
 			infra.GlobalLogger.Infof("下载算法 %v 成功。", firstTaskCache.AlgorithmObjectKey)
 
-			firstTaskCache, err = JsonToTaskCache(firstTaskCacheJson)
-			if err != nil {
-				infra.GlobalLogger.Error(err)
-				global.GpuNodeListMutex.Unlock()
-				continue
-			}
 			err = domain.AddRunningCluster(firstTaskCache, gpuNode.Hostname)
 			if err != nil {
 				infra.GlobalLogger.Error(err)