|
|
@@ -148,7 +148,11 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
if (conversation.getInstanceId() == null) {
|
|
|
conversation.setInstanceId(result.getInstanceId());
|
|
|
}
|
|
|
- conversationMapper.insert(conversation);
|
|
|
+ if (conversation.getId() == null) {
|
|
|
+ conversationMapper.insert(conversation);
|
|
|
+ }else {
|
|
|
+ conversationMapper.updateById(conversation);
|
|
|
+ }
|
|
|
return conversation;
|
|
|
}
|
|
|
|
|
|
@@ -504,7 +508,7 @@ public class MidjourneyServiceImpl implements MidjourneyService {
|
|
|
MidjourneyUserConversation dbConversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, conversation.getTaskId())
|
|
|
.eq(MidjourneyUserConversation::getUserId, userId).orderByDesc(MidjourneyUserConversation::getId).last("limit 1"));
|
|
|
if (dbConversation != null) {
|
|
|
- if ("SUCCESS".equals(dbConversation.getStatus()) || "FAILURE".equals(dbConversation.getStatus()) || "MODAL".equals(dbConversation.getStatus())) {
|
|
|
+ if ("SUCCESS".equals(dbConversation.getStatus()) || "FAILURE".equals(dbConversation.getStatus())) {
|
|
|
return;
|
|
|
}
|
|
|
conversation.setId(dbConversation.getId());
|