DataStaticService.java 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303
  1. package ieven.server.webapp.service.datastatic;
  2. import com.google.common.collect.Lists;
  3. import ieven.server.webapp.domain.data.DataMap;
  4. import ieven.server.webapp.domain.data.StaticInput;
  5. import ieven.server.webapp.domain.data.StaticModelInput;
  6. import ieven.server.webapp.infrastructure.wrapper.Mapped;
  7. import ieven.server.webapp.service.FieldsService;
  8. import ieven.server.webapp.service.Ops;
  9. import ieven.server.webapp.util.SplitCharacterEnum;
  10. import ieven.server.webapp.util.excel.PublicStatic;
  11. import lombok.extern.slf4j.Slf4j;
  12. import org.apache.commons.collections4.CollectionUtils;
  13. import org.apache.commons.lang3.StringUtils;
  14. import org.springframework.beans.factory.annotation.Autowired;
  15. import org.springframework.data.mongodb.core.MongoTemplate;
  16. import org.springframework.data.mongodb.core.query.Criteria;
  17. import org.springframework.data.mongodb.core.query.Query;
  18. import org.springframework.data.mongodb.core.query.Update;
  19. import org.springframework.stereotype.Service;
  20. import java.math.BigDecimal;
  21. import java.util.*;
  22. import java.util.regex.Pattern;
  23. @Slf4j
  24. @Service
  25. public class DataStaticService extends Ops {
  26. @Autowired
  27. FieldsService fieldsService;
  28. @Autowired
  29. private MongoTemplate mongoTemplate;
  30. public Mapped update(StaticInput staticInput) {
  31. //获取需要修改的数据
  32. DataMap dataMap = this.mongoTemplate.findById(staticInput.getLineId(), DataMap.class);
  33. if (dataMap == null) {
  34. return Mapped.ERROR("当前数据不存在");
  35. }
  36. if (StringUtils.isBlank(String.valueOf(dataMap.get("交易主体账号"))) || "null".equals(String.valueOf(dataMap.get("交易主体账号")))) {
  37. return Mapped.ERROR("交易主体账号有误");
  38. }
  39. //修改汇总表中的数据
  40. Query query = new Query();
  41. query.addCriteria(Criteria.where("_id").is(staticInput.getLineId()));
  42. Update update = new Update();
  43. if (StringUtils.isNotBlank(staticInput.getJyzthm())) {
  44. update.set("交易主体户名", PublicStatic.specialFormatStr(staticInput.getJyzthm()));
  45. }
  46. if (StringUtils.isNotBlank(staticInput.getZjhm())) {
  47. update.set("证件号码", PublicStatic.specialFormatStr(staticInput.getZjhm()));
  48. }
  49. if (StringUtils.isNotBlank(staticInput.getJyzthm()) || StringUtils.isNotBlank(staticInput.getZjhm())) {
  50. mongoTemplate.updateFirst(query, update, DataMap.class);
  51. }
  52. //修改明细表的交易主体户名
  53. if (StringUtils.isNotBlank(staticInput.getJyzthm())) {
  54. String modelId = staticInput.getModelId();
  55. List<String> fileIds = new ArrayList<>();
  56. String jyztzh = String.valueOf(dataMap.get("交易主体账号"));
  57. String dateFileId = getFileId("数据分析-主体信息汇总明细", modelId, mongoTemplate);
  58. if (StringUtils.isNotBlank(dateFileId)) {
  59. fileIds.add(dateFileId);
  60. }
  61. //五联单
  62. String dateFileIdwld = getFileId("五联单-账户明细", modelId, mongoTemplate);
  63. if (StringUtils.isNotBlank(dateFileIdwld)) {
  64. fileIds.add(dateFileIdwld);
  65. }
  66. //财付通
  67. String dateFileIdcft = getFileId("财付通-交易明细", modelId, mongoTemplate);
  68. if (StringUtils.isNotBlank(dateFileIdcft)) {
  69. fileIds.add(dateFileIdcft);
  70. }
  71. //反诈
  72. String dateFileIdfzza = getFileId("反诈治安-交易明细", modelId, mongoTemplate);
  73. if (StringUtils.isNotBlank(dateFileIdfzza)) {
  74. fileIds.add(dateFileIdfzza);
  75. }
  76. if (CollectionUtils.isNotEmpty(fileIds)) {
  77. Query queryjyzthm = new Query();
  78. queryjyzthm.addCriteria(Criteria.where("fileId").in(fileIds).and("交易主体账号").is(jyztzh));
  79. Update updatejy = new Update();
  80. updatejy.set("交易主体户名", PublicStatic.specialFormatStr(staticInput.getJyzthm()));
  81. mongoTemplate.updateMulti(queryjyzthm, updatejy, DataMap.class);
  82. }
  83. }
  84. return Mapped.OK();
  85. }
  86. public Mapped staticModel(StaticModelInput staticModelInput) {
  87. String modelId = staticModelInput.getModelId();
  88. String dateFileId = getFileId("数据分析-主体信息汇总明细", modelId, mongoTemplate);
  89. String dateFileIdUser = getFileId("数据分析-所有主体信息汇总", modelId, mongoTemplate);
  90. if (StringUtils.isBlank(dateFileId)) {
  91. return Mapped.ERROR("数据分析-主体信息汇总明细不存在");
  92. }
  93. if (StringUtils.isBlank(dateFileIdUser)) {
  94. return Mapped.ERROR("数据分析-所有主体信息汇总");
  95. }
  96. //查询所有的主体信息汇总
  97. List<DataMap> dataMaps = this.mongoTemplate.find((new Query(
  98. Criteria.where("fileId")
  99. .is(dateFileIdUser))), DataMap.class);
  100. Map<String, List<DataMap>> personMap = new HashMap<>();
  101. Map<String, List<DataMap>> personzhmcMap = new HashMap<>();
  102. for (DataMap dataMap : dataMaps) {
  103. String zh = String.valueOf(dataMap.get("账号"));
  104. String zhmc = String.valueOf(dataMap.get("账户名称"));
  105. List<DataMap> dataMaps1 = personMap.get(zh);
  106. List<DataMap> dataMapshmc = personzhmcMap.get(zhmc);
  107. if (CollectionUtils.isEmpty(dataMaps1)) {
  108. personMap.put(zh, Lists.newArrayList(dataMap));
  109. } else {
  110. dataMaps1.add(dataMap);
  111. personMap.put(zh, dataMaps1);
  112. }
  113. if (CollectionUtils.isEmpty(dataMapshmc)) {
  114. personzhmcMap.put(zhmc, Lists.newArrayList(dataMap));
  115. } else {
  116. dataMapshmc.add(dataMap);
  117. personzhmcMap.put(zhmc, dataMapshmc);
  118. }
  119. }
  120. //分两种情况,1:主体是否有数据如果有就按交易主体的计算
  121. Query query = new Query();
  122. //Pattern compile = Pattern.compile("^((?!" + "张"+ ").)*$",Pattern.CASE_INSENSITIVE);
  123. query.addCriteria(Criteria.where("fileId").is(dateFileId));
  124. query.addCriteria(new Criteria().andOperator(getQuery()));
  125. if (StringUtils.isNotBlank(staticModelInput.getJyztzh())) {
  126. query.addCriteria(Criteria.where("交易主体账号").in(Objects.requireNonNull(SplitCharacterEnum.splitByCharacter(staticModelInput.getJyztzh()))));
  127. }
  128. if (StringUtils.isNotBlank(staticModelInput.getJyzthm())) {
  129. query.addCriteria(Criteria.where("交易主体户名").in(Objects.requireNonNull(SplitCharacterEnum.splitByCharacter(staticModelInput.getJyzthm()))));
  130. }
  131. if (StringUtils.isNotBlank(staticModelInput.getJyztzhds())) {
  132. query.addCriteria(Criteria.where("交易对手账号").in(Objects.requireNonNull(SplitCharacterEnum.splitByCharacter(staticModelInput.getJyztzhds()))));
  133. }
  134. if (StringUtils.isNotBlank(staticModelInput.getJyzthmds())) {
  135. query.addCriteria(Criteria.where("交易对手户名").in(Objects.requireNonNull(SplitCharacterEnum.splitByCharacter(staticModelInput.getJyzthmds()))));
  136. }
  137. if (StringUtils.isNotBlank(staticModelInput.getJysjStart())&&StringUtils.isBlank(staticModelInput.getJysjEnd())) {
  138. query.addCriteria(Criteria.where("交易时间").gte(staticModelInput.getJysjStart()));
  139. }
  140. if (StringUtils.isNotBlank(staticModelInput.getJysjEnd())&&StringUtils.isBlank(staticModelInput.getJysjStart())) {
  141. query.addCriteria(Criteria.where("交易时间").lte(staticModelInput.getJysjEnd()));
  142. }
  143. if (StringUtils.isNotBlank(staticModelInput.getJysjEnd())&&StringUtils.isNotBlank(staticModelInput.getJysjStart())) {
  144. query.addCriteria(Criteria.where("交易时间").gte(staticModelInput.getJysjStart()).lte(staticModelInput.getJysjEnd()));
  145. }
  146. Map<String, Object> dataMap = new HashMap<>();
  147. List<Map<String, Object>> partyDataList = new ArrayList<>();
  148. List<Map<String, Object>> relationshipGraphList = new ArrayList<>();
  149. List<DataMap> dataMapDetail = this.mongoTemplate.find(query, DataMap.class);
  150. List<String> userList = new ArrayList<>();
  151. List<String> usermcList = new ArrayList<>();
  152. List<String> userAllList = new ArrayList<>();
  153. for (DataMap map : dataMapDetail) {
  154. String jythm = String.valueOf(map.get("交易主体户名"));
  155. String jydszh = String.valueOf(map.get("交易对手账号"));
  156. if (StringUtils.isBlank(jythm) || "null".equals(jythm)) {
  157. continue;
  158. }
  159. String jdlx = String.valueOf(map.get("借贷类型"));
  160. if (!jdlx.equals("借") && !jdlx.equals("贷")) {
  161. continue;
  162. }
  163. //获取节点名称
  164. if (!userList.contains(jythm)) {
  165. Map<String, Object> userMap = new HashMap<>();
  166. List<DataMap> dataMaps1 = personzhmcMap.get(jythm);
  167. if (CollectionUtils.isNotEmpty(dataMaps1)) {
  168. userMap.put("id", String.valueOf(dataMaps1.get(0).get("_id")));
  169. userMap.put("name", dataMaps1.get(0).get("账户名称"));
  170. userMap.put("relationPersons", dataMaps1);
  171. userMap.put("type", "主");
  172. partyDataList.add(userMap);
  173. userList.add(jythm);
  174. }
  175. }
  176. if (!usermcList.contains(jydszh)) {
  177. Map<String, Object> userMap = new HashMap<>();
  178. List<DataMap> dataMaps1 = personMap.get(jydszh);
  179. if (CollectionUtils.isNotEmpty(dataMaps1)) {
  180. String cid = String.valueOf(dataMaps1.get(0).get("_id"));
  181. userMap.put("id", cid);
  182. userMap.put("name", dataMaps1.get(0).get("账号"));
  183. userMap.put("relationPersons", dataMaps1);
  184. userMap.put("type", "从");
  185. partyDataList.add(userMap);
  186. usermcList.add(jydszh);
  187. }
  188. }
  189. //增加线上数据
  190. Map<String, Object> total = new HashMap<>();
  191. Map<String, Object> party1 = new HashMap<>();
  192. Map<String, Object> party2 = new HashMap<>();
  193. Map<String, Object> relationData = new HashMap<>();
  194. String userAllId = "";
  195. if (jdlx.equals("借")) {
  196. List<DataMap> dataMaps1 = personzhmcMap.get(jythm);
  197. List<DataMap> dataMaps2 = personMap.get(jydszh);
  198. if(CollectionUtils.isNotEmpty(dataMaps1)&&CollectionUtils.isNotEmpty(dataMaps2)){
  199. String one = String.valueOf(dataMaps1.get(0).get("_id"));
  200. party1.put("id", one);
  201. party1.put("name", dataMaps1.get(0).get("账户名称"));
  202. String two = String.valueOf(dataMaps2.get(0).get("_id"));
  203. party2.put("id", two);
  204. party2.put("name", dataMaps2.get(0).get("账号"));
  205. userAllId = one + "-" + two;
  206. }
  207. } else if (jdlx.equals("贷")) {
  208. List<DataMap> dataMaps1 = personzhmcMap.get(jythm);
  209. List<DataMap> dataMaps2 = personMap.get(jydszh);
  210. if(CollectionUtils.isNotEmpty(dataMaps1)&&CollectionUtils.isNotEmpty(dataMaps2)){
  211. String two = String.valueOf(dataMaps1.get(0).get("_id"));
  212. party2.put("id", dataMaps1.get(0).get("_id"));
  213. party2.put("name", dataMaps1.get(0).get("账户名称"));
  214. String one = String.valueOf(dataMaps2.get(0).get("_id"));
  215. party1.put("id", dataMaps2.get(0).get("_id"));
  216. party1.put("name", dataMaps2.get(0).get("账号"));
  217. userAllId = one + "-" + two;
  218. }
  219. }
  220. if (!userAllList.contains(userAllId)) {
  221. //查询所有的数据
  222. //查询所有的主体信息汇总
  223. BigDecimal sum = BigDecimal.ZERO;
  224. List<DataMap> dataDetailList = this.mongoTemplate.find((new Query(
  225. Criteria.where("fileId")
  226. .is(dateFileId)
  227. .and("交易主体户名")
  228. .is(jythm)
  229. .and("交易对手账号")
  230. .is(jydszh))), DataMap.class);
  231. for (DataMap dataMap1 : dataDetailList) {
  232. String jyje = String.valueOf(dataMap1.get("交易金额"));
  233. sum.add(new BigDecimal(jyje));
  234. }
  235. relationData.put("jeTotal", sum);
  236. relationData.put("sum", dataDetailList.size());
  237. total.put("party1", party1);
  238. total.put("party2", party2);
  239. total.put("relationData", relationData);
  240. total.put("relationships", dataDetailList);
  241. relationshipGraphList.add(total);
  242. userAllList.add(userAllId);
  243. }
  244. }
  245. dataMap.put("partyData", partyDataList);
  246. dataMap.put("relationshipGraph", relationshipGraphList);
  247. Mapped mapped = Mapped.OK();
  248. if(partyDataList.size()>1000){
  249. return Mapped.ERROR("节点超过300个,请重新查询");
  250. }
  251. mapped.put("result", dataMap);
  252. return mapped;
  253. }
  254. public List<Criteria> getQuery() {
  255. List<Criteria> dataList = new ArrayList<>();
  256. List<String> filedsList = Lists.newArrayList("交易对手账号", "交易对手户名");
  257. List<String> noQueryData = Lists.newArrayList(
  258. "公司", "慈善", "红十字", "店", "工作室", "商行", "中心", "经营部", "水果行", "餐厅", "超市", "市场", "商贸", "西饼屋", "科技", "酒楼", "停车场", "ETC", "etc", "服务", "批发",
  259. "渔场", "浴场", "设计", "物业", "咨询", "批发", "合作社", "工厂", "鞋厂", "动物园", "植物园", "土黑猪", "猫狗", "学校", "学院", "大学", "棋牌", "娱乐", "袜厂", "制品厂", "加油站",
  260. "体育馆", "研究所", "保健院", "龙虾", "羊蝎子", "足道", "公安局", "税务局", "法院", "菜馆", "棋牌", "水产行", "海鲜行", "公司", "服饰", "服装", "基金", "宠物", "车行", "保险", "理财",
  261. "支付宝", "财付通", "apple", "烟酒", "经营者", "商户"
  262. );
  263. for (String s : filedsList) {
  264. for (String noQueryDatum : noQueryData) {
  265. Criteria newCri = new Criteria(s);
  266. Pattern compile = Pattern.compile("^((?!" + noQueryDatum + ").)*$", Pattern.CASE_INSENSITIVE);
  267. newCri.regex(compile);
  268. dataList.add(newCri);
  269. }
  270. }
  271. return dataList;
  272. }
  273. }