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
53
54
55
56
57
58
59
60
61
62
package com.iailab.module.shasteel.mq.consumer;
 
import com.iailab.module.model.api.mcs.McsApi;
import com.iailab.module.model.api.mcs.dto.StScheduleSchemeDTO;
import com.iailab.module.model.api.mdk.MdkApi;
import com.iailab.module.model.api.mdk.dto.MdkScheduleReqDTO;
import com.iailab.module.model.enums.ScheduleTriggerMethodEnum;
import com.iailab.module.shasteel.mq.common.constant.CommonConstant;
import com.iailab.module.shasteel.mq.config.QueueElecPredictFinishConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;
 
import javax.annotation.Resource;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
 
/**
 * @description:
 * @author: dzd
 * @date: 2025/1/17 17:41
 **/
@Slf4j
@Component
public class ModelElecPredictFinishConsumer {
 
    @Resource
    private McsApi mcsApi;
 
    @Resource
    private MdkApi mdkApi;
 
    /**
     * 监听电力预测完成，执行电力调度
     *
     * @param message
     */
    @RabbitListener(queues = QueueElecPredictFinishConfig.QUEUE_NAME)
    public void listen(Message message) {
        String messageBody = new String(message.getBody());
        log.info("开始执行电力调度，messageBody:" + messageBody);
        // 查找需要执行的电力调度方案
        List<StScheduleSchemeDTO> scheduleSchemeList = mcsApi.listScheduleScheme(ScheduleTriggerMethodEnum.EVENT.getCode(), CommonConstant.TRIGGER_CONDITION_ELEC);
        // 电力预测完成，执行电力调度模型
        Calendar calendar = Calendar.getInstance();
        calendar.set(Calendar.MILLISECOND, 0);
        calendar.set(Calendar.SECOND, 0);
        calendar.set(Calendar.MINUTE, 0);
        calendar.set(Calendar.HOUR_OF_DAY, 0);
        Date scheduleTime = calendar.getTime();
        for (StScheduleSchemeDTO stScheduleSchemeDTO : scheduleSchemeList) {
            MdkScheduleReqDTO mdkScheduleReqDTO = new MdkScheduleReqDTO();
            mdkScheduleReqDTO.setScheduleCode(stScheduleSchemeDTO.getCode());
            mdkScheduleReqDTO.setScheduleTime(scheduleTime);
            mdkApi.doSchedule(mdkScheduleReqDTO);
        }
 
 
    }
}