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); + } + }