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/XsfsQktjServiceImpl.java | 144 ++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 144 insertions(+), 0 deletions(-)
diff --git a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/XsfsQktjServiceImpl.java b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/XsfsQktjServiceImpl.java
index 89eeab8..e610a01 100644
--- a/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/XsfsQktjServiceImpl.java
+++ b/skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sjztods/service/impl/XsfsQktjServiceImpl.java
@@ -1,20 +1,36 @@
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.entity.XsfsQktj;
import cn.gistack.sm.sjztods.excel.XsfsQktjExcel;
import cn.gistack.sm.sjztods.mapper.XsfsQktjMapper;
import cn.gistack.sm.sjztods.service.IXsfsQktjService;
import cn.gistack.sm.sjztods.vo.XsfsQktjVO;
+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.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Objects;
@@ -27,6 +43,9 @@
@Service
@DS("ztznwh")
public class XsfsQktjServiceImpl extends ServiceImpl<XsfsQktjMapper, XsfsQktj> implements IXsfsQktjService {
+
+ @Autowired
+ private RestTemplate restTemplate;
/**
* 上报
@@ -128,5 +147,130 @@
public List<XsfsQktjExcel> exportXsfsQktjList(XsfsQktjVO xsfsQktj) {
return baseMapper.exportXsfsQktjList(xsfsQktj);
}
+
+ /**
+ * 中台接口数据调用，数据入库
+ * @param xsfsQktj
+ */
+ @Override
+ public void saveOrUpdateData(XsfsQktjVO xsfsQktj) {
+ // 参数拼接
+ StringBuilder builder = new StringBuilder("?");
+ // 拼接日期，必传
+ builder.append("?dt=").append(xsfsQktj.getQueryTime());
+ if (null!=xsfsQktj.getType() && !xsfsQktj.getType().equals("")) {
+ builder.append("&").append(xsfsQktj.getType()).append("=").append(xsfsQktj.getTypeValue());
+ }
+ String params = builder.toString();
+ // 查询
+ JSONObject jsonObject = getZtData(params, ZtApiUrlConstant.cap_tot_status);
+ // 解析数据
+ JSONArray jsonArray = jsonObject.getJSONArray("data");
+ List<XsfsQktj> list = new ArrayList<>();
+ // 遍历,写入数据
+ for (int i = 0; i < jsonArray.size(); i++) {
+ String ad_nm = "";
+ if (xsfsQktj.getType().equals("city_cd")){
+ ad_nm = jsonArray.getJSONObject(i).getString("county_nm");
+ }else if (xsfsQktj.getType().equals("county_cd")){
+ ad_nm = jsonArray.getJSONObject(i).getString("town_nm");
+ }else {
+ ad_nm = jsonArray.getJSONObject(i).getString("city_nm");
+ }
+ // 查询库里是否已存在数据
+ XsfsQktj xsfsQktjBy = baseMapper.selectXsfsQktjBy(ad_nm,xsfsQktj.getQueryTime());
+ if (null!=xsfsQktjBy){
+ // 跳出本次循环
+ continue;
+ }
+ XsfsQktj qkTj = new XsfsQktj();
+ qkTj.setAdName(ad_nm);
+ // 设置各项值
+ setXsfsQktjInfo(qkTj,jsonArray,i,xsfsQktj.getQueryTime());
+ // 加入集合
+ list.add(qkTj);
+ }
+ // 批量插入
+ saveBatch(list);
+ }
+
+
+ /**
+ * 设置水库库容统计值
+ * @param skxs
+ * @param jsonArray
+ * @param i 第几个
+ * @param time 统计日期
+ */
+ private void setXsfsQktjInfo(XsfsQktj 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.setZxsTotal(jsonArray.getJSONObject(i).getDouble("tot_cap"));
+ skxs.setZxsSc(jsonArray.getJSONObject(i).getString("diff_last_day"));
+ skxs.setZxsLastYearTq(jsonArray.getJSONObject(i).getString("rate_last_year_cap"));
+ skxs.setZxsLastYearTqzj(jsonArray.getJSONObject(i).getString("diff_last_year"));
+ skxs.setZxsHisTq(jsonArray.getJSONObject(i).getString("diff_his_cap"));
+ skxs.setZxsHisTqzj(jsonArray.getJSONObject(i).getString("rate_avg_cap"));
+ // 大型
+ skxs.setDxskXsTotal(jsonArray.getJSONObject(i).getDouble("big_tot_cap"));
+ skxs.setDxskXsHis(jsonArray.getJSONObject(i).getString("big_diff_his"));
+ skxs.setDxskXsLastYear(jsonArray.getJSONObject(i).getString("big_diff_last_year"));
+ // 中型
+ skxs.setZxskXsTotal(jsonArray.getJSONObject(i).getDouble("mid_tot_cap"));
+ skxs.setZxskXsHis(jsonArray.getJSONObject(i).getString("mid_diff_his"));
+ skxs.setZxskXsLastYear(jsonArray.getJSONObject(i).getString("mid_diff_last_year"));
+ // 小型
+ skxs.setXxskXsTotal(jsonArray.getJSONObject(i).getDouble("small_tot_cap"));
+ }
+
+ /**
+ * 调用中台接口查询数据
+ *
+ * @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