From bb7971ee1b8dbfaa0e1ceb0d60daee974da77671 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Fri, 20 Oct 2023 17:57:31 +0800
Subject: [PATCH] 新增从中台读取数据（按水库和按行政区）
---
skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/SkxsQkTjServiceImpl.java | 133 ++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 128 insertions(+), 5 deletions(-)
diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/SkxsQkTjServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/SkxsQkTjServiceImpl.java
index 06d221a..8fd9b53 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/SkxsQkTjServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/SkxsQkTjServiceImpl.java
@@ -1,24 +1,34 @@
package cn.gistack.sm.sjztods.service.impl;
+import cn.gistack.common.utils.SpringContextUtil;
+import cn.gistack.sm.sjztods.constant.ZtApiUrlConstant;
+import cn.gistack.sm.sjztods.constant.ZtConfigConstant;
import cn.gistack.sm.sjztods.entity.SkxsQkTj;
import cn.gistack.sm.sjztods.excel.SkxsQkTjExcel;
import cn.gistack.sm.sjztods.mapper.SkxsQkTjMapper;
import cn.gistack.sm.sjztods.service.ISkxsQkTjService;
import cn.gistack.sm.sjztods.vo.SkxsQkTjVO;
-import cn.gistack.system.user.entity.User;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.alibaba.fastjson.parser.Feature;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.BeanUtil;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Service;
-
+import org.springframework.util.MultiValueMap;
+import org.springframework.web.client.RestTemplate;
import java.text.ParseException;
import java.text.SimpleDateFormat;
-import java.util.Date;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
/**
* @ClassName SkxsQkTjServiceImpl
@@ -28,6 +38,9 @@
@Service
@DS("ztznwh")
public class SkxsQkTjServiceImpl extends ServiceImpl<SkxsQkTjMapper, SkxsQkTj> implements ISkxsQkTjService {
+
+ @Autowired
+ private RestTemplate restTemplate;
/**
* 上报
@@ -134,5 +147,115 @@
public List<SkxsQkTjExcel> exportSkxsQkTjList(SkxsQkTjVO skxsQkTj) {
return baseMapper.exportSkxsQkTjList(skxsQkTj);
}
+
+ /**
+ * 中台接口数据调用，数据入库
+ * @param skxsQkTj
+ */
+ @Override
+ public void saveOrUpdateData(SkxsQkTjVO skxsQkTj) {
+ // 参数拼接
+ StringBuilder builder = new StringBuilder("?");
+ builder.append("pageNo=1").append("&pageSize=1000");
+ // 拼接日期，必传
+ builder.append("&dt=").append(skxsQkTj.getQueryTime());
+ String params = builder.toString();
+ // 查询
+ JSONObject jsonObject = getZtData(params, ZtApiUrlConstant.cap_water_detail);
+ // 保留 null值
+ JSONObject jsonData = jsonObject.getJSONObject("data");
+ JSONArray jsonArray = jsonData.getJSONArray("data");
+ List<SkxsQkTj> list = new ArrayList<>();
+ // 遍历,写入数据
+ for (int i = 0; i < jsonArray.size(); i++) {
+ String res_cd = jsonArray.getJSONObject(i).getString("res_cd");
+ // 查询库里是否已存在数据
+ SkxsQkTj skxs = baseMapper.selectSkxsQkTjBy(res_cd,jsonArray.getJSONObject(i).getString("res_nm"),skxsQkTj.getQueryTime());
+ if (null!=skxs){
+ // 跳出本次循环
+ continue;
+ }
+ SkxsQkTj qkTj = new SkxsQkTj();
+ // 设置各项值
+ setSkxsQkTjInfo(qkTj,jsonArray,i,skxsQkTj.getQueryTime());
+ // 加入集合
+ list.add(qkTj);
+ }
+ // 批量插入
+ saveBatch(list);
+ }
+
+ /**
+ * 设置水库库容统计值
+ * @param skxs
+ * @param jsonArray
+ * @param i 第几个
+ * @param time 统计日期
+ */
+ private void setSkxsQkTjInfo(SkxsQkTj skxs,JSONArray jsonArray,int i,String time) {
+ try {
+ skxs.setStatisticTime(new SimpleDateFormat("yyyy-MM-dd").parse(time));
+ } catch (ParseException e) {
+ e.printStackTrace();
+ }
+ skxs.setCreateUser(AuthUtil.getUserId());
+ skxs.setUpdateUser(AuthUtil.getUserId());
+ skxs.setCreateTime(new Date());
+ skxs.setUpdateTime(new Date());
+ skxs.setSysResource("zt");
+
+ skxs.setDeadCap(jsonArray.getJSONObject(i).getDouble("dead_cap"));
+ skxs.setResCode(jsonArray.getJSONObject(i).getString("res_cd"));
+ skxs.setResName(jsonArray.getJSONObject(i).getString("res_nm"));
+ skxs.setTotalCap(jsonArray.getJSONObject(i).getDouble("tot_cap"));
+ skxs.setRz(jsonArray.getJSONObject(i).getDouble("rz"));
+ skxs.setWs(jsonArray.getJSONObject(i).getDouble("now_cap"));
+ skxs.setBenResCap(jsonArray.getJSONObject(i).getDouble("ben_res_cap"));
+ skxs.setLastYearWs(jsonArray.getJSONObject(i).getDouble("last_year_cap"));
+ skxs.setHisWs(jsonArray.getJSONObject(i).getDouble("avg_cap"));
+ skxs.setUpMonthWs(jsonArray.getJSONObject(i).getDouble("last_month_cap"));
+ skxs.setTotalWsDif(jsonArray.getJSONObject(i).getString("diff_his"));
+ }
+
+ /**
+ * 调用中台接口查询数据
+ *
+ * @param params
+ * @param url
+ * @return
+ */
+ public JSONObject getZtData(String params, String url) {
+ // 获取环境
+ String activeProfile = SpringContextUtil.getActiveProfile();
+ if (activeProfile.equals("dev")){
+ url = ZtApiUrlConstant.url_prefix_dev + url;
+ }
+ if (activeProfile.equals("prod")){
+ url = ZtApiUrlConstant.url_prefix_prod + url;
+ }
+ if (activeProfile.equals("test")){
+ url = ZtApiUrlConstant.url_prefix_test + url;
+ }
+ //设置请求头
+ HttpHeaders headers = new HttpHeaders();
+ headers.add(ZtConfigConstant.header_key, ZtConfigConstant.header_value);
+ //封装请求头
+ HttpEntity<MultiValueMap<String, Object>> formEntity = new HttpEntity<MultiValueMap<String, Object>>(headers);
+ try {
+ //有请求头，有参数请求
+ ResponseEntity<String> responseEntity =
+ restTemplate.exchange(url + params,
+ HttpMethod.GET,
+ formEntity,
+ String.class);
+ // Feature.IgnoreNotMatch 保留null值的属性
+ JSONObject jsonObject = JSON.parseObject(responseEntity.getBody(),Feature.IgnoreNotMatch);
+ // 返回
+ return jsonObject;
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
}
--
Gitblit v1.9.3