|
|
@ -554,11 +554,11 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
sendForm.setUserId(task.getMiniUserId()); |
|
|
sendForm.setUserId(task.getMiniUserId()); |
|
|
sendForm.setTemplateId(subscribeTemplate); |
|
|
sendForm.setTemplateId(subscribeTemplate); |
|
|
sendForm.setPage("pages/index/index"); |
|
|
sendForm.setPage("pages/creation/list"); |
|
|
sendForm.setTemplateParams(Map.of( |
|
|
sendForm.setTemplateParams(Map.of( |
|
|
"thing1", "您的AI视频作品已完成", |
|
|
"thing5", "您的AI视频作品已完成", |
|
|
"time2", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm")), |
|
|
"time3", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy年MM月dd日 HH:mm")), |
|
|
"phrase3", "点击查看状态" |
|
|
"phrase4", "待查看" |
|
|
)); |
|
|
)); |
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
} catch (Exception e) { |
|
|
} catch (Exception e) { |
|
|
@ -657,23 +657,7 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
aiTaskMediaMapper.update(null, updateMediaWrapper); |
|
|
aiTaskMediaMapper.update(null, updateMediaWrapper); |
|
|
|
|
|
|
|
|
// 同步发送订阅消息通知
|
|
|
// 同步发送订阅消息通知
|
|
|
if (StrUtil.isNotBlank(subscribeTemplate)) { |
|
|
sendAiGenerateSuccessNotify(task.getMiniUserId(), subscribeTemplate, task.getId()); |
|
|
try { |
|
|
|
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
|
|
|
sendForm.setUserId(task.getMiniUserId()); |
|
|
|
|
|
sendForm.setTemplateId(subscribeTemplate); |
|
|
|
|
|
sendForm.setPage("pages/index/index"); |
|
|
|
|
|
// 根据实际模板字段调整参数
|
|
|
|
|
|
sendForm.setTemplateParams(Map.of( |
|
|
|
|
|
"thing1", "您的AI绘画作品已完成", |
|
|
|
|
|
"time2", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm")), |
|
|
|
|
|
"phrase3", "点击查看状态" |
|
|
|
|
|
)); |
|
|
|
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
|
|
|
} catch (Exception e) { |
|
|
|
|
|
log.error("单图任务{}发送订阅消息失败", task.getId(), e); |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
log.info("四宫格任务{}回调处理完成,状态:{}", taskUuid, status); |
|
|
log.info("四宫格任务{}回调处理完成,状态:{}", taskUuid, status); |
|
|
@ -765,12 +749,12 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
sendForm.setUserId(task.getMiniUserId()); |
|
|
sendForm.setUserId(task.getMiniUserId()); |
|
|
sendForm.setTemplateId(subscribeTemplate); |
|
|
sendForm.setTemplateId(subscribeTemplate); |
|
|
sendForm.setPage("pages/index/index"); |
|
|
sendForm.setPage("pages/creation/list"); |
|
|
// 根据实际模板字段调整参数
|
|
|
// 根据实际模板字段调整参数
|
|
|
sendForm.setTemplateParams(Map.of( |
|
|
sendForm.setTemplateParams(Map.of( |
|
|
"thing1", "您的AI绘画作品已完成", |
|
|
"thing5", "您的AI绘画作品已完成", |
|
|
"time2", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm")), |
|
|
"time3", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy年MM月dd日 HH:mm")), |
|
|
"phrase3", "点击查看状态" |
|
|
"phrase4", "待查看" |
|
|
)); |
|
|
)); |
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
} catch (Exception e) { |
|
|
} catch (Exception e) { |
|
|
@ -917,7 +901,7 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
if (query.getType() != null) { |
|
|
if (query.getType() != null) { |
|
|
taskQueryWrapper.eq(MiniAiGenerationTask::getType, query.getType()); |
|
|
taskQueryWrapper.eq(MiniAiGenerationTask::getType, query.getType()); |
|
|
} |
|
|
} |
|
|
taskQueryWrapper.orderByDesc(MiniAiGenerationTask::getCreateTime); |
|
|
taskQueryWrapper.orderByDesc(MiniAiGenerationTask::getCreateTimestamp); |
|
|
IPage<MiniAiGenerationTask> taskPage = aiGenerationTaskMapper.selectPage(new Page<>(query.getPageNum(), query.getPageSize()), taskQueryWrapper); |
|
|
IPage<MiniAiGenerationTask> taskPage = aiGenerationTaskMapper.selectPage(new Page<>(query.getPageNum(), query.getPageSize()), taskQueryWrapper); |
|
|
|
|
|
|
|
|
if (taskPage.getRecords().isEmpty()) { |
|
|
if (taskPage.getRecords().isEmpty()) { |
|
|
@ -932,7 +916,7 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
// 3. 批量查询这些任务对应的AI生成媒体内容
|
|
|
// 3. 批量查询这些任务对应的AI生成媒体内容
|
|
|
LambdaQueryWrapper<MiniAiTaskMedia> mediaQueryWrapper = new LambdaQueryWrapper<>(); |
|
|
LambdaQueryWrapper<MiniAiTaskMedia> mediaQueryWrapper = new LambdaQueryWrapper<>(); |
|
|
mediaQueryWrapper.in(MiniAiTaskMedia::getTaskId, taskIds); |
|
|
mediaQueryWrapper.in(MiniAiTaskMedia::getTaskId, taskIds); |
|
|
mediaQueryWrapper.eq(MiniAiTaskMedia::getFileSource, "ai_generate"); |
|
|
mediaQueryWrapper.eq(MiniAiTaskMedia::getFileSource, "ai_generated"); |
|
|
mediaQueryWrapper.eq(MiniAiTaskMedia::getDeleted, false); |
|
|
mediaQueryWrapper.eq(MiniAiTaskMedia::getDeleted, false); |
|
|
|
|
|
|
|
|
List<MiniAiTaskMedia> mediaList = aiTaskMediaMapper.selectList(mediaQueryWrapper); |
|
|
List<MiniAiTaskMedia> mediaList = aiTaskMediaMapper.selectList(mediaQueryWrapper); |
|
|
@ -960,7 +944,7 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
)); |
|
|
)); |
|
|
|
|
|
|
|
|
// 5. 组装VO
|
|
|
// 5. 组装VO
|
|
|
return taskPage.convert(task -> { |
|
|
IPage<AiGenerationTaskVO> convert = taskPage.convert(task -> { |
|
|
AiGenerationTaskVO vo = new AiGenerationTaskVO(); |
|
|
AiGenerationTaskVO vo = new AiGenerationTaskVO(); |
|
|
// vo.setId(task.getId());
|
|
|
// vo.setId(task.getId());
|
|
|
vo.setUuid(task.getUuid()); |
|
|
vo.setUuid(task.getUuid()); |
|
|
@ -971,6 +955,34 @@ public class AiGenerationServiceImpl implements AiGenerationService { |
|
|
vo.setGenerateContent(mediaGroupMap.getOrDefault(task.getId(), Collections.emptyList())); |
|
|
vo.setGenerateContent(mediaGroupMap.getOrDefault(task.getId(), Collections.emptyList())); |
|
|
return vo; |
|
|
return vo; |
|
|
}); |
|
|
}); |
|
|
|
|
|
log.info("用户{}查询AI生成任务历史成功,结果:{}", userId, convert.getRecords().toString()); |
|
|
|
|
|
return convert; |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
|
* 发送AI绘画完成订阅消息通知 |
|
|
|
|
|
* @param userId 接收用户ID |
|
|
|
|
|
* @param templateId 订阅消息模板ID |
|
|
|
|
|
* @param taskId 任务ID(仅用于日志) |
|
|
|
|
|
*/ |
|
|
|
|
|
private void sendAiGenerateSuccessNotify(Long userId, String templateId, Long taskId) { |
|
|
|
|
|
if (StrUtil.isBlank(templateId)) { |
|
|
|
|
|
return; |
|
|
|
|
|
} |
|
|
|
|
|
try { |
|
|
|
|
|
WxSubscribeSendForm sendForm = new WxSubscribeSendForm(); |
|
|
|
|
|
sendForm.setUserId(userId); |
|
|
|
|
|
sendForm.setTemplateId(templateId); |
|
|
|
|
|
sendForm.setPage("pages/creation/list"); |
|
|
|
|
|
sendForm.setTemplateParams(Map.of( |
|
|
|
|
|
"thing5", "您的AI绘画作品已完成", |
|
|
|
|
|
"time3", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy年MM月dd日 HH:mm")), |
|
|
|
|
|
"phrase4", "待查看" |
|
|
|
|
|
)); |
|
|
|
|
|
wxSubscribeService.sendMessage(sendForm); |
|
|
|
|
|
} catch (Exception e) { |
|
|
|
|
|
log.error("四宫格图片任务{}发送订阅消息失败", taskId, e); |
|
|
|
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|