chenbiao 2 anni fa
parent
commit
53ce574b1e

+ 0 - 1
midjourney/src/main/java/com/yhlxj/dao/model/entity/MidjourneyUserConversation.java

@@ -17,7 +17,6 @@ import java.io.Serializable;
  * @date 2024/4/23 11:10
  */
 @Data
-@Accessors(chain = true)
 @SearchBean(tables = "midjourney_user_conversation")
 @JsonInclude(JsonInclude.Include.NON_NULL)
 public class MidjourneyUserConversation extends BaseEntity implements Serializable {

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

@@ -136,15 +136,21 @@ public class MidjourneyServiceImpl implements MidjourneyService {
         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);
+            conversation = new MidjourneyUserConversation();
+            conversation.setUserId(userId);
+            conversation.setTaskId(taskId);
+            conversation.setAction(action);
+            conversation.setPrompt(prompt);
+            conversation.setMode(mode);
+            conversation.setStartTime(System.currentTimeMillis());
+            conversation.setProgress("0%");
+            conversation.setStatus(result.getCode() == 21 ? "NOT_START" : "IN_PROGRESS");
+            conversation.setTaskId(taskId);
         }
         if (MapUtil.isNotEmpty(properties)) {
             conversation.setProperties(Jsons.toJson(properties));
-            conversation.setChannelId(properties.get("discordChannelId") == null ? null : Long.parseLong(properties.get("discordChannelId").toString()))
-                    .setInstanceId(properties.get("discordInstanceId") == null ? null : Long.parseLong(properties.get("discordInstanceId").toString()));
+            conversation.setChannelId(properties.get("discordChannelId") == null ? null : Long.parseLong(properties.get("discordChannelId").toString()));
+            conversation.setInstanceId(properties.get("discordInstanceId") == null ? null : Long.parseLong(properties.get("discordInstanceId").toString()));
         }
         if (conversation.getInstanceId() == null) {
             conversation.setInstanceId(result.getInstanceId());