RelationShContainSpBaseController.java 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303
  1. package com.persagy.proxy.adm.controller;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.util.ObjectUtil;
  4. import cn.hutool.core.util.StrUtil;
  5. import com.alibaba.fastjson.JSONArray;
  6. import com.alibaba.fastjson.JSONObject;
  7. import com.fasterxml.jackson.databind.node.JsonNodeFactory;
  8. import com.fasterxml.jackson.databind.node.ObjectNode;
  9. import com.persagy.dmp.basic.dto.RequestData;
  10. import com.persagy.dmp.basic.model.QueryCriteria;
  11. import com.persagy.dmp.digital.client.DigitalObjectFacade;
  12. import com.persagy.dmp.digital.entity.ObjectDigital;
  13. import com.persagy.dmp.digital.entity.ObjectRelation;
  14. import com.persagy.proxy.adm.constant.AdmCommonConstant;
  15. import com.persagy.proxy.adm.constant.enums.AdmDictCategoryEnum;
  16. import com.persagy.proxy.adm.request.AdmResponse;
  17. import com.persagy.proxy.adm.service.IAdmRelationService;
  18. import com.persagy.proxy.adm.utils.AdmContextUtil;
  19. import com.persagy.proxy.common.client.DmpRwdClient;
  20. import com.persagy.proxy.common.entity.InstanceUrlParam;
  21. import com.persagy.proxy.common.entity.RelationDTO;
  22. import org.springframework.beans.factory.annotation.Autowired;
  23. import org.springframework.beans.factory.annotation.Value;
  24. import org.springframework.web.bind.annotation.*;
  25. import java.util.ArrayList;
  26. import java.util.List;
  27. import java.util.Set;
  28. import java.util.stream.Collectors;
  29. /**
  30. * @description:023、关系-竖井下的业务空间
  31. * @author:lgy
  32. * @data:2021/9/13 14:49
  33. */
  34. @RestController
  35. @RequestMapping("/rel/sh-contain-sp-base")
  36. public class RelationShContainSpBaseController {
  37. @Autowired
  38. private IAdmRelationService service;
  39. @Autowired
  40. DmpRwdClient rwdClient;
  41. @Value("${middleware.group.code}")
  42. private String groupCode;
  43. /**
  44. * 默认图类型
  45. */
  46. public static String DEFAULT_GRAPH_CODE = "ArchSubset";
  47. /**
  48. * 默认边类型
  49. */
  50. public static String DEFAULT_REL_CODE = "Sh2Sp";
  51. /**
  52. * 创建竖井下的业务空间关系
  53. * @param param 竖井和业务空间的关系对象
  54. * @return
  55. * @throws Exception
  56. */
  57. @PostMapping(value = {"/link"})
  58. public AdmResponse create(@RequestBody JSONObject param) throws Exception {
  59. String shaftId = param.getString("shaftId");
  60. String spaceId = param.getString("spaceId");
  61. String type = param.getString("objectType");
  62. if(!StrUtil.isAllNotEmpty(shaftId,spaceId)) {
  63. return AdmResponse.failure("必填项:ShaftID(竖井id)、SpaceID(业务空间id)");
  64. }
  65. // 创建关系对象
  66. List<RelationDTO> voList = new ArrayList<>();
  67. voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId , spaceId));
  68. service.doSave(AdmContextUtil.toDmpContext(), voList);
  69. return AdmResponse.success();
  70. }
  71. /**
  72. * 根据对象删除关系
  73. *根据对象删除竖井下的业务空间关系,只针对一个对象
  74. * @param param 对象
  75. * @param type 空间类型
  76. * @return 删除的结果信息
  77. */
  78. @PostMapping("/unlink")
  79. public AdmResponse delete(@RequestParam(value = "type", required = false) String type,
  80. @RequestBody JSONObject param) throws Exception {
  81. if(CollUtil.isEmpty(param)) {
  82. return AdmResponse.success();
  83. }
  84. // 组装上下文条件
  85. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  86. // 组装条件
  87. QueryCriteria criteria = new QueryCriteria();
  88. String shaftId = param.getString("shaftId");
  89. String spaceId = param.getString("spaceId");
  90. if(!StrUtil.isAllNotEmpty(shaftId,spaceId)) {
  91. return AdmResponse.failure("必填项:ShaftID(竖井id)、SpaceID(业务空间id)");
  92. }
  93. ObjectNode node = JsonNodeFactory.instance.objectNode();
  94. node.put("graphCode", DEFAULT_GRAPH_CODE);
  95. node.put("relCode", DEFAULT_REL_CODE);
  96. node.put("relValue", type);
  97. node.put("objFrom", shaftId);
  98. node.put("objTo",spaceId);
  99. criteria.setCriteria(node);
  100. service.doDelete(context, criteria);
  101. return AdmResponse.success();
  102. }
  103. /**
  104. * 创建竖井下的业务空间关系
  105. *
  106. * @param param 竖井和业务空间的关系对象
  107. * @param type 空间类型
  108. * @return 创建的结果信息
  109. */
  110. @PostMapping("/link-shsps")
  111. public AdmResponse createShsps(@RequestParam(value = "type", required = false) String type,
  112. @RequestBody JSONObject param) {
  113. String shaftId = param.getString("shaftId");
  114. JSONArray spaceIdList = param.getJSONArray("spaceIdList");
  115. if(StrUtil.isBlank(shaftId) || CollUtil.isEmpty(spaceIdList)) {
  116. return AdmResponse.failure("必填项:SpaceIdList(业务空间id列表)、ShaftID(竖井id),此方法不会覆盖以前的记录");
  117. }
  118. // 创建关系对象.先删除,后添加
  119. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  120. List<RelationDTO> voList = new ArrayList<>();
  121. for(int i = 0;i < spaceIdList.size();i++) {
  122. String spaceId = spaceIdList.getString(i);
  123. if(StrUtil.isBlank(spaceId)) {
  124. continue;
  125. }
  126. QueryCriteria queryRequest = new QueryCriteria();
  127. ObjectNode node = JsonNodeFactory.instance.objectNode();
  128. node.put("graphCode", DEFAULT_GRAPH_CODE);
  129. node.put("relCode", DEFAULT_REL_CODE);
  130. node.put("relValue", type);
  131. node.put("objFrom", shaftId);
  132. node.put("objTo", spaceId);
  133. queryRequest.setCriteria(node);
  134. service.doDelete(context,queryRequest);
  135. // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
  136. voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
  137. }
  138. // 组装上下文条件
  139. service.doSave(context, voList);
  140. return AdmResponse.success();
  141. }
  142. /**
  143. * 竖井下的业务空间,空间一对多
  144. * @param param 竖井和业务空间的关系对象
  145. * @param type 空间类型
  146. * @return 创建的结果信息
  147. */
  148. @PostMapping("/link-spshs")
  149. public AdmResponse createSpshs(@RequestParam(value = "type", required = false) String type,
  150. @RequestBody JSONObject param) {
  151. String spaceId = param.getString("spaceId");
  152. JSONArray shaftIdList = param.getJSONArray("shaftIdList");
  153. if(StrUtil.isBlank(spaceId) || CollUtil.isEmpty(shaftIdList)) {
  154. return AdmResponse.failure("必填项:SpaceId(业务空间id)、ShaftIDList(竖井id列表)");
  155. }
  156. // 创建关系对象.先删除,后添加
  157. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  158. List<RelationDTO> voList = new ArrayList<>();
  159. for(int i = 0;i < shaftIdList.size();i++) {
  160. String shaftId = shaftIdList.getString(i);
  161. if(StrUtil.isBlank(shaftId)) {
  162. continue;
  163. }
  164. QueryCriteria queryRequest = new QueryCriteria();
  165. ObjectNode node = JsonNodeFactory.instance.objectNode();
  166. node.put("graphCode", DEFAULT_GRAPH_CODE);
  167. node.put("relCode", DEFAULT_REL_CODE);
  168. node.put("relValue", type);
  169. node.put("objFrom", shaftId);
  170. node.put("objTo", spaceId);
  171. queryRequest.setCriteria(node);
  172. service.doDelete(context,queryRequest);
  173. // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
  174. voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
  175. }
  176. // 组装上下文条件
  177. service.doSave(context, voList);
  178. return AdmResponse.success();
  179. }
  180. /**
  181. * 创建竖井下的业务空间,竖井一对多,此方法会覆盖以前的记录
  182. *
  183. * @param param 竖井和业务空间的关系对象
  184. * @param type 空间类型
  185. * @return 创建的结果信息
  186. */
  187. @PostMapping("/link-shsps-replace")
  188. public AdmResponse createShSpsReplace(@RequestParam(value = "Type", required = false) String type,
  189. @RequestParam(value = "BuildingId", required = false) String buildingId,
  190. @RequestParam(value = "FloorId", required = false) String floorId,
  191. @RequestBody JSONObject param) throws Exception {
  192. String shaftId = param.getString("shaftId");
  193. JSONArray spaceIdList = param.getJSONArray("spaceIdList");
  194. if(StrUtil.isBlank(shaftId) || CollUtil.isEmpty(spaceIdList)) {
  195. return AdmResponse.failure("必填项:SpaceIdList(业务空间id列表)、ShaftID(竖井id),此方法会覆盖以前的记录");
  196. }
  197. // 创建关系对象.先删除,后添加
  198. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  199. RequestData requestData = RequestData.builder().buildingId(buildingId).classCode(type)
  200. .floorId(floorId).projectId(context.getProjectId()).objType(AdmDictCategoryEnum.SPACE.getValue()).build();
  201. QueryCriteria queryRequestTmp = new QueryCriteria();
  202. ObjectNode nodeTmp = JsonNodeFactory.instance.objectNode();
  203. nodeTmp.put("graphCode", DEFAULT_GRAPH_CODE);
  204. nodeTmp.put("relCode", DEFAULT_REL_CODE);
  205. nodeTmp.put("relValue", type);
  206. nodeTmp.put("objFrom", shaftId);
  207. queryRequestTmp.setCriteria(nodeTmp);
  208. //关系查询空间id
  209. Set<String> oldSpaceIds = service.queryByCondition(context,queryRequestTmp).stream().map(ObjectRelation::getObjTo).collect(Collectors.toSet());
  210. List<ObjectDigital> objectDigitals = DigitalObjectFacade.queryObjectListSuperiorId(context.getGroupCode(),
  211. context.getProjectId(), AdmCommonConstant.APP_ID, context.getUserId(), requestData);
  212. //楼层、建筑、对象类查询的空间id
  213. Set<String> spaceIdSet = objectDigitals.stream().map(ObjectDigital::getId).collect(Collectors.toSet());
  214. //空间id取交集
  215. if(ObjectUtil.isNotEmpty(oldSpaceIds)){
  216. spaceIdSet.retainAll(oldSpaceIds);
  217. }
  218. //删除空间关系
  219. for (String oldSpaceId : spaceIdSet) {
  220. QueryCriteria queryRequest = new QueryCriteria();
  221. ObjectNode node = JsonNodeFactory.instance.objectNode();
  222. node.put("graphCode", DEFAULT_GRAPH_CODE);
  223. node.put("relCode", DEFAULT_REL_CODE);
  224. node.put("relValue", type);
  225. node.put("objFrom", shaftId);
  226. node.put("objTo", oldSpaceId);
  227. queryRequest.setCriteria(node);
  228. service.doDelete(context,queryRequest);
  229. }
  230. //替换空间关系
  231. List<RelationDTO> voList = new ArrayList<>();
  232. for(int i = 0;i < spaceIdList.size();i++) {
  233. String spaceId = spaceIdList.getString(i);
  234. if(StrUtil.isBlank(spaceId)) {
  235. continue;
  236. }
  237. QueryCriteria queryRequest = new QueryCriteria();
  238. ObjectNode node = JsonNodeFactory.instance.objectNode();
  239. node.put("graphCode", DEFAULT_GRAPH_CODE);
  240. node.put("relCode", DEFAULT_REL_CODE);
  241. node.put("relValue", type);
  242. node.put("objFrom", shaftId);
  243. node.put("objTo", spaceId);
  244. queryRequest.setCriteria(node);
  245. service.doDelete(context,queryRequest);
  246. // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
  247. voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
  248. }
  249. // 组装上下文条件
  250. service.doSave(context, voList);
  251. return AdmResponse.success();
  252. }
  253. /**
  254. * 查询空间类型和名称
  255. * @param projectId projectId
  256. * @param shaftId shaftId 竖井id
  257. * @param param param
  258. * @return {@link AdmResponse}
  259. * @author lgy
  260. * @date 2021/9/23 18:37
  261. */
  262. @PostMapping("/space-type-query")
  263. public AdmResponse spaceTypeQuery(@RequestParam(value = "shaftId", required = false) String shaftId,
  264. @RequestBody JSONObject param) throws Exception {
  265. // 组装上下文条件
  266. InstanceUrlParam context = AdmContextUtil.toDmpContext();
  267. // 组装条件
  268. QueryCriteria queryRequestTmp = new QueryCriteria();
  269. ObjectNode nodeTmp = JsonNodeFactory.instance.objectNode();
  270. nodeTmp.put("graphCode", DEFAULT_GRAPH_CODE);
  271. nodeTmp.put("relCode", DEFAULT_REL_CODE);
  272. nodeTmp.put("objFrom", shaftId);
  273. queryRequestTmp.setCriteria(nodeTmp);
  274. //关系查询空间id
  275. Set<String> relValue = service.queryByCondition(context,queryRequestTmp).stream().map(ObjectRelation::getRelValue).collect(Collectors.toSet());
  276. JSONObject queryCriteria = new JSONObject();
  277. JSONObject criteria = new JSONObject();
  278. criteria.put("code",relValue);
  279. queryCriteria.put("criteria",criteria);
  280. JSONArray dmpResult = rwdClient.queryDefClass(context, queryCriteria).getData(JSONArray.class);
  281. JSONArray result = new JSONArray();
  282. for (int i = 0; i < dmpResult.size(); i++) {
  283. JSONObject defClass = dmpResult.getJSONObject(i);
  284. JSONObject temp = new JSONObject();
  285. temp.put("objectType",defClass.getString("code"));
  286. temp.put("objectTypeName",defClass.getString("name"));
  287. result.add(temp);
  288. }
  289. return AdmResponse.success(result);
  290. }
  291. }