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

package com.zhgd.xmgl.task;

import cn.hutool.core.collection.CollUtil;
import com.alibaba.fastjson.JSONObject;
import com.zhgd.xmgl.modules.doublecarbon.entity.PhotovoltaicPowerCurrentData;
import com.zhgd.xmgl.modules.doublecarbon.entity.PhotovoltaicPowerDev;
import com.zhgd.xmgl.modules.doublecarbon.entity.dto.Oauth2TokenDTO;
import com.zhgd.xmgl.modules.doublecarbon.service.PhotovoltaicPowerCurrentDataService;
import com.zhgd.xmgl.modules.doublecarbon.service.PhotovoltaicPowerDevService;
import com.zhgd.xmgl.util.Base64;
import lombok.extern.slf4j.Slf4j;
import net.javacrumbs.shedlock.core.SchedulerLock;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import javax.annotation.Resource;
import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * @author 邱平毅
 * @ClassName PhotovoltaicPowerTask
 * @date 2023/1/3
 * @Version 1.0
 */
@Slf4j
@Component
public class PhotovoltaicPowerTask {

    private final String token;

    PhotovoltaicPowerTask(@Value("${koyoe.clientId}") String clientId, @Value("${koyoe.secret}") String secret) {
        this.token = Base64.encodeBase64((clientId + ":" + secret).getBytes(StandardCharsets.UTF_8));
    }

    @Value("${koyoe.get-token}")
    private String getToken;

    @Value("${koyoe.nowData}")
    private String nowDataUrl;

    private Oauth2TokenDTO oauth2TokenDTO;

    @Resource
    private PhotovoltaicPowerDevService photovoltaicPowerDevService;

    @Resource
    private PhotovoltaicPowerCurrentDataService photovoltaicPowerCurrentDataService;

    @Resource
    private RestTemplate restTemplate;


    /**
     * 接收第三方厂家数据，获取最新光伏发电设备发电量
     */
    @SchedulerLock(name = "photovoltaicPowerNowData", lockAtMostFor = 1000 * 60 * 5, lockAtLeastFor = 1000 * 60 * 3)
    @Scheduled(cron = "0 0/5 * * * ?")
    public void nowData() {
        // 光伏发电设备列表
        List<PhotovoltaicPowerDev> photovoltaicPowerDevList = photovoltaicPowerDevService.list();
        if (photovoltaicPowerDevList == null) {
            return;
        }
        // token刷新
        if (oauth2TokenDTO == null || oauth2TokenDTO.getExpiresDate().before(new Date())) {
            setOauth2Token();
        }
        // 设备id对应总发电量
        Map<Long, Float> devIdSumMap = photovoltaicPowerCurrentDataService.devSumGeneratingCapacity();

        List<PhotovoltaicPowerCurrentData> dataList = new LinkedList<>();
        photovoltaicPowerDevList.forEach(dev -> {
            // 循环查询光伏发电设备的最后一条数据
            HttpHeaders requestHeaders = new HttpHeaders();
            requestHeaders.setBearerAuth(oauth2TokenDTO.getAccessToken());
            HttpEntity requestEntity = new HttpEntity<>(requestHeaders);
            ResponseEntity<JSONObject> responseEntity;
            try {
                responseEntity =
                        restTemplate.exchange(String.format(nowDataUrl, dev.getDevSn()), HttpMethod.GET, requestEntity, JSONObject.class);
            } catch (Exception e) {
                e.printStackTrace();
                log.error(e.toString());
                return;
            }
            JSONObject body = responseEntity.getBody();
            // 判断请求接口是否正确
            if (responseEntity.getStatusCodeValue() != HttpStatus.OK.value() || body == null) {
                log.error("光伏发电，查找最新一条数据异常！设备sn为:{},响应体为:{}", dev.getDevSn(), responseEntity);
                return;
            }
            // 最后一条数据
            JSONObject nowData = body.getJSONObject("nowData");
            if (nowData != null) {
                Float sum = nowData.getFloat("pvTotalOutputPwer");
                // 避免恶意数据导致发电值为负值
                Long devId = dev.getId();
                float generatingCapacity = sum - devIdSumMap.getOrDefault(devId, 0F);
                if (generatingCapacity > 0) {
                    dataList.add(new PhotovoltaicPowerCurrentData(devId, generatingCapacity));
                }
            }
        });
        if (CollUtil.isNotEmpty(dataList)) {
            photovoltaicPowerCurrentDataService.saveBatch(dataList);
        }
    }

    private void setOauth2Token() {
        HttpHeaders requestHeaders = new HttpHeaders();
        requestHeaders.set("Authorization", "Basic " + token);
        HttpEntity<String> httpEntity = new HttpEntity<>(requestHeaders);
        JSONObject jsonObject = restTemplate.postForObject(getToken, httpEntity, JSONObject.class);
        if (jsonObject != null) {
            oauth2TokenDTO = new Oauth2TokenDTO(jsonObject.getString("access_token"), jsonObject.getString("scope"),
                    jsonObject.getString("token_type")).setExpiresIn(jsonObject.getInteger("expires_in"));
        } else {
            throw new RuntimeException("setOauth2Token#getToken返回体为空！");
        }
    }

}