|
@@ -142,15 +142,14 @@ public class TaskUtil {
|
|
|
taskMapper.updateFailStateWithStopTime(taskId, state, TimeUtil.getNowForMysql(), DictConstants.TASK_ERROR_REASON_3);
|
|
|
} else if (DictConstants.TASK_ANALYSIS.equals(state)) {
|
|
|
taskMapper.updateSuccessStateWithStopTime(taskId, state, TimeUtil.getNowForMysql());
|
|
|
-
|
|
|
+
|
|
|
String isChoiceGpu = projectUtil.getProjectByProjectId(projectId).getIsChoiceGpu();
|
|
|
- if (DictConstants.VIDEO_GPU.equals(isChoiceGpu)) {
|
|
|
- log.info("项目 {} 使用 GPU 生成视频。", projectId);
|
|
|
- } else if (DictConstants.VIDEO_CPU.equals(isChoiceGpu)) {
|
|
|
+ if (DictConstants.VIDEO_CPU.equals(isChoiceGpu)) {
|
|
|
log.info("项目 {} 使用 CPU 生成视频。", projectId);
|
|
|
String generateVideoKey = "task:" + taskId + ":generateVideo";
|
|
|
customRedisClient.set(generateVideoKey, "0");
|
|
|
videoFeignClient.generateVideo(generateVideoKey, nodeName, projectId, projectType, maxSimulationTime, taskId);
|
|
|
+ log.info("任务 {} 的视频生成开始。", taskId);
|
|
|
while (true) {
|
|
|
TimeUnit.SECONDS.sleep(1);
|
|
|
final String generateVideoValue = customRedisClient.get(generateVideoKey);
|
|
@@ -158,8 +157,7 @@ public class TaskUtil {
|
|
|
break;
|
|
|
}
|
|
|
}
|
|
|
- } else {
|
|
|
- throw new RuntimeException("未设置视频生成。");
|
|
|
+ log.info("等待任务 {} 的视频生成结束。", taskId);
|
|
|
}
|
|
|
}
|
|
|
|