zwhui пре 2 година
родитељ
комит
3f3f17eaf2

+ 6 - 2
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -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());