diff --git a/src/main/java/com/biutag/supervisiondata/task/TaskService.java b/src/main/java/com/biutag/supervisiondata/task/TaskService.java index 189c20d..c97695d 100644 --- a/src/main/java/com/biutag/supervisiondata/task/TaskService.java +++ b/src/main/java/com/biutag/supervisiondata/task/TaskService.java @@ -1,9 +1,21 @@ package com.biutag.supervisiondata.task; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.biutag.supervisiondata.pojo.domain.TaskParamDomain; +import com.biutag.supervisiondata.pojo.entity.mine.RiskModelTaskClue; +import com.biutag.supervisiondata.pojo.entity.mine.RiskPersonal; +import com.biutag.supervisiondata.repository.RiskModelTaskClueRepository; +import com.biutag.supervisiondata.repository.RiskPersonalRepository; +import com.biutag.supervisiondata.service.PointService; +import jakarta.annotation.Resource; import lombok.extern.slf4j.Slf4j; import org.springframework.scheduling.annotation.Scheduled; import org.springframework.stereotype.Component; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + /** * @author kami on 2024-07-25 10:52:39 @@ -14,9 +26,30 @@ import org.springframework.stereotype.Component; @Slf4j public class TaskService { + @Resource + private PointService pointService; + + @Resource + private RiskPersonalRepository riskPersonalRepository; + + @Resource + private RiskModelTaskClueRepository riskModelTaskClueRepository; - @Scheduled(cron = "0 0/10 * * * ?") + /** + * 1点半执行 + */ + @Scheduled(cron = "0 30 01 * * ?") public void ageModel() { + pointService.syncPointCall(initDomain()); + } + + TaskParamDomain initDomain() { + List persons = riskPersonalRepository.list(); + List clues = riskModelTaskClueRepository.list(new LambdaQueryWrapper() + .eq(RiskModelTaskClue::getDel, 0)); + Map> map = clues.stream().collect(Collectors.groupingBy(RiskModelTaskClue::getModelId)); + TaskParamDomain domain = new TaskParamDomain(persons, map); + return domain; } }