RelationEqInShController.java 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
  1. package com.persagy.proxy.relation.controller;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.util.StrUtil;
  4. import com.alibaba.fastjson.JSONArray;
  5. import com.alibaba.fastjson.JSONObject;
  6. import com.fasterxml.jackson.databind.node.JsonNodeFactory;
  7. import com.fasterxml.jackson.databind.node.ObjectNode;
  8. import com.persagy.dmp.basic.model.QueryCriteria;
  9. import com.persagy.proxy.adm.request.AdmResponse;
  10. import com.persagy.proxy.adm.service.IAdmRelationService;
  11. import com.persagy.proxy.adm.utils.AdmContextUtil;
  12. import com.persagy.proxy.common.entity.InstanceUrlParam;
  13. import com.persagy.proxy.common.entity.RelationDTO;
  14. import org.springframework.beans.factory.annotation.Autowired;
  15. import org.springframework.beans.factory.annotation.Value;
  16. import org.springframework.web.bind.annotation.PostMapping;
  17. import org.springframework.web.bind.annotation.RequestBody;
  18. import org.springframework.web.bind.annotation.RequestMapping;
  19. import org.springframework.web.bind.annotation.RestController;
  20. import java.util.ArrayList;
  21. import java.util.List;
  22. /**
  23. * @description:020、关系-设备所在竖井
  24. * @author:lgy
  25. * @data:2021/9/13 10:00
  26. */
  27. @RequestMapping("/rel/eq-in-sh")
  28. @RestController
  29. public class RelationEqInShController {
  30. @Autowired
  31. private IAdmRelationService service;
  32. /**
  33. * 创建设备所在竖井关系,累计添加不会覆盖以前记录
  34. * @param param
  35. * @return
  36. * @throws Exception
  37. */
  38. @PostMapping(value = {"/link"})
  39. public AdmResponse create(@RequestBody JSONObject param) throws Exception {
  40. String equipId = param.getString("equipId");
  41. String shaftId = param.getString("shaftId");
  42. if(!StrUtil.isAllNotEmpty(equipId,shaftId)) {
  43. return AdmResponse.failure("必填项:EquipId(设备id)、ShaftId(竖井id)");
  44. }
  45. // 创建关系对象 EquipinShaft
  46. List<RelationDTO> voList = new ArrayList<>();
  47. voList.add(new RelationDTO(null, "MechInArch", "Eq2Sh", null, equipId , shaftId));
  48. service.doSave(AdmContextUtil.toDmpContext(), voList);
  49. return AdmResponse.success();
  50. }
  51. /**
  52. * 根据对象删除设备所在竖井关系,只针对一个对象
  53. * @param param 对象
  54. * @return 删除的结果信息
  55. */
  56. @PostMapping("/unlink")
  57. public AdmResponse delete(@RequestBody JSONObject param) throws Exception {
  58. if(CollUtil.isEmpty(param)) {
  59. return AdmResponse.success();
  60. }
  61. // 组装上下文条件
  62. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  63. // 组装条件
  64. QueryCriteria criteria = new QueryCriteria();
  65. String equipId = param.getString("equipId");
  66. String shaftId = param.getString("shaftId");
  67. if(!StrUtil.isAllNotEmpty(equipId,shaftId)) {
  68. return AdmResponse.failure("必填项:EquipId(设备id)、shaftId(竖井id)");
  69. }
  70. ObjectNode node = JsonNodeFactory.instance.objectNode();
  71. node.put("graphCode", "MechInArch");
  72. node.put("relCode", "Eq2Sh");
  73. node.put("objFrom", equipId);
  74. node.put("objTo",shaftId);
  75. criteria.setCriteria(node);
  76. service.doDelete(context, criteria);
  77. return AdmResponse.success();
  78. }
  79. /**
  80. * 设备所在竖井,设备一对多,此方法会覆盖以前的记录
  81. * @param param
  82. * @return
  83. * @throws Exception
  84. */
  85. @PostMapping("/link-sh")
  86. public AdmResponse createSh(@RequestBody JSONObject param) throws Exception {
  87. String equipId = param.getString("equipId");
  88. JSONArray shaftIdList = param.getJSONArray("shaftIdList");
  89. if(StrUtil.isBlank(equipId) || CollUtil.isEmpty(shaftIdList)) {
  90. return AdmResponse.failure("必填项:EquipId(设备id)、ShaftIdList(竖井id列表),此方法会覆盖以前的记录");
  91. }
  92. // 创建关系对象.先删除,后添加
  93. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  94. List<RelationDTO> voList = new ArrayList<>();
  95. for(int i = 0;i < shaftIdList.size();i++) {
  96. String shaftId = shaftIdList.getString(i);
  97. if(StrUtil.isBlank(shaftId)) {
  98. continue;
  99. }
  100. QueryCriteria queryRequest = new QueryCriteria();
  101. ObjectNode node = JsonNodeFactory.instance.objectNode();
  102. node.put("graphCode", "MechInArch");
  103. node.put("relCode", "Eq2Sh");
  104. node.put("objFrom", equipId);
  105. node.put("objTo", shaftId);
  106. queryRequest.setCriteria(node);
  107. service.doDelete(context,queryRequest);
  108. // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
  109. voList.add(new RelationDTO(null, "MechInArch", "Eq2Sh", null, equipId, shaftId));
  110. }
  111. // 组装上下文条件
  112. service.doSave(context, voList);
  113. return AdmResponse.success();
  114. }
  115. /**
  116. * 所在竖井的设备,竖井一对多,此方法会覆盖以前的记录
  117. * @param param
  118. * @return
  119. * @throws Exception
  120. */
  121. @PostMapping(value = {"/link-eq","/link-eq-sh"})
  122. public AdmResponse createEq(@RequestBody JSONObject param) throws Exception {
  123. String shaftId = param.getString("shaftId");
  124. JSONArray equipIdList = param.getJSONArray("equipIdList");
  125. if(StrUtil.isBlank(shaftId) || CollUtil.isEmpty(equipIdList)) {
  126. return AdmResponse.failure("必填项:EquipIdList(设备id列表)、ShaftId(竖井id),此方法会覆盖以前的记录");
  127. }
  128. // 创建关系对象.先删除,后添加
  129. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  130. List<RelationDTO> voList = new ArrayList<>();
  131. for(int i = 0;i < equipIdList.size();i++) {
  132. String equipId = equipIdList.getString(i);
  133. if(StrUtil.isBlank(shaftId)) {
  134. continue;
  135. }
  136. QueryCriteria queryRequest = new QueryCriteria();
  137. ObjectNode node = JsonNodeFactory.instance.objectNode();
  138. node.put("graphCode", "MechInArch");
  139. node.put("relCode", "Eq2Sh");
  140. node.put("objFrom", equipId);
  141. node.put("objTo", shaftId);
  142. queryRequest.setCriteria(node);
  143. service.doDelete(context,queryRequest);
  144. // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
  145. voList.add(new RelationDTO(null, "MechInArch", "Eq2Sh", null, equipId, shaftId));
  146. }
  147. // 组装上下文条件
  148. service.doSave(context, voList);
  149. return AdmResponse.success();
  150. }
  151. }