Bladeren bron

fix modal记录

zwhui 2 jaren geleden
bovenliggende
commit
539043cfc5

+ 9 - 4
midjourney/src/main/java/com/yhlxj/service/midjourney/impl/MidjourneyServiceImpl.java

@@ -131,10 +131,15 @@ public class MidjourneyServiceImpl implements MidjourneyService {
     public MidjourneyUserConversation saveConversation(Long userId,Integer mode,SubmitResult result, String action,String prompt, String botType) throws Exception {
         Long taskId = Long.parseLong(result.getResult());
         Map<String, Object> properties = result.getProperties();
-        MidjourneyUserConversation conversation = new MidjourneyUserConversation()
-                .setUserId(userId).setTaskId(taskId).setAction(action).setPrompt(prompt)
-                .setMode(mode).setStartTime(System.currentTimeMillis()).setProgress("0%")
-                .setStatus(result.getCode() == 21 ? "NOT_START" : "IN_PROGRESS").setTaskId(taskId);
+        MidjourneyUserConversation conversation;
+        if (action.equals("MODAL")){
+            conversation = conversationMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUserConversation.class).eq(MidjourneyUserConversation::getTaskId, taskId).last("limit 1"));
+        }else {
+            conversation = new MidjourneyUserConversation()
+                    .setUserId(userId).setTaskId(taskId).setAction(action).setPrompt(prompt)
+                    .setMode(mode).setStartTime(System.currentTimeMillis()).setProgress("0%")
+                    .setStatus(result.getCode() == 21 ? "NOT_START" : "IN_PROGRESS").setTaskId(taskId);
+        }
         if (MapUtil.isNotEmpty(properties)) {
             conversation.setProperties(Jsons.toJson(properties));
             conversation.setChannelId(properties.get("discordChannelId") == null ? null : Long.parseLong(properties.get("discordChannelId").toString()))