|
|
@@ -50,6 +50,8 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
|
|
|
private static final List<String> progress = Arrays.asList("25%","35%", "45%","65%", "75%", "85%", "95%","100%");
|
|
|
|
|
|
+ private static final List<String> status = Arrays.asList("MODAL","IN_PROGRESS", "SUCCESS", "FAILURE", "CANCEL");
|
|
|
+
|
|
|
private static final GlobalThreadPoolTaskExecutor TASK_EXECUTOR = GlobalThreadPoolTaskExecutor.getInstance();
|
|
|
|
|
|
|
|
|
@@ -196,7 +198,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
|
|
|
|
|
|
public void sync(MidjourneyUserConversation conversation) throws IOException {
|
|
|
- if (progress.contains(conversation.getProgress()) || StringUtils.equals(conversation.getStatus(), "MODAL")) {
|
|
|
+ if (progress.contains(conversation.getProgress()) || status.contains(conversation.getStatus())) {
|
|
|
log.info("同步任务:{},进度:{}",conversation.getTaskId(),conversation.getProgress());
|
|
|
MidjourneyUserConversation dbConversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, conversation.getTaskId())
|
|
|
.eq(MidjourneyUserConversation::getUserId, conversation.getUserId()).last("limit 1"));
|