From 6d0c29f5c3d5cd79649bd8e166f78ac8ab312455 Mon Sep 17 00:00:00 2001 From: xupei Date: Mon, 25 Aug 2025 18:16:57 +0800 Subject: [PATCH] =?UTF-8?q?TASK:motion=E7=94=9F=E6=88=90=E5=AE=8C=E6=88=90?= =?UTF-8?q?=E5=90=8E=E9=82=AE=E4=BB=B6=E9=80=9A=E7=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/com/ai/da/common/task/GenerateTask.java | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/ai/da/common/task/GenerateTask.java b/src/main/java/com/ai/da/common/task/GenerateTask.java index 01d95d0a..198ab2c5 100644 --- a/src/main/java/com/ai/da/common/task/GenerateTask.java +++ b/src/main/java/com/ai/da/common/task/GenerateTask.java @@ -5,9 +5,11 @@ import com.ai.da.common.utils.DateUtil; import com.ai.da.mapper.primary.PoseTransformationMapper; import com.ai.da.mapper.primary.ToProductImageResultMapper; import com.ai.da.mapper.primary.entity.*; +import com.ai.da.model.vo.PoseTransformationVO; import com.ai.da.service.APIGenerateService; import com.ai.da.service.CreditsService; import com.ai.da.service.GenerateService; +import com.ai.da.service.MessageCenterService; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import io.netty.util.internal.StringUtil; import lombok.extern.slf4j.Slf4j; @@ -31,6 +33,8 @@ public class GenerateTask { @Resource private GenerateService generateService; @Resource + private MessageCenterService messageCenterService; + @Resource private ToProductImageResultMapper toProductImageResultMapper; @Resource private PoseTransformationMapper poseTransformationMapper; @@ -99,7 +103,7 @@ public class GenerateTask { } // 万相 -> pose transformation 补偿 一小时执行一次 - @Scheduled(fixedDelay = 60 * 60 * 1000) + @Scheduled(fixedDelay = 5 * 60 * 1000) public void wxCompensationMechanism(){ List apiGenerates = apiGenerateService.getPendingTaskByStatus("wx"); if (apiGenerates != null && !apiGenerates.isEmpty()){ @@ -132,6 +136,15 @@ public class GenerateTask { } } + public void sendSysMsgToUser(Long accountId, String content){ + Notification notification = new Notification(); + notification.setType("system"); + notification.setReceiverId(accountId); + notification.setContent(content); + + messageCenterService.prePushMessage(notification); + } + }