RelationEqForShController.java 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. package com.persagy.proxy.relation.controller;
  2. import cn.hutool.core.util.StrUtil;
  3. import com.fasterxml.jackson.databind.node.JsonNodeFactory;
  4. import com.fasterxml.jackson.databind.node.ObjectNode;
  5. import com.persagy.dmp.basic.model.QueryCriteria;
  6. import com.persagy.proxy.adm.constant.AdmRelationTypeEnum;
  7. import com.persagy.proxy.relation.model.EqForSh;
  8. import com.persagy.proxy.relation.model.EqForShList;
  9. import com.persagy.proxy.relation.model.ShForEqList;
  10. import com.persagy.proxy.adm.request.AdmResponse;
  11. import com.persagy.proxy.adm.service.IAdmRelationService;
  12. import com.persagy.proxy.adm.utils.AdmContextUtil;
  13. import com.persagy.proxy.common.entity.InstanceUrlParam;
  14. import com.persagy.proxy.common.entity.RelationDTO;
  15. import org.springframework.beans.factory.annotation.Autowired;
  16. import org.springframework.beans.factory.annotation.Value;
  17. import org.springframework.util.CollectionUtils;
  18. import org.springframework.web.bind.annotation.*;
  19. import java.util.ArrayList;
  20. import java.util.List;
  21. /**
  22. * @author zhangwencheng
  23. * 设备服务于竖井
  24. */
  25. @RestController
  26. @RequestMapping("/rel/eq-for-sh")
  27. public class RelationEqForShController {
  28. @Autowired
  29. private IAdmRelationService service;
  30. /**
  31. * 创建设备服务于竖井关系
  32. * 必填项:EquipId(设备id)、ShaftId(竖井id)
  33. *
  34. * @return 创建的结果信息
  35. */
  36. @PostMapping("/link")
  37. public AdmResponse create(@RequestBody EqForSh eqForSh) {
  38. String equipId = eqForSh.getEquipId();
  39. String shaftId = eqForSh.getShaftId();
  40. String type = eqForSh.getType();
  41. //检验输入参数
  42. if (StrUtil.isEmpty(equipId) || StrUtil.isEmpty(shaftId)) {
  43. return AdmResponse.failure("必填项:equipId(设备id)、shaftId(竖井Id)");
  44. }
  45. AdmRelationTypeEnum typeEnum = AdmRelationTypeEnum.EQ2SH_FOR;
  46. String graphCode = typeEnum.getGraphCode();
  47. String relCode = typeEnum.getRelCode();
  48. List<RelationDTO> voList = new ArrayList<>();
  49. RelationDTO relation = new RelationDTO(null, graphCode, relCode, type, equipId, shaftId);
  50. voList.add(relation);
  51. service.doSave(AdmContextUtil.toDmpContext(), voList);
  52. return AdmResponse.success();
  53. }
  54. /**
  55. * 创建设备服务于竖井关系
  56. * 必填项:ShaftIdList(竖井id列表)、EquipId(设备id),测方法会覆盖以前的记录
  57. *
  58. * @param
  59. * @return 创建的结果信息
  60. */
  61. @PostMapping("/link-sh")
  62. public AdmResponse createSh(@RequestBody EqForShList eqForShList) {
  63. String equipId = eqForShList.getEquipId();
  64. List<String> shaftIdList = eqForShList.getShaftIdList();
  65. //检验输入参数
  66. if (StrUtil.isEmpty(equipId) || CollectionUtils.isEmpty(shaftIdList)) {
  67. return AdmResponse.failure("必填项:equipId(设备id)、shaftIdList(竖井Id列表)");
  68. }
  69. AdmRelationTypeEnum typeEnum = AdmRelationTypeEnum.EQ2SH_FOR;
  70. String graphCode = typeEnum.getGraphCode();
  71. String relCode = typeEnum.getRelCode();
  72. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  73. deleteByObjId(equipId, null, null, typeEnum, context);
  74. List<RelationDTO> voList = new ArrayList<>();
  75. for (String shaftId : shaftIdList) {
  76. RelationDTO relation = new RelationDTO(null, graphCode, relCode, "", equipId, shaftId);
  77. voList.add(relation);
  78. }
  79. service.doSave(context, voList);
  80. return AdmResponse.success();
  81. }
  82. /**
  83. * 服务于竖井的设备,竖井一对多
  84. * 必填项:ShaftId(竖井id)、EquipIdList(设备id列表),测方法会覆盖以前的记录
  85. *
  86. * @param shForEqList 设备和竖井的关系对象
  87. * @return 创建的结果信息
  88. */
  89. @PostMapping("/link-eq")
  90. public AdmResponse createEq(@RequestBody ShForEqList shForEqList) {
  91. List<String> equipIdList = shForEqList.getEquipIdList();
  92. String shaftId = shForEqList.getShaftId();
  93. //检验输入参数
  94. if (StrUtil.isEmpty(shaftId) || CollectionUtils.isEmpty(equipIdList)) {
  95. return AdmResponse.failure("必填项:equipId(设备id)、shaftIdList(竖井Id列表)");
  96. }
  97. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  98. AdmRelationTypeEnum typeEnum = AdmRelationTypeEnum.EQ2SH_FOR;
  99. String graphCode = typeEnum.getGraphCode();
  100. String relCode = typeEnum.getRelCode();
  101. deleteByObjId(null, shaftId, null, typeEnum, context);
  102. List<RelationDTO> voList = new ArrayList<>();
  103. for (String equipId : equipIdList) {
  104. RelationDTO relation = new RelationDTO(null, graphCode, relCode, "", equipId, shaftId);
  105. voList.add(relation);
  106. }
  107. service.doSave(context, voList);
  108. return AdmResponse.success();
  109. }
  110. /**
  111. * 根据对象删除设备服务于竖井关系,只针对一个对象
  112. * 必填项:EquipId(设备id)、shaftId(竖井id)
  113. *
  114. * @param eqForSh 对象
  115. * @return 删除的结果信息
  116. */
  117. @PostMapping("/unlink")
  118. public AdmResponse delete(@RequestBody EqForSh eqForSh) {
  119. String shaftId = eqForSh.getShaftId();
  120. String equipId = eqForSh.getEquipId();
  121. //检验输入参数
  122. if (StrUtil.isEmpty(equipId) || StrUtil.isEmpty(shaftId)) {
  123. return AdmResponse.failure("必填项:equipId(设备id)、shaftId(竖井Id)");
  124. }
  125. String type = eqForSh.getType();
  126. AdmRelationTypeEnum typeEnum = AdmRelationTypeEnum.EQ2SH_FOR;
  127. deleteByObjId(equipId, shaftId, type, typeEnum, AdmContextUtil.toDmpContext());
  128. return AdmResponse.success();
  129. }
  130. private void deleteByObjId(String objFrom, String objTo, String type, AdmRelationTypeEnum typeEnum, InstanceUrlParam context) {
  131. QueryCriteria criteria = new QueryCriteria();
  132. ObjectNode node = JsonNodeFactory.instance.objectNode();
  133. node.put("graphCode", typeEnum.getGraphCode());
  134. node.put("relCode", typeEnum.getRelCode());
  135. node.put("relValue", type);
  136. node.put("objFrom", objFrom);
  137. node.put("objTo", objTo);
  138. criteria.setCriteria(node);
  139. service.doDelete(context, criteria);
  140. }
  141. }