|
@@ -142,15 +142,14 @@ public class TaskUtil {
|
|
taskMapper.updateFailStateWithStopTime(taskId, state, TimeUtil.getNowForMysql(), DictConstants.TASK_ERROR_REASON_3);
|
|
taskMapper.updateFailStateWithStopTime(taskId, state, TimeUtil.getNowForMysql(), DictConstants.TASK_ERROR_REASON_3);
|
|
} else if (DictConstants.TASK_ANALYSIS.equals(state)) { // 该状态只会获得一次
|
|
} else if (DictConstants.TASK_ANALYSIS.equals(state)) { // 该状态只会获得一次
|
|
taskMapper.updateSuccessStateWithStopTime(taskId, state, TimeUtil.getNowForMysql());
|
|
taskMapper.updateSuccessStateWithStopTime(taskId, state, TimeUtil.getNowForMysql());
|
|
- // 查询项目是否使用 gpu 生成视频(0是1否)
|
|
|
|
|
|
+ // 查询项目是否使用 CPU 生成视频
|
|
String isChoiceGpu = projectUtil.getProjectByProjectId(projectId).getIsChoiceGpu();
|
|
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);
|
|
log.info("项目 {} 使用 CPU 生成视频。", projectId);
|
|
String generateVideoKey = "task:" + taskId + ":generateVideo";
|
|
String generateVideoKey = "task:" + taskId + ":generateVideo";
|
|
customRedisClient.set(generateVideoKey, "0");
|
|
customRedisClient.set(generateVideoKey, "0");
|
|
videoFeignClient.generateVideo(generateVideoKey, nodeName, projectId, projectType, maxSimulationTime, taskId);
|
|
videoFeignClient.generateVideo(generateVideoKey, nodeName, projectId, projectType, maxSimulationTime, taskId);
|
|
|
|
+ log.info("任务 {} 的视频生成开始。", taskId);
|
|
while (true) {
|
|
while (true) {
|
|
TimeUnit.SECONDS.sleep(1);
|
|
TimeUnit.SECONDS.sleep(1);
|
|
final String generateVideoValue = customRedisClient.get(generateVideoKey);
|
|
final String generateVideoValue = customRedisClient.get(generateVideoKey);
|
|
@@ -158,8 +157,7 @@ public class TaskUtil {
|
|
break;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
- } else {
|
|
|
|
- throw new RuntimeException("未设置视频生成。");
|
|
|
|
|
|
+ log.info("等待任务 {} 的视频生成结束。", taskId);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// -------------------------------- 判断项目是否结束 --------------------------------
|
|
// -------------------------------- 判断项目是否结束 --------------------------------
|