arima预测和hst预测更新最新数据；对预报水位去重

guoshilong

2023-08-11 3c096d3cfd56a23e6794f30ccd150bfe19aa4ae8

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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package cn.gistack.nky.service.impl;
 
import cn.gistack.nky.entity.ArimaPredict;
import cn.gistack.nky.mapper.ArimaPredictMapper;
import cn.gistack.nky.requestpojo.ArimaPredictReqPo;
import cn.gistack.nky.requestpojo.QueryReqPo;
import cn.gistack.nky.resultpojo.DataResChildrenPo;
import cn.gistack.nky.service.IArimaPredictService;
import cn.gistack.nky.service.INkyService;
import cn.gistack.nky.service.IZtApiService;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.AllArgsConstructor;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.tool.utils.ObjectUtil;
import org.springframework.stereotype.Service;
 
import java.util.ArrayList;
import java.util.List;
 
@Service
@AllArgsConstructor
public class ArimaPredictServiceImpl extends ServiceImpl<ArimaPredictMapper, ArimaPredict> implements IArimaPredictService {
    private final INkyService nkyService;
    private final IZtApiService ztApiService;
 
    @Override
    public void getData(String type) {
        List<DataResChildrenPo> cdList = getCdList(type);
        cdList.forEach(dataResChildrenPo -> {
                //循环请求预测结果
                QueryReqPo queryReqPo = new QueryReqPo(dataResChildrenPo,type);
                //请求预测结果
                ArimaPredict arimaPredict = nkyService.arimaQuery(queryReqPo);
                if (arimaPredict != null){
                    //存储预测结果至数据库
                    save(arimaPredict);
                }
        });
    }
 
    @Override
    public void predictData(String type) {
        List<DataResChildrenPo> cdList = getCdList(type);
 
        if (type.equals("1")){
            //渗压
                cdList.forEach(dataResChildrenPo -> {
                    ArimaPredictReqPo arimaPredictReqPo = new ArimaPredictReqPo(dataResChildrenPo, type);
                    //循环预测请求
                    nkyService.arimaPredict(arimaPredictReqPo);
                });
        }else if (type.equals("2")){
            //渗流
        }else if (type.equals("3") || type.equals("4") || type.equals("5")){
            cdList.forEach(dataResChildrenPo -> {
                //循环预测请求
                ArimaPredictReqPo arimaPredictReqPoX = new ArimaPredictReqPo(dataResChildrenPo, "3");
                nkyService.arimaPredict(arimaPredictReqPoX);
 
                ArimaPredictReqPo arimaPredictReqPoY = new ArimaPredictReqPo(dataResChildrenPo, "4");
                nkyService.arimaPredict(arimaPredictReqPoY);
 
                ArimaPredictReqPo arimaPredictReqPoZ = new ArimaPredictReqPo(dataResChildrenPo, "5");
                nkyService.arimaPredict(arimaPredictReqPoZ);
            });
        }
 
 
    }
 
    @Override
    public List<ArimaPredict> getFutureData(String resCd,String type) {
        List<ArimaPredict> futureData = baseMapper.getFutureData(resCd, type);
        futureData.forEach(arimaPredict -> {
            arimaPredict.setPredicts(JSON.parse(arimaPredict.getPredicts().toString()));
        });
        return futureData;
    }
 
    @Override
    public boolean save(ArimaPredict entity) {
 
        if (ObjectUtil.isEmpty(entity.getId())){
            return false;
        }
 
        List<ArimaPredict> list = baseMapper.selectTodayData(entity);
 
        if (list.size()>0){
            //删除该数据，添加最新数据
            baseMapper.deleteById(list.get(0));
            return super.save(entity);
        }else {
            return super.save(entity);
        }
    }
 
    public  List<DataResChildrenPo> getCdList(String type){
        List<DataResChildrenPo> cdList = new ArrayList<>();
        if (type.equals("1")){
            //获取渗压测点
            cdList = ztApiService.getSy();
        }else if (type.equals("2")){
            //获取渗流测点
            return null;
        }else if (type.equals("3") || type.equals("4") || type.equals("5")){
            //获取位移测点
            cdList = ztApiService.getWy();
        }else {
            return null;
        }
        return cdList;
    }
}