RyTask.java 84 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558
  1. package com.slibra.quartz.task;
  2. import cn.hutool.core.date.DateUtil;
  3. import cn.hutool.http.HttpRequest;
  4. import com.alibaba.fastjson2.JSON;
  5. import com.alibaba.fastjson2.JSONArray;
  6. import com.alibaba.fastjson2.JSONObject;
  7. import com.alibaba.fastjson2.JSONWriter;
  8. import com.google.protobuf.ByteString;
  9. import com.slibra.business.domain.*;
  10. import com.slibra.business.mapper.*;
  11. import com.slibra.business.req.ChatReq;
  12. import com.slibra.business.req.DecisionReq;
  13. import com.slibra.common.constant.MyConstants;
  14. import com.slibra.common.enums.BusinessEnum;
  15. import com.slibra.common.enums.DataSourceType;
  16. import com.slibra.common.utils.DateUtils;
  17. import com.slibra.common.utils.ip.IpUtils;
  18. import com.slibra.common.utils.uuid.IdUtils;
  19. import com.slibra.framework.datasource.DynamicDataSourceContextHolder;
  20. import inference.InferenceAPIsServiceGrpc;
  21. import inference.PredictionResponse;
  22. import inference.PredictionsRequest;
  23. import io.grpc.ManagedChannel;
  24. import io.grpc.ManagedChannelBuilder;
  25. import lombok.extern.slf4j.Slf4j;
  26. import org.springframework.beans.factory.annotation.Autowired;
  27. import org.springframework.stereotype.Component;
  28. import com.slibra.common.utils.StringUtils;
  29. import org.springframework.util.CollectionUtils;
  30. import java.io.IOException;
  31. import java.math.BigDecimal;
  32. import java.math.RoundingMode;
  33. import java.time.LocalDateTime;
  34. import java.time.format.DateTimeFormatter;
  35. import java.util.*;
  36. import static com.slibra.common.constant.MyConstants.*;
  37. /**
  38. * 定时任务调度测试
  39. *
  40. *
  41. */
  42. @Component("ryTask")
  43. @Slf4j
  44. public class RyTask
  45. {
  46. public void ryMultipleParams(String s, Boolean b, Long l, Double d, Integer i)
  47. {
  48. System.out.println(StringUtils.format("执行多参方法: 字符串类型{},布尔类型{},长整型{},浮点型{},整形{}", s, b, l, d, i));
  49. }
  50. public void ryParams(String params)
  51. {
  52. System.out.println("执行有参方法:" + params);
  53. }
  54. public void ryNoParams()
  55. {
  56. System.out.println("执行无参方法");
  57. }
  58. //----------------------------------------------下面是新增的方法----------------------------------------------
  59. @Autowired
  60. private TXinyiIndustryMapper xinyiIndustryMapper;
  61. @Autowired
  62. private TXinyiRobotMapper xinyiRobotMapper;
  63. @Autowired
  64. private TXinyiNormConfigMapper xinyiNormConfigMapper;
  65. @Autowired
  66. private TXinyiWarningRecordMapper xinyiWarningRecordMapper;
  67. @Autowired
  68. private TXinyiChatRecordMapper xinyiChatRecordMapper;
  69. @Autowired
  70. private TXinyiDailyMapper xinyiDailyMapper;
  71. // public final static StopWatch watch = new StopWatch("task");
  72. public static final String[] queryTags = {"信义污水厂JS_COD_Value","信义污水厂JS_PH_Value","信义污水厂JS_SS_Value","信义污水厂JS_ZL_Value","信义污水厂JS_ZA_Value","信义污水厂JS_AD_Value","信义污水厂JS_T_Value","信义污水厂进水泵房液位","信义污水厂出水瞬时流量","信义污水厂升级出水COD","信义污水厂升级出水PH","信义污水厂升级出水SS","信义污水厂升级出水TN","信义污水厂升级出水TP","信义污水厂升级出水氨氮","信义污水厂AIT202_Value","信义污水厂AIT203_Value","信义污水厂AIT207_Value","信义污水厂AIT206_Value","信义污水厂AIT209_Value","信义污水厂AIT210_Value","信义污水厂进水TDS","信义污水厂FT101_Value","信义污水厂SWCHHYHLB1_R_Value","信义污水厂SWCHHYHLB2_R_Value","信义污水厂SWCHHYHLB3_R_Value","信义污水厂SWCHHYHLB4_R_Value","信义污水厂SWCHHYHLB5_R_Value","信义污水厂SWCHHYHLB6_R_Value","信义污水厂SWCWNHLB1_R_Value","信义污水厂SWCWNHLB2_R_Value","信义污水厂SWCWNHLB3_R_Value","信义污水厂SWCWNHLB4_R_Value","信义污水厂SWCWNHLB5_R_Value","信义污水厂GFJ1_R_Value","信义污水厂GFJ2_R_Value","信义污水厂GFJ3_R_Value","信义污水厂GFJ4_R_Value","信义污水厂GFJ5_R_Value","信义污水厂GFJ6_R_Value","信义污水厂GFJ1_KQLL_Value","信义污水厂GFJ2_KQLL_Value","信义污水厂GFJ3_KQLL_Value","信义污水厂GFJ4_KQLL_Value","信义污水厂GFJ5_KQLL_Value","信义污水厂GFJ6_KQLL_Value","信义污水厂实际碳源加药量","信义污水厂除磷加药瞬时流量", "信义污水厂_除磷P04预测值_"};
  73. /**
  74. * 定时从工业库获取数据
  75. *
  76. * 2024年4月17日17:44:15 调整逻辑:考虑到因断电等情况导致服务断电,所以不再同步最近一小时,而是同步从上次成功的最后一条数据开始。
  77. */
  78. public void getIndustryData(){
  79. log.info("进入了定时同步工业库数据的任务");
  80. //耗时工具
  81. // watch.start("parseJob");
  82. // 给定时间段的起始时间和结束时间
  83. LocalDateTime endTime = LocalDateTime.now();
  84. // LocalDateTime startTime = endTime.plusMinutes(-60);
  85. //获取上次最后一条同步的数据的日期到 分钟维度
  86. String lastDateHour = this.xinyiIndustryMapper.getLastMinute();
  87. log.info("获取上次同步工业库的最后一条记录的时间是{}", lastDateHour);
  88. lastDateHour = lastDateHour + ":00";
  89. //开始时间
  90. LocalDateTime startTime = LocalDateTime.parse(lastDateHour.replaceAll("/", "-").replace(" ", "T"));
  91. startTime = startTime.plusMinutes(60L);//加一分钟 从上次最后一条记录的下一分钟开始
  92. /*LocalDateTime startTime = LocalDateTime.parse("2024-02-26T00:00:00");
  93. LocalDateTime endTime = LocalDateTime.parse("2024-02-27T00:00:00");*/
  94. // 每个小时的时间格式
  95. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
  96. // 循环按小时分割
  97. LocalDateTime currentHour = startTime;
  98. //最终获取的数据
  99. Map<String, TXinyiIndustry> needMap = new LinkedHashMap<>();
  100. while (currentHour.isBefore(endTime)) {
  101. String begin = currentHour.format(formatter);
  102. String end = currentHour.plusMinutes(5).format(formatter);
  103. // 输出当前小时的起始时间和结束时间
  104. System.out.println("起始时间:" + begin);
  105. System.out.println("结束时间:" + end);
  106. // 当前小时加一小时,作为下一个小时的起始时间
  107. currentHour = currentHour.plusMinutes(5);
  108. //每个小时查询一次数据
  109. String url = "http://10.0.0.27:4568/api/v1/khquerydata";
  110. HashMap<String, Object> req = new HashMap<>();
  111. req.put("tagNames", queryTags);
  112. req.put("startTime", begin);
  113. req.put("endTime", end);
  114. req.put("recordNumbers", 100000);
  115. String body = HttpRequest.post(url).header("Authorization", "c2E6c2E=").header("clientName", "hongshan").body(JSON.toJSONString(req)).execute().body();
  116. // System.out.println("body = " + body);
  117. List<HashMap<String, String>> list = new ArrayList<>();
  118. //行转列数据处理
  119. for (String queryTag : queryTags) {
  120. JSONArray array = JSON.parseObject(body).getJSONArray(queryTag);
  121. //特殊数据处理一
  122. if(Objects.isNull(array) || array.isEmpty()){
  123. System.out.println(queryTag + "查询到了空的数据,跳过本次循环");
  124. continue;
  125. }
  126. int size = array.size();
  127. //特殊数据处理二
  128. if("0".equals(array.get(1) + "")){
  129. System.out.println(queryTag + "查询到了数据,但是数据集合只有一条,且都是0");
  130. continue;
  131. }
  132. //结合至少62个数据才满足条件(有可能获取不到)
  133. /*if(size < 62){
  134. System.out.println(queryTag + "查询到了不符合条件的数据,跳过本次循环");
  135. continue;
  136. }*/
  137. //存放的数据集
  138. //利用map去重
  139. HashMap<String, String> map = new LinkedHashMap<>();
  140. for (int i = 2; i < size; i++) {
  141. // System.out.println(i + "" + array.get(i));
  142. JSONArray oneRecord = JSON.parseArray(JSON.toJSONString(array.get(i)));
  143. //处理为空或者为0的数据
  144. Object timeStampValue = oneRecord.get(2);
  145. if(Objects.isNull(timeStampValue) || "0".equals(timeStampValue + ""))
  146. continue;
  147. BigDecimal value = Objects.isNull(oneRecord.get(0)) ? null : new BigDecimal(oneRecord.get(0) + "");
  148. long timestamp = (long) timeStampValue;
  149. String format = DateUtil.format(new Date(timestamp), DateUtils.YYYYMMDDHH_TS);
  150. map.put(format, queryTag + "-" + value);
  151. }
  152. list.add(map);
  153. }
  154. Set<String> recordTimeSet = new HashSet<>();
  155. Map<String, TXinyiIndustry> recordMap = new HashMap<>();
  156. for (int i = 0; i < list.size(); i++) {
  157. HashMap<String, String> map = list.get(i);
  158. int finalJ = i;
  159. map.forEach((k, v) ->{
  160. TXinyiIndustry industry = null;
  161. if(!recordTimeSet.contains(k)){//第一次
  162. industry = new TXinyiIndustry();
  163. recordTimeSet.add(k);
  164. recordMap.put(k, industry);
  165. }else{
  166. industry = recordMap.get(k);
  167. }
  168. industry.setTestTime(k + ":00");
  169. //2024年4月15日11:19:52 额外增加2个字段
  170. industry.setTestDate(k.substring(0,10));
  171. industry.setTestHour(k.substring(0,13));
  172. //解析值
  173. String[] split = v.split("-");
  174. String type = split[0];
  175. BigDecimal value = new BigDecimal(split[1]);
  176. if ("信义污水厂JS_COD_Value".equals(type)) {
  177. industry.setJsCod(value);
  178. } else if ("信义污水厂JS_PH_Value".equals(type)) {
  179. industry.setJsPh(value);
  180. } else if ("信义污水厂JS_SS_Value".equals(type)) {
  181. industry.setJsSs(value);
  182. } else if ("信义污水厂JS_ZL_Value".equals(type)) {
  183. industry.setJsTp(value);
  184. } else if ("信义污水厂JS_ZA_Value".equals(type)) {
  185. industry.setJsTn(value);
  186. } else if ("信义污水厂JS_AD_Value".equals(type)) {
  187. industry.setJsNh3(value);
  188. } else if ("信义污水厂JS_T_Value".equals(type)) {
  189. industry.setJsSwPh(value);
  190. } else if ("信义污水厂进水泵房液位".equals(type)) {
  191. industry.setJsBfyw(value);
  192. } else if ("信义污水厂出水瞬时流量".equals(type)) {
  193. industry.setCsSlqc(value);
  194. } else if ("信义污水厂升级出水COD".equals(type)) {
  195. industry.setCsCod(value);
  196. } else if ("信义污水厂升级出水PH".equals(type)) {
  197. industry.setCsPh(value);
  198. } else if ("信义污水厂升级出水SS".equals(type)) {
  199. industry.setCsSs(value);
  200. } else if ("信义污水厂升级出水TN".equals(type)) {
  201. industry.setCsTn(value);
  202. } else if ("信义污水厂升级出水TP".equals(type)) {
  203. industry.setCsTp(value);
  204. } else if ("信义污水厂升级出水氨氮".equals(type)) {
  205. industry.setCsNh3(value);
  206. } else if ("信义污水厂AIT202_Value".equals(type)) {
  207. industry.setOneHyzdDo(value);
  208. } else if ("信义污水厂AIT203_Value".equals(type)) {
  209. industry.setOneHymdDo(value);
  210. } else if ("信义污水厂AIT207_Value".equals(type)) {
  211. industry.setTwoHyzdDo(value);
  212. } else if ("信义污水厂AIT206_Value".equals(type)) {
  213. industry.setTwoHymdDo(value);
  214. } else if ("信义污水厂AIT209_Value".equals(type)) {
  215. industry.setOneMlss(value);
  216. } else if ("信义污水厂AIT210_Value".equals(type)) {
  217. industry.setTwoMlss(value);
  218. } else if ("信义污水厂进水TDS".equals(type)) {
  219. industry.setJsTds(value);
  220. } else if ("信义污水厂FT101_Value".equals(type)) {
  221. industry.setJsSlq(value);
  222. } else if ("信义污水厂SWCHHYHLB1_R_Value".equals(type)) {
  223. industry.setNHlbOneGp(value);
  224. } else if ("信义污水厂SWCHHYHLB2_R_Value".equals(type)) {
  225. industry.setNHlbTwoGp(value);
  226. } else if ("信义污水厂SWCHHYHLB3_R_Value".equals(type)) {
  227. industry.setNHlbThreeGp(value);
  228. } else if ("信义污水厂SWCHHYHLB4_R_Value".equals(type)) {
  229. industry.setNHlbFourGp(value);
  230. } else if ("信义污水厂SWCHHYHLB5_R_Value".equals(type)) {
  231. industry.setNhlBFiveGp(value);
  232. } else if ("信义污水厂SWCHHYHLB6_R_Value".equals(type)) {
  233. industry.setNHlbSixGp(value);
  234. } else if ("信义污水厂SWCWNHLB1_R_Value".equals(type)) {
  235. industry.setWHlbOneGp(value);
  236. } else if ("信义污水厂SWCWNHLB2_R_Value".equals(type)) {
  237. industry.setWHlbTwoGp(value);
  238. } else if ("信义污水厂SWCWNHLB3_R_Value".equals(type)) {
  239. industry.setWHlbThreeGp(value);
  240. } else if ("信义污水厂SWCWNHLB4_R_Value".equals(type)) {
  241. industry.setWHlbFourGp(value);
  242. } else if ("信义污水厂SWCWNHLB5_R_Value".equals(type)) {
  243. industry.setWHlbFiveGp(value);
  244. } else if ("信义污水厂GFJ1_R_Value".equals(type)) {
  245. industry.setFjOne(value);
  246. } else if ("信义污水厂GFJ2_R_Value".equals(type)) {
  247. industry.setFjTwo(value);
  248. } else if ("信义污水厂GFJ3_R_Value".equals(type)) {
  249. industry.setFjThree(value);
  250. } else if ("信义污水厂GFJ4_R_Value".equals(type)) {
  251. industry.setFjFour(value);
  252. } else if ("信义污水厂GFJ5_R_Value".equals(type)) {
  253. industry.setFjFive(value);
  254. } else if ("信义污水厂GFJ6_R_Value".equals(type)) {
  255. industry.setFjSix(value);
  256. } else if ("信义污水厂GFJ1_KQLL_Value".equals(type)) {
  257. industry.setKqllOne(value);
  258. } else if ("信义污水厂GFJ2_KQLL_Value".equals(type)) {
  259. industry.setKqllTwo(value);
  260. } else if ("信义污水厂GFJ3_KQLL_Value".equals(type)) {
  261. industry.setKqllThree(value);
  262. } else if ("信义污水厂GFJ4_KQLL_Value".equals(type)) {
  263. industry.setKqllFour(value);
  264. } else if ("信义污水厂GFJ5_KQLL_Value".equals(type)) {
  265. industry.setKqllFive(value);
  266. } else if ("信义污水厂GFJ6_KQLL_Value".equals(type)) {
  267. industry.setKqllSix(value);
  268. }else if ("信义污水厂实际碳源加药量".equals(type)) {
  269. industry.setSJTYJLY(value);
  270. }else if ("信义污水厂除磷加药瞬时流量".equals(type)) {
  271. industry.setCLJYSSLL(value);
  272. } else if ("信义污水厂_除磷P04预测值_".equals(type)) {
  273. industry.setCLP04YCZ(value);
  274. }
  275. //只有最后一次才执行数据库添加
  276. if(finalJ == list.size()-1){
  277. needMap.put(industry.getTestHour(), industry);
  278. }
  279. });
  280. }
  281. }
  282. //保存数据 触发告警 决策 问答记录等等
  283. needMap.forEach((k, industry) ->{
  284. //2024年4月22日15:45:24 额外保存两个字段 数组
  285. List<BigDecimal> extraList = new ArrayList<>();
  286. extraList.add(industry.getOneHymdDo());
  287. extraList.add(industry.getTwoHymdDo());
  288. industry.setHycRjyAll(JSON.toJSONString(extraList));
  289. extraList.clear();
  290. extraList.add(industry.getOneHyzdDo());
  291. extraList.add(industry.getTwoHyzdDo());
  292. industry.setHycRjyZdAll(JSON.toJSONString(extraList));
  293. extraList.clear();
  294. extraList.add(industry.getOneMlss());
  295. extraList.add(industry.getTwoMlss());
  296. industry.setHycWnndAll(JSON.toJSONString(extraList));
  297. //插入数据库
  298. xinyiIndustryMapper.insertTXinyiIndustry(industry);
  299. //判断是否触发告警、接触告警、保存决策等等
  300. this.handleWarning(industry);
  301. });
  302. //执行完成 测试执行时间
  303. //计时结束
  304. // watch.stop();
  305. // System.out.println(watch.getLastTaskName() + " 执行耗时:" + watch.getLastTaskTimeMillis() + " ms");
  306. }
  307. private void handleWarning(TXinyiIndustry tXinyiIndustry) {
  308. log.info("进入了定时任务保存工业库数据并触发报警操作");
  309. //获取配置表
  310. List<TXinyiNormConfig> tXinyiNormConfigs = this.xinyiNormConfigMapper.selectTXinyiNormConfigList(null);
  311. if(CollectionUtils.isEmpty(tXinyiNormConfigs))
  312. throw new RuntimeException("未查询到配置信息");
  313. TXinyiNormConfig normConfig = tXinyiNormConfigs.get(0);
  314. //水质报警
  315. this.handleSZWarning(tXinyiIndustry, normConfig);
  316. //2024年5月28日14:14:26 下面是新增的 生化报警处理
  317. this.handleSHWarning(tXinyiIndustry, normConfig);
  318. }
  319. private void handleSHWarning(TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  320. //判断对应指标是否报警 然后调研大模型获取决策信息
  321. BigDecimal jsSlq = tXinyiIndustry.getJsSlq();
  322. //内回流比报警
  323. BigDecimal nhlbSjz = normConfig.getNhlbSjz();//400
  324. BigDecimal nhlbnkSxz = normConfig.getNhlbnkSxz();//360
  325. BigDecimal nhlbqdsl = normConfig.getNHLBQDSL();
  326. BigDecimal nhlbdsjll = normConfig.getNHLBDSJLL();
  327. BigDecimal nhlbgzxl = normConfig.getNHLBGZXL();
  328. if(!Objects.isNull(jsSlq) && !Objects.isNull(nhlbnkSxz) && !Objects.isNull(nhlbqdsl)
  329. && !Objects.isNull(nhlbdsjll) && !Objects.isNull(nhlbgzxl)){
  330. BigDecimal divide = nhlbqdsl.multiply(nhlbdsjll).multiply(nhlbgzxl).divide(jsSlq, 4, RoundingMode.HALF_UP);
  331. if(!Objects.isNull(nhlbSjz)){
  332. handleXinYiWarningsSH(nhlbSjz, divide, nhlbnkSxz, BusinessEnum.WarningCategoryEnum.NHLB.getCode(), tXinyiIndustry, normConfig);
  333. }
  334. }
  335. //外回流比报警
  336. BigDecimal whlbqdsl = normConfig.getWHLBQDSL();
  337. BigDecimal whlbdsjll = normConfig.getWHLBDSJLL();
  338. BigDecimal whlbgzxl = normConfig.getWHLBGZXL();
  339. BigDecimal whlbnkXxz = normConfig.getWhlbnkXxz();
  340. BigDecimal whlbSjz = normConfig.getWhlbSjz();
  341. if(!Objects.isNull(whlbqdsl) && !Objects.isNull(whlbdsjll) && !Objects.isNull(whlbgzxl)){
  342. BigDecimal divide = whlbqdsl.multiply(whlbdsjll).multiply(whlbgzxl).divide(jsSlq, 4, RoundingMode.HALF_UP);
  343. if(!Objects.isNull(whlbSjz)){
  344. handleXinYiWarningsSH(whlbSjz, divide, whlbnkXxz, BusinessEnum.WarningCategoryEnum.WHLB.getCode(), tXinyiIndustry, normConfig);
  345. }
  346. }
  347. //最新的一条日报数据
  348. List<TXinyiDaily> tXinyiDailies = this.xinyiDailyMapper.selectTXinyiDailyList(null);
  349. if(CollectionUtils.isEmpty(tXinyiDailies))
  350. return;//肯定不会出现这种情况 因为有很多历史数据了
  351. TXinyiDaily tXinyiDaily = tXinyiDailies.get(0);
  352. log.info("生化报警获取日报的最新的一条数据为{}", JSON.toJSONString(tXinyiDaily));
  353. BigDecimal jsBod5 = tXinyiDaily.getJsBod5();
  354. //污泥负荷(需要从日报获取数据) 计算 + 部分数据从日报获取
  355. //污泥负荷=[Ls]=24*([BOD_in]-[BOD_off])*[Q_in]/[MLSS]/([V_hao]+[V_que]+[V_yan])/2 kgBOD/(kgMLSS·d)
  356. //todo 后面再加 计算太复杂
  357. //碳氮比(需要从日报获取数据)
  358. //进水碳氮比=[b]=[BOD_in]/[TN_in]
  359. BigDecimal jsTn = tXinyiIndustry.getJsTn();
  360. if(!Objects.isNull(jsBod5) && !Objects.isNull(jsTn)){
  361. BigDecimal divide = jsBod5.divide(jsTn, 4, RoundingMode.HALF_UP);
  362. BigDecimal jstdbnkzSxz = normConfig.getJstdbnkzSxz();
  363. BigDecimal cstdbSjz = normConfig.getCstdbSjz();
  364. if(!Objects.isNull(whlbSjz)){
  365. handleXinYiWarningsSH(cstdbSjz, divide, jstdbnkzSxz, BusinessEnum.WarningCategoryEnum.TDB.getCode(), tXinyiIndustry, normConfig);
  366. }
  367. }
  368. //碳磷比(需要从日报获取数据)
  369. //进水碳磷比=[c]=[BOD_in]/[TP_in]
  370. BigDecimal jsTp = tXinyiIndustry.getJsTp();
  371. BigDecimal jstlbSjz = normConfig.getJstlbSjz();
  372. if(!Objects.isNull(jsBod5) && !Objects.isNull(jsTp)){
  373. BigDecimal divide = jsBod5.divide(jsTp, 4, RoundingMode.HALF_UP);
  374. BigDecimal jstlbNkz = normConfig.getJstlbNkz();
  375. if(!Objects.isNull(jstlbSjz)){
  376. handleXinYiWarningsSH(jstlbSjz, divide, jstlbNkz, BusinessEnum.WarningCategoryEnum.TLB.getCode(), tXinyiIndustry, normConfig);
  377. }
  378. }
  379. //BOD比COD(需要从日报获取数据)
  380. //进水BOD与COD比值数据=[d]=[BOD_in]/[COD_in]
  381. BigDecimal jsCod = tXinyiIndustry.getJsCod();
  382. if(!Objects.isNull(jsBod5) && !Objects.isNull(jsCod)){
  383. BigDecimal jsbodycodbzSzj = normConfig.getJsbodycodbzSzj();
  384. String jsbodycodbzGkz = normConfig.getJsbodycodbzGkz();
  385. if(StringUtils.isNotBlank(jsbodycodbzGkz) && jsbodycodbzGkz.contains("-")){
  386. String[] split = jsbodycodbzGkz.split("-");
  387. BigDecimal gkz = new BigDecimal(split[1]);
  388. BigDecimal divide = jsBod5.divide(jsCod, 4, RoundingMode.HALF_UP);
  389. handleXinYiWarningsSH(gkz, divide, jsbodycodbzSzj, BusinessEnum.WarningCategoryEnum.BODCODB.getCode(), tXinyiIndustry, normConfig);
  390. }
  391. }
  392. //好氧区DO(一池)
  393. BigDecimal shcHyOneDo = tXinyiIndustry.getOneHymdDo();
  394. BigDecimal hycrjysjzSxz = normConfig.getHycrjysjzSxz();
  395. BigDecimal hycrjyNkz = normConfig.getHycrjyNkz();
  396. if(!Objects.isNull(shcHyOneDo) && !Objects.isNull(hycrjysjzSxz)){
  397. handleXinYiWarningsSH(hycrjysjzSxz, shcHyOneDo, hycrjyNkz, BusinessEnum.WarningCategoryEnum.HYQDO_ONE.getCode(), tXinyiIndustry, normConfig);
  398. }
  399. //好氧区DO(二池)
  400. BigDecimal shcHyTwoDo = tXinyiIndustry.getTwoHymdDo();
  401. if(!Objects.isNull(shcHyTwoDo) && !Objects.isNull(hycrjysjzSxz)){
  402. handleXinYiWarningsSH(hycrjysjzSxz, shcHyTwoDo, hycrjyNkz, BusinessEnum.WarningCategoryEnum.HYQDO_TWO.getCode(), tXinyiIndustry, normConfig);
  403. }
  404. //气水比(需要从日报获取数据)
  405. BigDecimal gfjgzts = normConfig.getGFJGZTS();
  406. BigDecimal gfjckll = normConfig.getGFJCKLL();
  407. BigDecimal shcqbSjz = normConfig.getShcqbSjz();
  408. if(!Objects.isNull(gfjgzts) && !Objects.isNull(gfjckll) && !Objects.isNull(shcqbSjz) && !Objects.isNull(jsSlq)){
  409. BigDecimal qsb = gfjgzts.multiply(gfjckll).divide(jsSlq, 4, RoundingMode.HALF_UP);
  410. BigDecimal shcqbNkz = normConfig.getShcqbNkz();
  411. handleXinYiWarningsSH(shcqbSjz, qsb, shcqbNkz, BusinessEnum.WarningCategoryEnum.WHLB.getCode(), tXinyiIndustry, normConfig);
  412. }
  413. }
  414. /**
  415. * 处理信义生化报警的逻辑 统一处理
  416. * @param bzz
  417. * @param currentVal
  418. * @param gkz
  419. * @param category
  420. * @param tXinyiIndustry
  421. * @param normConfig
  422. */
  423. private void handleXinYiWarningsSH(BigDecimal bzz, BigDecimal currentVal, BigDecimal gkz, String category, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  424. BigDecimal multiply = bzz.multiply(new BigDecimal(MyConstants.SCALE_VALUE));
  425. TXinyiWarningRecord tXinyiWarningRecord = new TXinyiWarningRecord();
  426. /*String category = BusinessEnum.WarningCategoryEnum.CS_AD.getCode();*/
  427. tXinyiWarningRecord.setStatus(0);
  428. tXinyiWarningRecord.setType(1);
  429. tXinyiWarningRecord.setCategory(category);
  430. tXinyiWarningRecord.setTime(DateUtils.getNowDate());
  431. tXinyiWarningRecord.setWarningVal(currentVal);
  432. tXinyiWarningRecord.setDesignVal(bzz);
  433. tXinyiWarningRecord.setControlVal(gkz);
  434. tXinyiWarningRecord.setCreateBy(WARNING_DEFAULT_CREATE);
  435. tXinyiWarningRecord.setCreateTime(DateUtils.getNowDate());
  436. //2024年5月25日17:52:33 如果工业库获取不到数据,也触发报警,但是不调用决策接口
  437. if (Objects.isNull(currentVal)) {
  438. tXinyiWarningRecord.setReason(category + EXCEPTION_WARNING);
  439. tXinyiWarningRecord.setLevel(WARNING_LEVEL_NO_DATE);
  440. } else if (currentVal.compareTo(multiply) > 0) {//一级
  441. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  442. tXinyiWarningRecord.setLevel(WARNING_LEVEL_ONE);
  443. } else if (currentVal.compareTo(bzz) >= 0 && currentVal.compareTo(multiply) <= 0) {//二级
  444. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  445. tXinyiWarningRecord.setLevel(WARNING_LEVEL_TWO);
  446. } else if (!Objects.isNull(gkz) && currentVal.compareTo(gkz) > 0) {
  447. tXinyiWarningRecord.setReason(category + CHAOGUANKONG_WARNING);
  448. tXinyiWarningRecord.setLevel(WARNING_LEVEL_THREE);
  449. } else {
  450. tXinyiWarningRecord = null;//这种的无需处理
  451. }
  452. //当前状态正常 需要查询历史有无正在报警的数据,如果有,将报警状态改完2(系统自动关闭)
  453. List<TXinyiWarningRecord> tXinyiWarningRecords = this.xinyiWarningRecordMapper.selectTXinyiWarningRecordList(TXinyiWarningRecord.builder().delFlag(0).type(1).category(category).status(0).build());
  454. if(Objects.isNull(tXinyiWarningRecord)){//数据正常,无告警信息
  455. if(!CollectionUtils.isEmpty(tXinyiWarningRecords)){
  456. log.info( "{}:现在恢复正常,历史报警数据为{}", category,JSON.toJSONString(tXinyiWarningRecords));
  457. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {
  458. xinyiWarningRecord.setStatus(2);
  459. Date nowDate = DateUtils.getNowDate();
  460. xinyiWarningRecord.setOffTime(nowDate);
  461. xinyiWarningRecord.setUpdateTime(nowDate);
  462. xinyiWarningRecord.setUpdateBy(WARNING_DEFAULT_CREATE);
  463. this.xinyiWarningRecordMapper.updateTXinyiWarningRecord(xinyiWarningRecord);
  464. }
  465. }
  466. }else{//有新的告警信息
  467. if(CollectionUtils.isEmpty(tXinyiWarningRecords)){//之前没有告警记录
  468. //保存到数据库中
  469. this.xinyiWarningRecordMapper.insertTXinyiWarningRecord(tXinyiWarningRecord);
  470. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  471. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  472. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  473. }else {
  474. //继续调用决策(普通问答)
  475. this.askBigModelForSHWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  476. }
  477. }else{
  478. log.info("{}:之前已经有过告警记录了,且还是继续报警,无需重复添加报警,但是决策仍然要调用", category);
  479. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {//理论上只有一个的
  480. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  481. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  482. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  483. }else {
  484. //继续调用决策(普通问答)
  485. this.askBigModelForSHWarning(xinyiWarningRecord, tXinyiIndustry, normConfig);
  486. }
  487. }
  488. }
  489. }
  490. }
  491. private void askBigModelForSHWarning(TXinyiWarningRecord xinyiWarningRecord, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  492. log.info("进入了后台接口调⽤⼤模型获取问答结果处理");
  493. StringBuilder sb = new StringBuilder();
  494. String sessionId = IdUtils.simpleUUID();
  495. ChatReq chatReq = new ChatReq();
  496. // String ipAddr = IpUtils.getIpAddr();//获取用户的ip地址 传给大模型
  497. String ipAddr = "";//获取用户的ip地址 传给大模型 定时任务获取不到ip地址
  498. int counts = 1;//默认是第一次
  499. //这种问答 没有历史问答的概念 直接把问题扔进去就行 无需查询历史记录
  500. List<String> historyDates = new ArrayList<>();
  501. //构建问题(替换提示词中的占位符)
  502. String shWarningPrompt = SH_WARNING_PROMPT;
  503. shWarningPrompt =shWarningPrompt.replace("#{0}", xinyiWarningRecord.getReason());
  504. shWarningPrompt =shWarningPrompt.replace("#{1}", String.valueOf(xinyiWarningRecord.getDesignVal()));
  505. shWarningPrompt =shWarningPrompt.replace("#{2}", String.valueOf(xinyiWarningRecord.getControlVal()));
  506. shWarningPrompt =shWarningPrompt.replace("#{3}", String.valueOf(xinyiWarningRecord.getWarningVal()));
  507. historyDates.add(shWarningPrompt);
  508. // 获取输出流
  509. ManagedChannel channel = null;
  510. try {
  511. channel = ManagedChannelBuilder.forAddress("10.0.0.24", 17070)
  512. .usePlaintext()
  513. .build();
  514. InferenceAPIsServiceGrpc.InferenceAPIsServiceBlockingStub stub = InferenceAPIsServiceGrpc.newBlockingStub(channel);
  515. String dataJson = "{\"bot_id\":\"721\",\"exp_id\":\"721\",\"session_id\":\"" + sessionId + "\",\"use_rag\":\"true\",\"prompt\":\"你是⼀个资深⽔务领域专家,能回答各种⽔务相关问题\",\"history_dia\":" + JSON.toJSONString(historyDates) + ",\"generate_args\":{\"max_new_tokens\":2048,\"max_length\":4096,\"num_beams\":1,\"do_sample\":true,\"top_p\":0.7,\"temperature\":0.95},\"extra\":{ \"ip_address\": \"" + ipAddr + "\" },\"strengthen\":" + (true) + "}";
  516. log.info("请求大模型的问答参数为{}", dataJson);
  517. PredictionsRequest request = PredictionsRequest.newBuilder()
  518. .setModelName("slibra_bot")
  519. .putInput("method", ByteString.copyFrom("infer_stream", "utf-8"))//推理
  520. .putInput("data", ByteString.copyFrom(dataJson, "utf-8"))
  521. .buildPartial();
  522. Iterator<PredictionResponse> predictions = stub.streamPredictions(request);
  523. //将结果记录到问答表
  524. while (predictions.hasNext()) {
  525. String responseStr = predictions.next().getPrediction().toStringUtf8();
  526. log.info("大模型问答返回的原始结果为{}", responseStr);
  527. responseStr = JSON.parseObject(responseStr).getString("message");
  528. if("complete".equals(responseStr)){
  529. log.info("结尾语句并且是非JSON,无需处理");
  530. }else{
  531. sb.append(responseStr);
  532. }
  533. }
  534. //将问答更新到数据库中
  535. chatReq.setSessionId(sessionId);
  536. chatReq.setType(2);//0问答 1决策 2本地
  537. chatReq.setModule(3);//0专家问答 1智能工单 2智能体助手 3告警 4简报
  538. String showVal = this.buildShowValue(xinyiWarningRecord, tXinyiIndustry, normConfig);
  539. chatReq.setShowVal(showVal);
  540. chatReq.setQuestion(shWarningPrompt);
  541. chatReq.setAnswer(sb.toString());
  542. chatReq.setWarningId(String.valueOf(xinyiWarningRecord.getId()));
  543. chatReq.setCounts(counts);//问答次数
  544. chatReq.setUserId(WARNING_DEFAULT_CREATE);
  545. chatReq.setCreateBy(WARNING_DEFAULT_CREATE);
  546. chatReq.setCreateTime(DateUtils.getNowDate());
  547. this.xinyiChatRecordMapper.insertTXinyiChatRecord(chatReq);
  548. } catch (IOException e) {
  549. throw new RuntimeException(e);
  550. } finally {
  551. // 关闭输出流
  552. channel.shutdown();
  553. }
  554. }
  555. private void handleSZWarning(TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  556. //出水相关
  557. //出水COD报警
  558. BigDecimal csCod = tXinyiIndustry.getCsCod();
  559. BigDecimal cscodBzz = normConfig.getCscodBzz();
  560. BigDecimal cscodGkz = normConfig.getCscodGkz();
  561. if(!Objects.isNull(cscodBzz)){
  562. handleXinYiWarningsCs(cscodBzz, csCod, cscodGkz, BusinessEnum.WarningCategoryEnum.CS_COD.getCode(), tXinyiIndustry, normConfig);
  563. }
  564. //出水总磷超标报警
  565. BigDecimal csTp = tXinyiIndustry.getCsTp();
  566. BigDecimal cszlBzz = normConfig.getCszlBzz();
  567. BigDecimal cszlGkz = normConfig.getCszlGkz();
  568. if(!Objects.isNull(cszlBzz)){
  569. handleXinYiWarningsCs(cszlBzz, csTp, cszlGkz, BusinessEnum.WarningCategoryEnum.CS_ZL.getCode(), tXinyiIndustry, normConfig);
  570. }
  571. //出水总氮超标报警
  572. BigDecimal csTn = tXinyiIndustry.getCsTn();
  573. BigDecimal cszzBzz = normConfig.getCszzBzz();
  574. BigDecimal cszzGkz = normConfig.getCszzGkz();
  575. if(!Objects.isNull(cszzBzz)){
  576. handleXinYiWarningsCs(cszzBzz, csTn, cszzGkz, BusinessEnum.WarningCategoryEnum.CS_ZD.getCode(), tXinyiIndustry, normConfig);
  577. }
  578. //出水氨氮超标报警
  579. BigDecimal csNh3 = tXinyiIndustry.getCsNh3();
  580. BigDecimal csadBzz = normConfig.getCsadBzz();
  581. BigDecimal csadGkz = normConfig.getCsadGkz();
  582. if(!Objects.isNull(csadBzz)){
  583. handleXinYiWarningsCs(csadBzz, csNh3, csadGkz, BusinessEnum.WarningCategoryEnum.CS_AD.getCode(), tXinyiIndustry, normConfig);
  584. }
  585. //出水SS超标报警
  586. BigDecimal csSS = tXinyiIndustry.getCsSs();
  587. BigDecimal csSSBzz = normConfig.getCsssBzz();
  588. BigDecimal csssGkz = normConfig.getCsssGkz();
  589. if(!Objects.isNull(csSSBzz)){
  590. handleXinYiWarningsCs(csSSBzz, csSS, csssGkz, BusinessEnum.WarningCategoryEnum.CS_SS.getCode(), tXinyiIndustry, normConfig);
  591. }
  592. //进水相关报警
  593. //进水总磷超标报警
  594. BigDecimal jsTp = tXinyiIndustry.getJsTp();
  595. BigDecimal jszlSjz = normConfig.getJszlSjz();
  596. if(!Objects.isNull(jszlSjz)){
  597. handleXinYiWarningRecordJS(jszlSjz, jsTp, BusinessEnum.WarningCategoryEnum.JS_ZL.getCode(), tXinyiIndustry, normConfig);
  598. }
  599. //进水COD超标报警
  600. BigDecimal jsCod = tXinyiIndustry.getJsCod();
  601. BigDecimal jscodSjz = normConfig.getJscodSjz();
  602. if(!Objects.isNull(jscodSjz)){
  603. handleXinYiWarningRecordJS(jscodSjz, jsCod, BusinessEnum.WarningCategoryEnum.JS_COD.getCode(), tXinyiIndustry, normConfig);
  604. }
  605. //进水总氮超标报警
  606. BigDecimal jsTn = tXinyiIndustry.getJsTn();
  607. BigDecimal jszdSjz = normConfig.getJszdSjz();
  608. if(!Objects.isNull(jszdSjz)){
  609. handleXinYiWarningRecordJS(jszdSjz, jsTn, BusinessEnum.WarningCategoryEnum.JS_ZD.getCode(), tXinyiIndustry, normConfig);
  610. }
  611. //进水氨氮超标报警
  612. BigDecimal jsNh3 = tXinyiIndustry.getJsNh3();
  613. BigDecimal jsadSjz = normConfig.getJsadSjz();
  614. if(!Objects.isNull(jsadSjz)){
  615. handleXinYiWarningRecordJS(jsadSjz, jsNh3, BusinessEnum.WarningCategoryEnum.JS_AD.getCode(), tXinyiIndustry, normConfig);
  616. }
  617. //进水SS超标报警
  618. BigDecimal jsSS = tXinyiIndustry.getJsSs();
  619. BigDecimal jsSSSjz = normConfig.getJsssSjz();
  620. if(!Objects.isNull(jsSSSjz)){
  621. handleXinYiWarningRecordJS(jsSSSjz, jsSS, BusinessEnum.WarningCategoryEnum.JS_SS.getCode(), tXinyiIndustry, normConfig);
  622. }
  623. }
  624. /**
  625. * 调用大模型获取决策结果 并同时记录对应信息到聊天记录表中
  626. *
  627. * @param tXinyiWarningRecord
  628. * @param tXinyiIndustry
  629. * @param normConfig
  630. */
  631. private void handleDecision(TXinyiWarningRecord tXinyiWarningRecord, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  632. log.info("进入了调⽤大模型决策接口");
  633. ChatReq chatReq = new ChatReq();
  634. // StringBuilder sb = new StringBuilder();
  635. //大模型结果 放入一个结合中
  636. List<String> resultData = new ArrayList<>();
  637. //决策和问答不一样 没有历史的概念 所以sessionId都是新的 次数都是1
  638. String sessionId = IdUtils.simpleUUID();
  639. String feedback = chatReq.getFeedback();
  640. //决策请求的业务参数
  641. // List<HashMap<String, Object>> list = this.xinyiIndustryMapper.selectLast10RecordsForDecision();
  642. //2024年5月21日15:23:07 这里不能用关联查询处理,日报要获取最新的一条而不是今日的数据。
  643. List<DecisionReq> decisionReqs = this.xinyiIndustryMapper.selectLast10RecordsForDecisionOnlyIndustry();
  644. if(!CollectionUtils.isEmpty(decisionReqs)){
  645. for (DecisionReq decisionReq : decisionReqs) {
  646. //处理日报数据
  647. TXinyiDaily daily = this.xinyiDailyMapper.selectNewestData();
  648. if(!Objects.isNull(daily)){
  649. decisionReq.setT(daily.getJsSw());
  650. decisionReq.setSVI(daily.getWntjzsAll());
  651. decisionReq.setSV(daily.getWncjbAll());
  652. decisionReq.setMlvss(daily.getHfxwnndAll());
  653. decisionReq.setDoAna(daily.getYycRjyAll());
  654. decisionReq.setDoQue(daily.getQycRjyAll());
  655. //2024年5月26日11:59:02 干污泥量数据做了同步
  656. decisionReq.setGwnl(daily.getGWNL());
  657. BigDecimal jsBod5 = daily.getJsBod5();
  658. decisionReq.setBodIn(jsBod5);
  659. BigDecimal tpIn = decisionReq.getTpIn();
  660. BigDecimal tnIn = decisionReq.getTnIn();
  661. if(!Objects.isNull(jsBod5)){
  662. if(!Objects.isNull(tpIn) && tpIn.compareTo(new BigDecimal("0")) >0){
  663. decisionReq.setC(jsBod5.divide(tpIn, 4, RoundingMode.HALF_UP));
  664. }
  665. if(!Objects.isNull(tnIn) && tnIn.compareTo(new BigDecimal("0")) >0){
  666. decisionReq.setB(jsBod5.divide(tnIn, 4, RoundingMode.HALF_UP));
  667. }
  668. }
  669. }
  670. }
  671. }
  672. String rows = JSON.toJSONString(decisionReqs, JSONWriter.Feature.WriteNulls);
  673. // 获取输出流
  674. ManagedChannel channel = null;
  675. String dataJson = "";
  676. try {
  677. channel = ManagedChannelBuilder.forAddress("10.0.0.24", 17070)
  678. .usePlaintext()
  679. .build();
  680. InferenceAPIsServiceGrpc.InferenceAPIsServiceBlockingStub stub = InferenceAPIsServiceGrpc.newBlockingStub(channel);
  681. dataJson = "{\"bot_id\":\"b00001\",\"exp_id\":\"721\",\"norm\":\"" + tXinyiWarningRecord.getCategory() + "\",\"feedback\":" + feedback + ",\"session_id\":" + "\"" + sessionId + "\"" + ",\"generate_args\":{\"max_new_tokens\":1024,\"max_length\":4096,\"num_beams\":1,\"do_sample\":true,\"top_p\":0.7,\"temperature\":0.95},\"extra\":{\"rows\":" + rows + "}}";
  682. log.info("请求大模型的决策的参数为{}", dataJson);
  683. PredictionsRequest request = PredictionsRequest.newBuilder()
  684. .setModelName("slibra_bot")
  685. .putInput("method", ByteString.copyFrom("decision_stream", "utf-8"))//推理
  686. .putInput("data", ByteString.copyFrom(dataJson, "utf-8"))
  687. .buildPartial();
  688. Iterator<PredictionResponse> predictions = stub.streamPredictions(request);
  689. while (predictions.hasNext()) {
  690. String responseStr = predictions.next().getPrediction().toStringUtf8();
  691. log.info("决策流式返回的结果是{}", responseStr);
  692. //2024年5月25日16:37:16 按照大模型返回的类型解析数据
  693. String biz = JSON.parseObject(responseStr).getString("biz");
  694. if(BusinessEnum.BigModelBizEnum.OK.getCode().equals(biz)){
  695. log.info("结尾语句并且是非JSON,无需处理");
  696. //结束语句也流式输出,但是并不记录下来 2024年5月24日11:15:23 也不返回前端
  697. /*outputStream.write(responseStr.getBytes());
  698. outputStream.flush();*/
  699. }else if(BusinessEnum.BigModelBizEnum.DECISION_DEBUGGER.getCode().equals(biz)){
  700. log.info("中间过程,目前只打印日志,不记录数据,也不返回给前端,返回数据为{}", responseStr);
  701. //结束语句也流式输出,但是并不记录下来 2024年5月24日11:15:23 也不返回前端
  702. /*outputStream.write(responseStr.getBytes());
  703. outputStream.flush();*/
  704. }else{//其他 要么错误 要么alert 要么出的报告
  705. // sb.append(responseStr);
  706. resultData.add(responseStr);
  707. }
  708. }
  709. } catch (Exception e) {
  710. // throw new RuntimeException(e);
  711. log.error("定时任务处理告警调用决策异常,异常信息为{}", JSON.toJSONString(e));
  712. resultData.add("{\"biz\":\"ERROR\",\"message\":\"大模型分析数据异常,请稍后再试\"}");
  713. } finally {
  714. log.info("决策最终要保存的数据是{}", JSON.toJSONString(resultData));
  715. //保存聊天记录
  716. //将问答更新到数据库中
  717. chatReq.setSessionId(sessionId);
  718. chatReq.setType(1);//0问答 1决策
  719. chatReq.setModule(3);
  720. /*String userId = SecurityUtils.getUserId().toString();
  721. String username = SecurityUtils.getUsername();*/
  722. chatReq.setUserId(WARNING_DEFAULT_CREATE);
  723. String showVal = this.buildShowValue(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  724. chatReq.setShowVal(showVal);//前端展示的数据和提问的数据不一致
  725. chatReq.setQuestion(dataJson);
  726. chatReq.setAnswer(JSON.toJSONString(resultData));
  727. chatReq.setWarningId(String.valueOf(tXinyiWarningRecord.getId()));
  728. chatReq.setCounts(1);//问答次数
  729. chatReq.setCreateBy(WARNING_DEFAULT_CREATE);
  730. chatReq.setCreateTime(DateUtils.getNowDate());
  731. this.xinyiChatRecordMapper.insertTXinyiChatRecord(chatReq);
  732. // 关闭输出流
  733. if(!Objects.isNull(channel))
  734. channel.shutdown();
  735. }
  736. }
  737. private String buildShowValue(TXinyiWarningRecord tXinyiWarningRecord, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  738. JSONObject result = new JSONObject();
  739. JSONObject basic = new JSONObject();
  740. Integer status = tXinyiWarningRecord.getStatus();
  741. Date warningTime = tXinyiWarningRecord.getTime();
  742. int count = DateUtils.differentHoursByMillisecond(warningTime, DateUtils.getNowDate()) + 1;
  743. basic.put("title", tXinyiWarningRecord.getReason());
  744. basic.put("报警时间", DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD_HH_MM ,warningTime));
  745. basic.put("报警值", tXinyiWarningRecord.getWarningVal());
  746. basic.put("标准值", tXinyiWarningRecord.getDesignVal());
  747. basic.put("管控值", tXinyiWarningRecord.getControlVal());
  748. basic.put("报警次数", Math.min(count, MAX_COUNT));
  749. if(tXinyiWarningRecord.getType() != 2)
  750. basic.put("状态", status == 0 ? "告警中" : status == 1 ? "用户关闭" : status == 2 ? "系统自动关闭" : "用户转为应急处理中");
  751. else
  752. basic.put("状态", status == 0 ? "预警中" : "已完成");
  753. //2024年5月27日14:04:22 额外返回2个字段 [管控值 和 告警级别] 返回的json没有数据是因为value没有值
  754. // basic.put("管控值", tXinyiWarningRecord.getControlVal());
  755. basic.put("告警级别", tXinyiWarningRecord.getLevel());
  756. result.put("basic", basic);
  757. JSONObject jsData = getJsonObject(tXinyiIndustry, normConfig);//进水数据
  758. result.put("jsData", jsData);
  759. JSONObject csData = getCsonObject(tXinyiIndustry, normConfig);//出水数据
  760. result.put("csData", csData);
  761. return JSON.toJSONString(result, JSONWriter.Feature.WriteNulls);
  762. }
  763. private static JSONObject getJsonObject(TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  764. JSONObject jsData = new JSONObject();
  765. HashMap<Object, Object> temp1 = new HashMap<>();
  766. BigDecimal jsSlq = tXinyiIndustry.getJsSlq();
  767. temp1.put("value", jsSlq);
  768. temp1.put("exceed", jsSlq.compareTo(normConfig.getJsslSjz()) >0);
  769. jsData.put("流量", temp1);
  770. HashMap<Object, Object> temp2 = new HashMap<>();
  771. BigDecimal jsCod = tXinyiIndustry.getJsCod();
  772. temp2.put("value", jsCod);
  773. temp2.put("exceed", jsCod.compareTo(normConfig.getJscodSjz()) > 0);
  774. jsData.put("COD", temp2);
  775. HashMap<Object, Object> temp3 = new HashMap<>();
  776. BigDecimal jsNh3 = tXinyiIndustry.getJsNh3();
  777. temp3.put("value", jsNh3);
  778. temp3.put("exceed", jsNh3.compareTo(normConfig.getJsadSjz()) > 0);
  779. jsData.put("NH3-N", temp3);
  780. HashMap<Object, Object> temp4 = new HashMap<>();
  781. BigDecimal jsTp = tXinyiIndustry.getJsTp();
  782. temp4.put("value", jsTp);
  783. temp4.put("exceed", jsTp.compareTo(normConfig.getJszlSjz()) > 0);
  784. jsData.put("TP", temp4);
  785. HashMap<Object, Object> temp5 = new HashMap<>();
  786. BigDecimal jsSs = tXinyiIndustry.getJsSs();
  787. temp5.put("value", jsSs);
  788. temp5.put("exceed", jsSs.compareTo(normConfig.getJsssSjz()) > 0);
  789. jsData.put("SS", temp5);
  790. HashMap<Object, Object> temp6 = new HashMap<>();
  791. BigDecimal jsTn = tXinyiIndustry.getJsTn();
  792. temp6.put("value", jsTn);
  793. temp6.put("exceed", jsTn.compareTo(normConfig.getJszdSjz()) > 0);
  794. jsData.put("TN", temp6);
  795. return jsData;
  796. }
  797. private static JSONObject getCsonObject(TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  798. JSONObject csData = new JSONObject();
  799. HashMap<Object, Object> temp1 = new HashMap<>();
  800. BigDecimal csSlq = tXinyiIndustry.getCsSlqc();
  801. temp1.put("value", csSlq);
  802. temp1.put("exceed", false);//出水水量没有管控值
  803. csData.put("流量", temp1);
  804. HashMap<Object, Object> temp2 = new HashMap<>();
  805. BigDecimal csCod = tXinyiIndustry.getCsCod();
  806. temp2.put("value", csCod);
  807. temp2.put("exceed", csCod.compareTo(normConfig.getCscodGkz()) > 0);
  808. csData.put("COD", temp2);
  809. HashMap<Object, Object> temp3 = new HashMap<>();
  810. BigDecimal csNh3 = tXinyiIndustry.getCsNh3();
  811. temp3.put("value", csNh3);
  812. temp3.put("exceed", csNh3.compareTo(normConfig.getCsadGkz()) > 0);
  813. csData.put("NH3-N", temp3);
  814. HashMap<Object, Object> temp4 = new HashMap<>();
  815. BigDecimal csTp = tXinyiIndustry.getCsTp();
  816. temp4.put("value", csTp);
  817. temp4.put("exceed", csTp.compareTo(normConfig.getCszlGkz()) > 0);
  818. csData.put("TP", temp4);
  819. HashMap<Object, Object> temp5 = new HashMap<>();
  820. BigDecimal csSs = tXinyiIndustry.getCsSs();
  821. temp5.put("value", csSs);
  822. temp5.put("exceed", csSs.compareTo(normConfig.getCsssGkz()) > 0);
  823. csData.put("SS", temp5);
  824. HashMap<Object, Object> temp6 = new HashMap<>();
  825. BigDecimal csTn = tXinyiIndustry.getCsTn();
  826. temp6.put("value", csTn);
  827. temp6.put("exceed", csTn.compareTo(normConfig.getCszzGkz()) > 0);
  828. csData.put("TN", temp6);
  829. return csData;
  830. }
  831. /**
  832. * 通过输入的值 生成对应类型的报警对象(出水)
  833. *
  834. * @param csBzz
  835. * @param currentVal
  836. * @param csGkz
  837. * @param category
  838. * @param tXinyiIndustry
  839. * @param normConfig
  840. * @return
  841. */
  842. private void handleXinYiWarningsCs(BigDecimal csBzz, BigDecimal currentVal, BigDecimal csGkz, String category, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  843. BigDecimal multiply = csBzz.multiply(new BigDecimal(MyConstants.SCALE_VALUE));
  844. TXinyiWarningRecord tXinyiWarningRecord = new TXinyiWarningRecord();
  845. /*String category = BusinessEnum.WarningCategoryEnum.CS_AD.getCode();*/
  846. tXinyiWarningRecord.setStatus(0);
  847. tXinyiWarningRecord.setType(0);
  848. tXinyiWarningRecord.setCategory(category);
  849. tXinyiWarningRecord.setTime(DateUtils.getNowDate());
  850. tXinyiWarningRecord.setWarningVal(currentVal);
  851. tXinyiWarningRecord.setDesignVal(csBzz);
  852. tXinyiWarningRecord.setControlVal(csGkz);
  853. tXinyiWarningRecord.setCreateBy(WARNING_DEFAULT_CREATE);
  854. tXinyiWarningRecord.setCreateTime(DateUtils.getNowDate());
  855. //2024年5月25日17:52:33 如果工业库获取不到数据,也触发报警,但是不调用决策接口
  856. if(Objects.isNull(currentVal)){
  857. tXinyiWarningRecord.setReason(category + EXCEPTION_WARNING);
  858. tXinyiWarningRecord.setLevel(WARNING_LEVEL_NO_DATE);
  859. }else if(currentVal.compareTo(multiply) > 0){//一级
  860. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  861. tXinyiWarningRecord.setLevel(WARNING_LEVEL_ONE);
  862. }else if(currentVal.compareTo(csBzz) >= 0 && currentVal.compareTo(multiply) <= 0){//二级
  863. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  864. tXinyiWarningRecord.setLevel(WARNING_LEVEL_TWO);
  865. }else if(!Objects.isNull(csGkz) && currentVal.compareTo(csGkz) > 0){
  866. tXinyiWarningRecord.setReason(category + CHAOGUANKONG_WARNING);
  867. tXinyiWarningRecord.setLevel(WARNING_LEVEL_THREE);
  868. }else{
  869. tXinyiWarningRecord = null;//这种的无需处理
  870. }
  871. //当前状态正常 需要查询历史有无正在报警的数据,如果有,将报警状态改完2(系统自动关闭)
  872. List<TXinyiWarningRecord> tXinyiWarningRecords = this.xinyiWarningRecordMapper.selectTXinyiWarningRecordList(TXinyiWarningRecord.builder().delFlag(0).type(0).category(category).warningStatus(0).build());
  873. if(Objects.isNull(tXinyiWarningRecord)){//数据正常,无告警信息
  874. if(!CollectionUtils.isEmpty(tXinyiWarningRecords)){
  875. log.info( "{}:现在恢复正常,历史报警数据为{}", category,JSON.toJSONString(tXinyiWarningRecords));
  876. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {
  877. xinyiWarningRecord.setStatus(2);
  878. Date nowDate = DateUtils.getNowDate();
  879. xinyiWarningRecord.setOffTime(nowDate);
  880. xinyiWarningRecord.setUpdateTime(nowDate);
  881. xinyiWarningRecord.setUpdateBy(WARNING_DEFAULT_CREATE);
  882. this.xinyiWarningRecordMapper.updateTXinyiWarningRecord(xinyiWarningRecord);
  883. }
  884. }
  885. }else{//有告警信息
  886. if(CollectionUtils.isEmpty(tXinyiWarningRecords)){//之前没有告警记录
  887. //保存到数据库中
  888. this.xinyiWarningRecordMapper.insertTXinyiWarningRecord(tXinyiWarningRecord);
  889. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  890. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  891. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  892. }else {
  893. //继续调用决策
  894. this.handleDecision(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  895. }
  896. }else{
  897. log.info("{}:之前已经有过告警记录了,且还是继续报警,无需重复添加报警,但是决策仍然要调用", category);
  898. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {//理论上只有一个的
  899. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  900. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  901. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  902. }else {
  903. //继续调用决策
  904. this.handleDecision(xinyiWarningRecord, tXinyiIndustry, normConfig);
  905. }
  906. }
  907. }
  908. }
  909. }
  910. /**
  911. * 通过输入的值 生成对应类型的报警对象(进水)
  912. *
  913. * @param jsBzz
  914. * @param currentVal
  915. * @param category
  916. * @param tXinyiIndustry
  917. * @param normConfig
  918. * @return
  919. */
  920. private void handleXinYiWarningRecordJS(BigDecimal jsBzz, BigDecimal currentVal, String category, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  921. BigDecimal multiply = jsBzz.multiply(new BigDecimal(MyConstants.SCALE_VALUE));
  922. TXinyiWarningRecord tXinyiWarningRecord = new TXinyiWarningRecord();
  923. /*String category = BusinessEnum.WarningCategoryEnum.CS_AD.getCode();*/
  924. tXinyiWarningRecord.setStatus(0);
  925. tXinyiWarningRecord.setType(0);
  926. tXinyiWarningRecord.setCategory(category);
  927. tXinyiWarningRecord.setTime(DateUtils.getNowDate());
  928. tXinyiWarningRecord.setWarningVal(currentVal);
  929. tXinyiWarningRecord.setDesignVal(jsBzz);
  930. // tXinyiWarningRecord.setControlVal(csGkz);
  931. tXinyiWarningRecord.setCreateBy(WARNING_DEFAULT_CREATE);
  932. tXinyiWarningRecord.setCreateTime(DateUtils.getNowDate());
  933. //2024年5月25日17:52:33 如果工业库获取不到数据,也触发报警,但是不调用决策接口
  934. if(Objects.isNull(currentVal)){
  935. tXinyiWarningRecord.setReason(category + EXCEPTION_WARNING);
  936. tXinyiWarningRecord.setLevel(WARNING_LEVEL_NO_DATE);
  937. }else if(currentVal.compareTo(multiply) > 0){//一级
  938. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  939. tXinyiWarningRecord.setLevel(WARNING_LEVEL_ONE);
  940. }else if(currentVal.compareTo(jsBzz) >= 0 && currentVal.compareTo(multiply) <= 0){//二级
  941. tXinyiWarningRecord.setReason(category + CHAOBIAO_WARNING);
  942. tXinyiWarningRecord.setLevel(WARNING_LEVEL_TWO);
  943. }else{
  944. tXinyiWarningRecord = null;//这种的无需处理
  945. }
  946. /*else if(!Objects.isNull(csGkz) && currentVal.compareTo(csGkz) > 0){
  947. tXinyiWarningRecord.setReason(category + CHAOGUANKONG_WARNING);
  948. tXinyiWarningRecord.setLevel(WARNING_LEVEL_THREE);
  949. }*/
  950. //当前状态正常 需要查询历史有无正在报警的数据,如果有,将报警状态改完2(系统自动关闭)
  951. List<TXinyiWarningRecord> tXinyiWarningRecords = this.xinyiWarningRecordMapper.selectTXinyiWarningRecordList(TXinyiWarningRecord.builder().delFlag(0).type(0).category(category).status(0).build());
  952. if(Objects.isNull(tXinyiWarningRecord)){//数据正常,无告警信息
  953. if(!CollectionUtils.isEmpty(tXinyiWarningRecords)){
  954. log.info( "{}:现在恢复正常,历史报警数据为{}", category,JSON.toJSONString(tXinyiWarningRecords));
  955. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {
  956. xinyiWarningRecord.setStatus(2);
  957. Date nowDate = DateUtils.getNowDate();
  958. xinyiWarningRecord.setOffTime(nowDate);
  959. xinyiWarningRecord.setUpdateTime(nowDate);
  960. xinyiWarningRecord.setUpdateBy(WARNING_DEFAULT_CREATE);
  961. this.xinyiWarningRecordMapper.updateTXinyiWarningRecord(xinyiWarningRecord);
  962. }
  963. }
  964. }else{//有告警信息
  965. if(CollectionUtils.isEmpty(tXinyiWarningRecords)){//之前没有告警记录
  966. //保存到数据库中
  967. this.xinyiWarningRecordMapper.insertTXinyiWarningRecord(tXinyiWarningRecord);
  968. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  969. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  970. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  971. }else {
  972. //继续调用决策
  973. this.handleDecision(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  974. }
  975. }else{
  976. log.info("{}:之前已经有过告警记录了,且还是继续报警,无需重复添加报警,但是决策仍然要调用", category);
  977. for (TXinyiWarningRecord xinyiWarningRecord : tXinyiWarningRecords) {//理论上只有一个的
  978. if(WARNING_LEVEL_NO_DATE.equals(tXinyiWarningRecord.getLevel())){
  979. //只保存一个普通的问答记录 不需要调用决策信息,但是实时数据还是要记录的
  980. this.addChatRecordByWarning(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  981. }else {
  982. //继续调用决策
  983. this.handleDecision(xinyiWarningRecord, tXinyiIndustry, normConfig);
  984. }
  985. }
  986. }
  987. }
  988. }
  989. private void addChatRecordByWarning(TXinyiWarningRecord tXinyiWarningRecord, TXinyiIndustry tXinyiIndustry, TXinyiNormConfig normConfig) {
  990. ChatReq chatReq = new ChatReq();
  991. //保存聊天记录
  992. //将问答更新到数据库中
  993. chatReq.setSessionId(IdUtils.simpleUUID());
  994. chatReq.setType(2);//0问答 1决策 2本地
  995. chatReq.setModule(3);
  996. /*String userId = SecurityUtils.getUserId().toString();
  997. String username = SecurityUtils.getUsername();*/
  998. chatReq.setUserId(WARNING_DEFAULT_CREATE);
  999. String showVal = this.buildShowValue(tXinyiWarningRecord, tXinyiIndustry, normConfig);
  1000. chatReq.setShowVal(showVal);//前端展示的数据和提问的数据不一致
  1001. chatReq.setQuestion(WARNING_DEFAULT_QUESTION);//本地问题
  1002. chatReq.setAnswer(tXinyiWarningRecord.getReason() + ",请检查设备是否正常运行");
  1003. chatReq.setWarningId(String.valueOf(tXinyiWarningRecord.getId()));
  1004. chatReq.setCounts(1);//问答次数
  1005. chatReq.setCreateBy(WARNING_DEFAULT_CREATE);
  1006. chatReq.setCreateTime(DateUtils.getNowDate());
  1007. this.xinyiChatRecordMapper.insertTXinyiChatRecord(chatReq);
  1008. }
  1009. /**
  1010. * 定时从sqlserver获取数据
  1011. */
  1012. public void sqlserverData(){
  1013. log.info("进入了定时同步SqlServer的任务");
  1014. //主库获取上次最新的同步日期
  1015. String lastTime = this.xinyiRobotMapper.selectLastTime();
  1016. log.info("上次同步的日期是{}", lastTime);
  1017. //从
  1018. DynamicDataSourceContextHolder.setDataSourceType(DataSourceType.SLAVE.name());
  1019. List<TXinyiRobot> tXinyiRobots = xinyiRobotMapper.selectTXinyiRobotListByTime(lastTime);
  1020. DynamicDataSourceContextHolder.clearDataSourceType();
  1021. // System.out.println(JSON.toJSONString(tXinyiRobots));
  1022. // System.out.println("-------------");
  1023. //主
  1024. if(!CollectionUtils.isEmpty(tXinyiRobots)){
  1025. for (TXinyiRobot tXinyiRobot : tXinyiRobots) {
  1026. String date = handleDate(tXinyiRobot.getVDate().replaceAll(" ", ""));//有空格
  1027. String time = handleDate(tXinyiRobot.getVTime().replaceAll(" ", ""));//有空格
  1028. tXinyiRobot.setVDate(date);
  1029. tXinyiRobot.setVTime(time);
  1030. tXinyiRobot.setVDateTime(date + " " + time);
  1031. //处理给前端展示的字段
  1032. tXinyiRobot.setTestDate(date);//日期
  1033. tXinyiRobot.setTestHour(date + " " + time.substring(0, 2));//小时
  1034. tXinyiRobot.setTestTime(date + " " + time.substring(0, 5));//分钟
  1035. tXinyiRobot.setCreatedTime(new Date());
  1036. this.xinyiRobotMapper.insertTXinyiRobot(tXinyiRobot);
  1037. }
  1038. }
  1039. }
  1040. /**
  1041. *
  1042. * 定时生成每日简报数据
  1043. *
  1044. */
  1045. public void generageShortReport(){
  1046. log.info("进入了定时生成每日简报数据");
  1047. List<TXinyiDaily> dailyTwoRecords = this.xinyiDailyMapper.selectNRecords(2);
  1048. //正常不会有这种问题 因为日报有很多条
  1049. if(CollectionUtils.isEmpty(dailyTwoRecords) || dailyTwoRecords.size() < 2){
  1050. log.error("进入了定时生成每日简报数据 获取最新的2条数据不足,终止");
  1051. return;
  1052. }
  1053. //暂时不考虑因为没有填写日报 导致生成重复数据的问题(后续需要的话再添加)
  1054. //处理数据 并 拼装
  1055. String queryData = buildShortReportQueryData(dailyTwoRecords);
  1056. log.info("定时生成简报,组装好的请求大模型的参数为:{}", queryData);
  1057. if(StringUtils.isBlank(queryData)){
  1058. log.error("无法拼装请求数据!!!!!!");
  1059. return;
  1060. }
  1061. String showVal = formateDateStr(dailyTwoRecords.get(0).getTestDate()) + JIAN_BAO_END;
  1062. //调用模型 并保存结果
  1063. this.askBigModel(queryData, showVal);
  1064. log.info("定时生成简报任务结束~~~~~~~~~~~~~~");
  1065. }
  1066. /**
  1067. *
  1068. * 2022/01/01 转成2022年01月01日 数据
  1069. * @param testDate
  1070. * @return
  1071. */
  1072. private String formateDateStr(String testDate) {
  1073. if(StringUtils.isBlank(testDate))
  1074. return "";
  1075. if(!testDate.contains("/"))
  1076. return testDate;
  1077. String[] split = testDate.split("/");
  1078. return split[0] + "年" + split[1] + "月" + split[2] + "日";
  1079. }
  1080. private void askBigModel(String question, String showVal) {
  1081. log.info("进入了后台接口调⽤⼤模型获取问答结果处理");
  1082. StringBuilder sb = new StringBuilder();
  1083. String sessionId = IdUtils.simpleUUID();
  1084. ChatReq chatReq = new ChatReq();
  1085. // String ipAddr = IpUtils.getIpAddr();//获取用户的ip地址 传给大模型
  1086. String ipAddr = "";//获取用户的ip地址 传给大模型 定时任务获取不到ip地址
  1087. int counts = 1;//默认是第一次
  1088. //这种问答 没有历史问答的概念 直接把问题扔进去就行 无需查询历史记录
  1089. List<String> historyDates = new ArrayList<>();
  1090. historyDates.add(question);
  1091. // 获取输出流
  1092. ManagedChannel channel = null;
  1093. try {
  1094. channel = ManagedChannelBuilder.forAddress("10.0.0.24", 17070)
  1095. .usePlaintext()
  1096. .build();
  1097. InferenceAPIsServiceGrpc.InferenceAPIsServiceBlockingStub stub = InferenceAPIsServiceGrpc.newBlockingStub(channel);
  1098. String dataJson = "{\"bot_id\":\"721\",\"exp_id\":\"721\",\"session_id\":\"" + sessionId + "\",\"use_rag\":\"true\",\"prompt\":\"你是⼀个资深⽔务领域专家,能回答各种⽔务相关问题\",\"history_dia\":" + JSON.toJSONString(historyDates) + ",\"generate_args\":{\"max_new_tokens\":2048,\"max_length\":4096,\"num_beams\":1,\"do_sample\":true,\"top_p\":0.7,\"temperature\":0.95},\"extra\":{ \"ip_address\": \"" + ipAddr + "\" },\"strengthen\":" + (true) + "}";
  1099. log.info("请求大模型的问答参数为{}", dataJson);
  1100. PredictionsRequest request = PredictionsRequest.newBuilder()
  1101. .setModelName("slibra_bot")
  1102. .putInput("method", ByteString.copyFrom("infer_stream", "utf-8"))//推理
  1103. .putInput("data", ByteString.copyFrom(dataJson, "utf-8"))
  1104. .buildPartial();
  1105. Iterator<PredictionResponse> predictions = stub.streamPredictions(request);
  1106. //将结果记录到问答表
  1107. while (predictions.hasNext()) {
  1108. String responseStr = predictions.next().getPrediction().toStringUtf8();
  1109. log.info("大模型问答返回的原始结果为{}", responseStr);
  1110. responseStr = JSON.parseObject(responseStr).getString("message");
  1111. if("complete".equals(responseStr)){
  1112. System.out.println("结尾语句并且是非JSON,无需处理");
  1113. //结束语句也流式输出,但是并不记录下来 2024年5月24日11:15:23 也不返回前端
  1114. /*outputStream.write(responseStr.getBytes());
  1115. outputStream.flush();*/
  1116. }else{
  1117. sb.append(responseStr);
  1118. }
  1119. }
  1120. //将问答更新到数据库中
  1121. chatReq.setSessionId(sessionId);
  1122. chatReq.setType(2);//0问答 1决策 2本地
  1123. chatReq.setModule(4);//0专家问答 1智能工单 2智能体助手 3告警 4简报
  1124. chatReq.setShowVal(showVal);
  1125. chatReq.setQuestion(question);
  1126. chatReq.setAnswer(sb.toString());
  1127. chatReq.setCounts(counts);//问答次数
  1128. chatReq.setUserId(WARNING_DEFAULT_CREATE);
  1129. chatReq.setCreateBy(WARNING_DEFAULT_CREATE);
  1130. chatReq.setCreateTime(DateUtils.getNowDate());
  1131. this.xinyiChatRecordMapper.insertTXinyiChatRecord(chatReq);
  1132. } catch (IOException e) {
  1133. throw new RuntimeException(e);
  1134. } finally {
  1135. // 关闭输出流
  1136. channel.shutdown();
  1137. }
  1138. }
  1139. private String buildShortReportQueryData(List<TXinyiDaily> dailyTwoRecords) {
  1140. //查询配置信息
  1141. List<TXinyiNormConfig> tXinyiNormConfigs = this.xinyiNormConfigMapper.selectTXinyiNormConfigList(null);
  1142. if(CollectionUtils.isEmpty(tXinyiNormConfigs))
  1143. return null;
  1144. TXinyiNormConfig normConfig = tXinyiNormConfigs.get(0);
  1145. //获取数据
  1146. TXinyiDaily yesterdayData = dailyTwoRecords.get(0);
  1147. TXinyiDaily beforeYesterdayData = dailyTwoRecords.get(1);
  1148. String originStr = JIAN_BAO_PROMPT;
  1149. String yesterdayStr = yesterdayData.getTestDate().substring(5).replace("/", "月") + "日";
  1150. String beforeYesterdayStr = beforeYesterdayData.getTestDate().substring(5).replace("/", "月") + "日";
  1151. originStr = originStr.replace("#{0}", yesterdayStr);
  1152. originStr = originStr.replace("#{1}", beforeYesterdayStr);
  1153. originStr = originStr.replace("#{2}", String.valueOf(yesterdayData.getJsCod()));
  1154. originStr = originStr.replace("#{3}", String.valueOf(yesterdayData.getJsTn()));
  1155. originStr = originStr.replace("#{4}", String.valueOf(yesterdayData.getJsTp()));
  1156. originStr = originStr.replace("#{5}", String.valueOf(yesterdayData.getJsNh3()));
  1157. originStr = originStr.replace("#{6}", String.valueOf(yesterdayData.getJsSs()));
  1158. originStr = originStr.replace("#{7}", String.valueOf(yesterdayData.getJSL()));
  1159. originStr = originStr.replace("#{8}", String.valueOf(yesterdayData.getCsCod()));
  1160. originStr = originStr.replace("#{9}", String.valueOf(yesterdayData.getCsTn()));
  1161. originStr = originStr.replace("#{10}", String.valueOf(yesterdayData.getCsTp()));
  1162. originStr = originStr.replace("#{11}", String.valueOf(yesterdayData.getCsNh3()));
  1163. originStr = originStr.replace("#{12}", String.valueOf(yesterdayData.getCsSs()));
  1164. originStr = originStr.replace("#{13}", String.valueOf(yesterdayData.getCSL()));
  1165. originStr = originStr.replace("#{14}", String.valueOf(beforeYesterdayData.getJsCod()));
  1166. originStr = originStr.replace("#{15}", String.valueOf(beforeYesterdayData.getJsTn()));
  1167. originStr = originStr.replace("#{16}", String.valueOf(beforeYesterdayData.getJsTp()));
  1168. originStr = originStr.replace("#{17}", String.valueOf(beforeYesterdayData.getJsNh3()));
  1169. originStr = originStr.replace("#{18}", String.valueOf(beforeYesterdayData.getJsSs()));
  1170. originStr = originStr.replace("#{19}", String.valueOf(beforeYesterdayData.getJSL()));
  1171. originStr = originStr.replace("#{20}", String.valueOf(beforeYesterdayData.getCsCod()));
  1172. originStr = originStr.replace("#{21}", String.valueOf(beforeYesterdayData.getCsTn()));
  1173. originStr = originStr.replace("#{22}", String.valueOf(beforeYesterdayData.getCsTp()));
  1174. originStr = originStr.replace("#{23}", String.valueOf(beforeYesterdayData.getCsNh3()));
  1175. originStr = originStr.replace("#{24}", String.valueOf(beforeYesterdayData.getCsSs()));
  1176. originStr = originStr.replace("#{25}", String.valueOf(beforeYesterdayData.getCSL()));
  1177. originStr = originStr.replace("#{26}", String.valueOf(normConfig.getJscodSjz()));
  1178. originStr = originStr.replace("#{27}", String.valueOf(normConfig.getJszdSjz()));
  1179. originStr = originStr.replace("#{28}", String.valueOf(normConfig.getJszlSjz()));
  1180. originStr = originStr.replace("#{29}", String.valueOf(normConfig.getJsadSjz()));
  1181. originStr = originStr.replace("#{30}", String.valueOf(normConfig.getJsssSjz()));
  1182. originStr = originStr.replace("#{31}", String.valueOf(normConfig.getCscodBzz()));
  1183. originStr = originStr.replace("#{32}", String.valueOf(normConfig.getCszzBzz()));
  1184. originStr = originStr.replace("#{33}", String.valueOf(normConfig.getCszlBzz()));
  1185. originStr = originStr.replace("#{34}", String.valueOf(normConfig.getCsadBzz()));
  1186. originStr = originStr.replace("#{35}", String.valueOf(normConfig.getCsssBzz()));
  1187. return originStr;
  1188. }
  1189. public static String handleDate(String str){
  1190. StringBuilder sb = new StringBuilder();
  1191. if(str.contains(" ")){//包含空格 就是年月日时分秒了
  1192. String[] split = str.split(" ");
  1193. addBeforeZero(sb, split[0], "/");
  1194. sb.append(" ");
  1195. addBeforeZero(sb, split[1], ":");
  1196. }else{
  1197. if(str.contains("/")){//年月日
  1198. addBeforeZero(sb, str, "/");
  1199. }else if(str.contains(":")){//时分秒
  1200. addBeforeZero(sb, str, ":");
  1201. }else {
  1202. sb.append(str);
  1203. }
  1204. }
  1205. return sb.toString();
  1206. }
  1207. public static StringBuilder addBeforeZero(StringBuilder sb, String str, String tag){
  1208. String[] split = str.split(tag);
  1209. int length = split.length;
  1210. for (int i = 0; i < length; i++) {
  1211. String value = split[i];
  1212. Integer intValue = Integer.parseInt(value);
  1213. if(intValue < 10 && value.length() == 1){////防止有正确的情况 额外再补充字符串
  1214. sb.append(0).append(value);
  1215. }else{
  1216. sb.append(value);
  1217. }
  1218. if(i < length-1){
  1219. sb.append(tag);
  1220. }
  1221. }
  1222. return sb;
  1223. }
  1224. public static void main(String[] args) {
  1225. /*LocalDateTime endTime = LocalDateTime.now();
  1226. System.out.println("endTime = " + endTime);
  1227. endTime = endTime.plusMinutes(60);
  1228. System.out.println("endTime = " + endTime);*/
  1229. /*String str = "2024/04/15 09:55";
  1230. System.out.println(str);
  1231. System.out.println(str.substring(0,10));
  1232. System.out.println(str.substring(0,13));*/
  1233. /*String str = "2024/04/18 08:00";
  1234. str = str + ":00";
  1235. System.out.println(str);
  1236. LocalDateTime startTime = LocalDateTime.parse(str.replaceAll("/", "-").replace(" ", "T"));
  1237. System.out.println(startTime.plusMinutes(1L).format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));*/
  1238. /*ArrayList<Integer> objects = new ArrayList<>();
  1239. objects.add(1);
  1240. objects.add(2);
  1241. objects.add(3);
  1242. System.out.println(objects);
  1243. objects.clear();
  1244. System.out.println(objects);
  1245. // test();
  1246. Date date = new Date();
  1247. System.out.println(DateUtils.differentHoursByMillisecond(date, date));
  1248. HashMap<Object, Object> map = new HashMap<>();
  1249. map.put("a", null);
  1250. map.put("b", "ab");
  1251. map.put("c", "");
  1252. map.put("d", '1');
  1253. System.out.println(JSON.toJSONString(map, JSONWriter.Feature.WriteNulls));
  1254. TXinyiIndustry tXinyiIndustry = new TXinyiIndustry();
  1255. System.out.println(JSON.toJSONString(tXinyiIndustry, JSONWriter.Feature.WriteNulls));*/
  1256. String s = "2022/01/01".substring(5).replace("/", "月") + "日";
  1257. System.out.println("s = " + s);
  1258. System.out.println(new BigDecimal("1").compareTo(null));//空指针 要判断
  1259. }
  1260. //测试工业库 没小时保存一条记录是否可行
  1261. public static void test() {
  1262. // 每个小时的时间格式
  1263. DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
  1264. LocalDateTime startTime = LocalDateTime.parse("2024-05-18T00:00:00");
  1265. LocalDateTime endTime = LocalDateTime.now();
  1266. // 循环按小时分割
  1267. LocalDateTime currentHour = startTime;
  1268. //最终获取的数据
  1269. Map<String, TXinyiIndustry> needMap = new LinkedHashMap<>();
  1270. while (currentHour.isBefore(endTime)) {
  1271. String begin = currentHour.format(formatter);
  1272. String end = currentHour.plusMinutes(5).format(formatter);
  1273. // 输出当前小时的起始时间和结束时间
  1274. System.out.println("起始时间:" + begin);
  1275. System.out.println("结束时间:" + end);
  1276. // 当前小时加一小时,作为下一个小时的起始时间
  1277. currentHour = currentHour.plusMinutes(5);
  1278. //每个小时查询一次数据
  1279. String url = "http://10.0.0.27:4568/api/v1/khquerydata";
  1280. HashMap<String, Object> req = new HashMap<>();
  1281. req.put("tagNames", queryTags);
  1282. req.put("startTime", begin);
  1283. req.put("endTime", end);
  1284. req.put("recordNumbers", 100000);
  1285. String body = HttpRequest.post(url).header("Authorization", "c2E6c2E=").header("clientName", "hongshan").body(JSON.toJSONString(req)).execute().body();
  1286. // System.out.println("body = " + body);
  1287. List<HashMap<String, String>> list = new ArrayList<>();
  1288. //行转列数据处理
  1289. for (String queryTag : queryTags) {
  1290. JSONArray array = JSON.parseObject(body).getJSONArray(queryTag);
  1291. //特殊数据处理一
  1292. if(Objects.isNull(array) || array.isEmpty()){
  1293. System.out.println(queryTag + "查询到了空的数据,跳过本次循环");
  1294. continue;
  1295. }
  1296. int size = array.size();
  1297. //特殊数据处理二
  1298. if("0".equals(array.get(1) + "")){
  1299. System.out.println(queryTag + "查询到了数据,但是数据集合只有一条,且都是0");
  1300. continue;
  1301. }
  1302. //结合至少62个数据才满足条件(有可能获取不到)
  1303. /*if(size < 62){
  1304. System.out.println(queryTag + "查询到了不符合条件的数据,跳过本次循环");
  1305. continue;
  1306. }*/
  1307. //存放的数据集
  1308. //利用map去重
  1309. HashMap<String, String> map = new LinkedHashMap<>();
  1310. for (int i = 2; i < size; i++) {
  1311. // System.out.println(i + "" + array.get(i));
  1312. JSONArray oneRecord = JSON.parseArray(JSON.toJSONString(array.get(i)));
  1313. //处理为空或者为0的数据
  1314. Object timeStampValue = oneRecord.get(2);
  1315. if(Objects.isNull(timeStampValue) || "0".equals(timeStampValue + ""))
  1316. continue;
  1317. BigDecimal value = Objects.isNull(oneRecord.get(0)) ? null : new BigDecimal(oneRecord.get(0) + "");
  1318. long timestamp = (long) timeStampValue;
  1319. String format = DateUtil.format(new Date(timestamp), DateUtils.YYYYMMDDHH_TS);
  1320. map.put(format, queryTag + "-" + value);
  1321. }
  1322. list.add(map);
  1323. }
  1324. Set<String> recordTimeSet = new HashSet<>();
  1325. Map<String, TXinyiIndustry> recordMap = new HashMap<>();
  1326. for (int i = 0; i < list.size(); i++) {
  1327. HashMap<String, String> map = list.get(i);
  1328. int finalJ = i;
  1329. map.forEach((k, v) ->{
  1330. TXinyiIndustry industry = null;
  1331. if(!recordTimeSet.contains(k)){//第一次
  1332. industry = new TXinyiIndustry();
  1333. recordTimeSet.add(k);
  1334. recordMap.put(k, industry);
  1335. }else{
  1336. industry = recordMap.get(k);
  1337. }
  1338. industry.setTestTime(k + ":00");
  1339. //2024年4月15日11:19:52 额外增加2个字段
  1340. industry.setTestDate(k.substring(0,10));
  1341. industry.setTestHour(k.substring(0,13));
  1342. //解析值
  1343. String[] split = v.split("-");
  1344. String type = split[0];
  1345. BigDecimal value = new BigDecimal(split[1]);
  1346. if ("信义污水厂JS_COD_Value".equals(type)) {
  1347. industry.setJsCod(value);
  1348. } else if ("信义污水厂JS_PH_Value".equals(type)) {
  1349. industry.setJsPh(value);
  1350. } else if ("信义污水厂JS_SS_Value".equals(type)) {
  1351. industry.setJsSs(value);
  1352. } else if ("信义污水厂JS_ZL_Value".equals(type)) {
  1353. industry.setJsTp(value);
  1354. } else if ("信义污水厂JS_ZA_Value".equals(type)) {
  1355. industry.setJsTn(value);
  1356. } else if ("信义污水厂JS_AD_Value".equals(type)) {
  1357. industry.setJsNh3(value);
  1358. } else if ("信义污水厂JS_T_Value".equals(type)) {
  1359. industry.setJsSwPh(value);
  1360. } else if ("信义污水厂进水泵房液位".equals(type)) {
  1361. industry.setJsBfyw(value);
  1362. } else if ("信义污水厂出水瞬时流量".equals(type)) {
  1363. industry.setCsSlqc(value);
  1364. } else if ("信义污水厂升级出水COD".equals(type)) {
  1365. industry.setCsCod(value);
  1366. } else if ("信义污水厂升级出水PH".equals(type)) {
  1367. industry.setCsPh(value);
  1368. } else if ("信义污水厂升级出水SS".equals(type)) {
  1369. industry.setCsSs(value);
  1370. } else if ("信义污水厂升级出水TN".equals(type)) {
  1371. industry.setCsTn(value);
  1372. } else if ("信义污水厂升级出水TP".equals(type)) {
  1373. industry.setCsTp(value);
  1374. } else if ("信义污水厂升级出水氨氮".equals(type)) {
  1375. industry.setCsNh3(value);
  1376. } else if ("信义污水厂AIT202_Value".equals(type)) {
  1377. industry.setOneHyzdDo(value);
  1378. } else if ("信义污水厂AIT203_Value".equals(type)) {
  1379. industry.setOneHymdDo(value);
  1380. } else if ("信义污水厂AIT207_Value".equals(type)) {
  1381. industry.setTwoHyzdDo(value);
  1382. } else if ("信义污水厂AIT206_Value".equals(type)) {
  1383. industry.setTwoHymdDo(value);
  1384. } else if ("信义污水厂AIT209_Value".equals(type)) {
  1385. industry.setOneMlss(value);
  1386. } else if ("信义污水厂AIT210_Value".equals(type)) {
  1387. industry.setTwoMlss(value);
  1388. } else if ("信义污水厂进水TDS".equals(type)) {
  1389. industry.setJsTds(value);
  1390. } else if ("信义污水厂FT101_Value".equals(type)) {
  1391. industry.setJsSlq(value);
  1392. } else if ("信义污水厂SWCHHYHLB1_R_Value".equals(type)) {
  1393. industry.setNHlbOneGp(value);
  1394. } else if ("信义污水厂SWCHHYHLB2_R_Value".equals(type)) {
  1395. industry.setNHlbTwoGp(value);
  1396. } else if ("信义污水厂SWCHHYHLB3_R_Value".equals(type)) {
  1397. industry.setNHlbThreeGp(value);
  1398. } else if ("信义污水厂SWCHHYHLB4_R_Value".equals(type)) {
  1399. industry.setNHlbFourGp(value);
  1400. } else if ("信义污水厂SWCHHYHLB5_R_Value".equals(type)) {
  1401. industry.setNhlBFiveGp(value);
  1402. } else if ("信义污水厂SWCHHYHLB6_R_Value".equals(type)) {
  1403. industry.setNHlbSixGp(value);
  1404. } else if ("信义污水厂SWCWNHLB1_R_Value".equals(type)) {
  1405. industry.setWHlbOneGp(value);
  1406. } else if ("信义污水厂SWCWNHLB2_R_Value".equals(type)) {
  1407. industry.setWHlbTwoGp(value);
  1408. } else if ("信义污水厂SWCWNHLB3_R_Value".equals(type)) {
  1409. industry.setWHlbThreeGp(value);
  1410. } else if ("信义污水厂SWCWNHLB4_R_Value".equals(type)) {
  1411. industry.setWHlbFourGp(value);
  1412. } else if ("信义污水厂SWCWNHLB5_R_Value".equals(type)) {
  1413. industry.setWHlbFiveGp(value);
  1414. } else if ("信义污水厂GFJ1_R_Value".equals(type)) {
  1415. industry.setFjOne(value);
  1416. } else if ("信义污水厂GFJ2_R_Value".equals(type)) {
  1417. industry.setFjTwo(value);
  1418. } else if ("信义污水厂GFJ3_R_Value".equals(type)) {
  1419. industry.setFjThree(value);
  1420. } else if ("信义污水厂GFJ4_R_Value".equals(type)) {
  1421. industry.setFjFour(value);
  1422. } else if ("信义污水厂GFJ5_R_Value".equals(type)) {
  1423. industry.setFjFive(value);
  1424. } else if ("信义污水厂GFJ6_R_Value".equals(type)) {
  1425. industry.setFjSix(value);
  1426. } else if ("信义污水厂GFJ1_KQLL_Value".equals(type)) {
  1427. industry.setKqllOne(value);
  1428. } else if ("信义污水厂GFJ2_KQLL_Value".equals(type)) {
  1429. industry.setKqllTwo(value);
  1430. } else if ("信义污水厂GFJ3_KQLL_Value".equals(type)) {
  1431. industry.setKqllThree(value);
  1432. } else if ("信义污水厂GFJ4_KQLL_Value".equals(type)) {
  1433. industry.setKqllFour(value);
  1434. } else if ("信义污水厂GFJ5_KQLL_Value".equals(type)) {
  1435. industry.setKqllFive(value);
  1436. } else if ("信义污水厂GFJ6_KQLL_Value".equals(type)) {
  1437. industry.setKqllSix(value);
  1438. }else if ("信义污水厂实际碳源加药量".equals(type)) {
  1439. industry.setSJTYJLY(value);
  1440. }else if ("信义污水厂除磷加药瞬时流量".equals(type)) {
  1441. industry.setCLJYSSLL(value);
  1442. } else if ("信义污水厂_除磷P04预测值_".equals(type)) {
  1443. industry.setCLP04YCZ(value);
  1444. }
  1445. //只有最后一次才执行数据库添加
  1446. if(finalJ == list.size()-1){
  1447. needMap.put(industry.getTestHour(), industry);
  1448. }
  1449. });
  1450. }
  1451. }
  1452. //保存数据 触发告警 决策 问答记录等等
  1453. needMap.forEach((k, industry) ->{
  1454. System.out.println(JSON.toJSONString(industry));
  1455. });
  1456. }
  1457. }