wisdomisite-java/src/main/java/com/zhgd/xmgl/task/ElectricalTask.java

package com.zhgd.xmgl.task;

import cn.hutool.core.collection.CollUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zhgd.xmgl.modules.electrical.entity.ElectricalData;
import com.zhgd.xmgl.modules.electrical.entity.ElectricalDev;
import com.zhgd.xmgl.modules.electrical.mapper.ElectricalDataMapper;
import com.zhgd.xmgl.modules.electrical.service.IElectricalDataService;
import com.zhgd.xmgl.modules.electrical.service.IElectricalDevService;
import com.zhgd.xmgl.modules.project.entity.Project;
import com.zhgd.xmgl.modules.project.mapper.ProjectMapper;
import com.zhgd.xmgl.util.XiwonUtil;
import lombok.extern.slf4j.Slf4j;
import net.javacrumbs.shedlock.core.SchedulerLock;
import org.springframework.http.HttpStatus;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;

/**
 * @author 邱平毅
 * @ClassName ElectricalTask
 * @date 2022/11/15 14:02
 * @Version 1.0
 * 电箱定时任务
 */
@Slf4j
@Component
@RestController
public class ElectricalTask {
    @Resource
    private IElectricalDataService electricalDataService;
    @Resource
    private ElectricalDataMapper electricalDataMapper;
    @Resource
    private ProjectMapper projectMapper;

    @Resource
    private IElectricalDevService electricalDevService;

    @Resource
    private XiwonUtil xiwonUtil;

    /**
     * 获取最新电量使用情况 每5分钟触发任务
     */
    @SchedulerLock(name = "getElectricRealTimeData", lockAtMostFor = 1000 * 60 * 5, lockAtLeastFor = 1000 * 60 * 3)
    @Scheduled(cron = "0 0/5 * * * ?")
    @GetMapping("/xmgl/task/getElectricRealTimeData")
    public void getElectricRealTimeData() {
        log.info("获取最新电量使用情况 每5分钟触发任务");
        List<Project> projectList = projectMapper.selectList(new LambdaQueryWrapper<Project>()
                .ne(Project::getXiwonAppId, "")
                .isNotNull(Project::getXiwonAppId)
                .ne(Project::getXiwonAppSecret, "")
                .isNotNull(Project::getXiwonAppSecret)
        );
        for (Project project : projectList) {
            // 获取所有
            List<ElectricalDev> devList = electricalDevService.list(new LambdaQueryWrapper<ElectricalDev>().eq(ElectricalDev::getProjectSn, project.getProjectSn()));
            devList.forEach(dev -> CompletableFuture.supplyAsync(() -> {
                        return doGetElectricRealTimeData(dev, project);
                    }).exceptionally(throwable -> {
                        log.error("err:", throwable);
                        return null;
                    })
            );
        }
    }

    private boolean doGetElectricRealTimeData(ElectricalDev dev, Project project) {
        // 请求参数
        Map<String, Object> map = new HashMap<>(4);
        map.put("state", 0);
        map.put("pageSize", "1");
        map.put("pageNum", "1");
        // 设备sn
        String devSn = dev.getDevSn();
        ElectricalData newestData = electricalDataMapper.selectOne(new LambdaQueryWrapper<ElectricalData>().eq(ElectricalData::getDevSn, devSn).orderByDesc(ElectricalData::getUploadTime).last("limit 1"));
        // 请求携稳的接口
        JSONObject realTimeData = xiwonUtil.postForm("/electric/realTimeData", project.getXiwonAppId(), project.getXiwonAppSecret(), map, JSONObject.class, (() -> {
            if (newestData != null && newestData.getXiewenId() != null) {
                map.put("dataId", newestData.getXiewenId());
            }
            return map.put("deviceSn", devSn);
        }));

        log.info("携稳接口响应数据（配电箱）：{}", JSONArray.toJSONString(realTimeData));
        Integer code = realTimeData.getInteger("code");
        // code校验是否成功请求
        if (code == HttpStatus.OK.value()) {
            JSONArray dataArray = realTimeData.getJSONArray("data");
            if (CollUtil.isNotEmpty(dataArray)) {
                // 解析请求到的参数，保存到我们的数据库
                JSONObject data = dataArray.getJSONObject(0);
                ElectricalData electricalData = new ElectricalData(devSn, data.getString("dataTime"), data.getString("voltageA"), data.getString("voltageB"),
                        data.getString("voltageC"), data.getString("currentA"), data.getString("currentB"), data.getString("currentC"),
                        data.getString("leakage"), data.getString("temperatureA"), data.getString("temperatureB"), data.getString("temperatureC"), data.getString("temperatureD"));
                electricalData.setXiewenId(data.getLong("id"));
                electricalDataService.saveElectricalData(electricalData);
            } else {
                log.error("设备sn：{}，当前无数据！", devSn);
            }
        } else {
            log.error("设备sn：{}，请求失败！当前code：{}，msg：{}", devSn, code, realTimeData.getString("msg"));
        }
        return true;
    }
}