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