1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package com.iailab.module.ansteel.job.task;

 

import com.iailab.module.model.api.mdk.MdkApi;

import com.iailab.module.model.api.mdk.dto.MdkPredictReqDTO;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.amqp.rabbit.core.RabbitTemplate;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

 

import java.util.*;

 

/**

 * 执行模块预测定时任务

 */

@Component("runPredictModuleTask")

public class RunPredictModuleTask implements ITask {

    private Logger logger = LoggerFactory.getLogger(getClass());

 

    @Autowired

    private MdkApi mdkApi;

 

    @Autowired

    private RabbitTemplate rabbitTemplate;

 

    @Override

    public void run(String params) {

        logger.info("runPredictModuleTask定时任务正在执行，参数为：{}", params);

        try {

            Calendar calendar = Calendar.getInstance();

            calendar.set(Calendar.SECOND, 0);

            calendar.set(Calendar.MILLISECOND, 0);

            Date predictTime = calendar.getTime();

            if (StringUtils.isEmpty(params)) {

                logger.info("参数为空");

                return;

            }

            List<String> moduleTypeList = Arrays.asList(params.split(","));

            moduleTypeList.forEach(type -> {

                MdkPredictReqDTO dto = new MdkPredictReqDTO();

                dto.setPredictTime(predictTime);

                dto.setModuleType(type);

                mdkApi.predictModule(dto);

                logger.info(type + "模块预测完成。");

            });

        } catch (Exception ex) {

            logger.error("runPredictModuleTask运行异常",ex);

        }

        logger.info("runPredictModuleTask运行完成");

    }

}