AdmSpaceServiceImpl.java 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531
  1. package com.persagy.proxy.adm.service.impl;
  2. import cn.hutool.core.collection.CollUtil;
  3. import cn.hutool.core.util.StrUtil;
  4. import cn.hutool.json.JSONUtil;
  5. import com.alibaba.fastjson.JSONArray;
  6. import com.alibaba.fastjson.JSONObject;
  7. import com.fasterxml.jackson.core.JsonProcessingException;
  8. import com.fasterxml.jackson.databind.ObjectMapper;
  9. import com.fasterxml.jackson.databind.node.ArrayNode;
  10. import com.fasterxml.jackson.databind.node.JsonNodeFactory;
  11. import com.fasterxml.jackson.databind.node.ObjectNode;
  12. import com.persagy.dmp.basic.model.QueryCriteria;
  13. import com.persagy.dmp.define.entity.ObjectTypeDefine;
  14. import com.persagy.dmp.digital.entity.ObjectRelation;
  15. import com.persagy.proxy.adm.model.*;
  16. import com.persagy.proxy.adm.request.AdmQueryCriteria;
  17. import com.persagy.proxy.adm.request.AdmResponse;
  18. import com.persagy.proxy.adm.service.IAdmEquipmentService;
  19. import com.persagy.proxy.adm.service.IAdmRelationService;
  20. import com.persagy.proxy.adm.service.IAdmSpaceService;
  21. import com.persagy.proxy.adm.service.IAdmSystemService;
  22. import com.persagy.proxy.adm.utils.AdmContextUtil;
  23. import com.persagy.proxy.common.client.DmpRwdClient;
  24. import com.persagy.proxy.common.entity.DmpResult;
  25. import com.persagy.proxy.common.entity.InstanceUrlParam;
  26. import com.persagy.proxy.common.entity.RelationDTO;
  27. import lombok.extern.slf4j.Slf4j;
  28. import org.springframework.beans.factory.annotation.Autowired;
  29. import org.springframework.stereotype.Service;
  30. import java.util.*;
  31. import java.util.stream.Collectors;
  32. /**
  33. * ADM 空间 实现类
  34. * @author lvxy
  35. * @date 2021/8/18
  36. */
  37. @Slf4j
  38. @Service
  39. public class AdmSpaceServiceImpl extends AbstractAdmBaseServiceImpl<AdmSpace> implements IAdmSpaceService {
  40. @Autowired
  41. private DmpRwdClient rwdClient;
  42. @Autowired
  43. private ObjectMapper objectMapper;
  44. @Autowired
  45. private IAdmRelationService relationService;
  46. @Autowired
  47. private IAdmEquipmentService equipmentService;
  48. @Autowired
  49. private IAdmSystemService systemService;
  50. @Override
  51. public AdmResponse doQuery(InstanceUrlParam context, AdmQueryCriteria request, Class<AdmSpace> clazz) {
  52. AdmResponse response = super.doQuery(context, request, clazz);
  53. // 系统需要补充分类名称属性
  54. ensureSpecialValues(context, (List<AdmSpace>) response.getContent());
  55. return response;
  56. }
  57. @Override
  58. public List<AdmSpace> doInsertExt(InstanceUrlParam context, Class<AdmSpace> clazz, List<AdmSpace> voList) {
  59. Map<String, AdmSpace> spaceMapByFlId = CollUtil.fieldValueMap(voList, "floorId");
  60. //查询空间中floorId 对应的floor
  61. DmpResult<JSONArray> floors = queryFloorsByFloorIds(context, spaceMapByFlId.keySet());
  62. if(CollUtil.isNotEmpty(floors.getData())){
  63. ArrayList<AdmFloor> floorList = CollUtil.newArrayList(JSONUtil.toList(floors.getData().toString(),AdmFloor.class));
  64. //space.height = floor.infos.height
  65. voList = processHeight(spaceMapByFlId, floorList);
  66. }
  67. //插入空间
  68. List<AdmSpace> vosR = doInsert(context, clazz, voList);
  69. //添加绑定关系
  70. processRelations(vosR);
  71. //处理关系标记
  72. //TODO 处理关系标记
  73. //processRelationType();
  74. return vosR;
  75. }
  76. @Override
  77. public AdmResponse doQueryType(InstanceUrlParam context, AdmDCSQueryRequest request) {
  78. //查询定义类型
  79. ObjectNode criteriaEq = JsonNodeFactory.instance.objectNode();
  80. criteriaEq.put("parentCode", "space");
  81. DmpResult<JSONArray> resultDef = queryDefineType(context, criteriaEq);
  82. List<JSONObject> defineList = null;
  83. if(resultDef.getResult().equals(DmpResult.SUCCESS)){
  84. defineList = JSONUtil.toList(resultDef.getData().toString(), JSONObject.class);
  85. //获取实际空间对象的数量
  86. //根据classCode floorId 查询查询实际存在的空间
  87. assembleDefineTypes(context, defineList, request.getFloorId());
  88. // 设置返回值
  89. AdmResponse response = AdmResponse.success(defineList);
  90. Long total = resultDef.getCount() == null ? null : resultDef.getCount().longValue();
  91. response.setTotal(total);
  92. return response;
  93. }else{
  94. return null;
  95. }
  96. }
  97. @Override
  98. public AdmResponse doQuerySpaces(InstanceUrlParam context, AdmDCSQueryRequest request) {
  99. AdmQueryCriteria admQueryCriteria = new AdmQueryCriteria();
  100. admQueryCriteria.setCascade(request.getCascade());
  101. admQueryCriteria.setName(AdmSpace.OBJ_TYPE);
  102. String classCodeFilter = "";
  103. if(StrUtil.isNotEmpty(request.getZoneType())){
  104. classCodeFilter = "classCode = '"+request.getZoneType()+"'";
  105. }else{
  106. StringBuffer zoneTypefilters = new StringBuffer(" classCode ");
  107. zoneTypefilters.append("in ['");
  108. request.getZoneTypeList().stream().forEach(code ->{
  109. zoneTypefilters.append(code).append("',");
  110. });
  111. classCodeFilter = StrUtil.sub(zoneTypefilters.toString(),0,-1)+"']";
  112. }
  113. String filters = request.getFilters();
  114. if(StrUtil.isBlank(filters)){
  115. admQueryCriteria.setFilters(classCodeFilter);
  116. }else{
  117. admQueryCriteria.setFilters(filters+";"+classCodeFilter);
  118. }
  119. if(StrUtil.isNotEmpty(request.getBuildingId()) && StrUtil.isNotEmpty(request.getFloorId()) && !request.getFloorId().equals("isnull")){
  120. //明确建筑,明确楼层
  121. ObjectNode relationTo = JsonNodeFactory.instance.objectNode();
  122. relationTo.put("relCode", "Fl2Sp");
  123. relationTo.put("graphCode", "MechInArch");
  124. relationTo.put("objFrom", request.getFloorId());
  125. admQueryCriteria.setRelationFrom(relationTo);
  126. }else if(StrUtil.isNotEmpty(request.getBuildingId()) && StrUtil.isNotEmpty(request.getFloorId()) && request.getFloorId().equals("isnull")){
  127. //明确建筑,不明确楼层
  128. //查询建筑下的楼层
  129. QueryCriteria queryCriteriaRelation = new QueryCriteria();
  130. ObjectNode criteria = JsonNodeFactory.instance.objectNode();
  131. criteria.put("relCode", "Bd2Fl");
  132. criteria.put("graphCode", "ArchSubset");
  133. criteria.put("objFrom", request.getBuildingId());
  134. queryCriteriaRelation.setCriteria(criteria);
  135. List<ObjectRelation> relations = relationService.queryByCondition(AdmContextUtil.toDmpContext(), queryCriteriaRelation);
  136. if(CollUtil.isNotEmpty(relations)){
  137. Map<String, ObjectRelation> objectRelationMap = CollUtil.fieldValueMap(relations, "objTo");
  138. //查询 建筑 不在楼层下 的空间
  139. ObjectNode relationTo = JsonNodeFactory.instance.objectNode();
  140. relationTo.put("relCode", "Fl2Sp");
  141. relationTo.put("graphCode", "MechInArch");
  142. ArrayNode arrayNode = relationTo.putObject("objFrom").putArray("$in");
  143. objectRelationMap.keySet().stream().forEach(floorId -> arrayNode.add(floorId));
  144. relationTo.put("$bindFlag", false);
  145. admQueryCriteria.setRelationFrom(relationTo);
  146. admQueryCriteria.setFilters(admQueryCriteria.getFilters() + "; buildingId = '"+request.getBuildingId()+"'");
  147. }else{
  148. //查询建筑下的空间
  149. ObjectNode relationTo = JsonNodeFactory.instance.objectNode();
  150. relationTo.put("relCode", "Bd2Sp");
  151. relationTo.put("graphCode", "MechInArch");
  152. relationTo.put("objFrom", request.getBuildingId());
  153. admQueryCriteria.setRelationFrom(relationTo);
  154. }
  155. }else if(StrUtil.isNotEmpty(request.getBuildingId()) && StrUtil.isEmpty(request.getFloorId()) && !request.getBuildingId().equals("isnull")){
  156. //查询建筑下的空间
  157. ObjectNode relationTo = JsonNodeFactory.instance.objectNode();
  158. relationTo.put("relCode", "Bd2Sp");
  159. relationTo.put("graphCode", "MechInArch");
  160. relationTo.put("objFrom", request.getBuildingId());
  161. admQueryCriteria.setRelationFrom(relationTo);
  162. }else if(StrUtil.isNotEmpty(request.getBuildingId()) && StrUtil.isEmpty(request.getFloorId()) && request.getBuildingId().equals("isnull")){
  163. //未明确建筑下的空间
  164. //查询建筑下的空间
  165. ObjectNode relationTo = JsonNodeFactory.instance.objectNode();
  166. relationTo.put("relCode", "Bd2Sp");
  167. relationTo.put("graphCode", "MechInArch");
  168. relationTo.put("$bindFlag", false);
  169. admQueryCriteria.setRelationFrom(relationTo);
  170. }else if(StrUtil.isEmpty(request.getBuildingId()) && StrUtil.isEmpty(request.getFloorId())){
  171. }
  172. AdmResponse response = doQuery(AdmContextUtil.toDmpContext(), admQueryCriteria, AdmSpace.class);
  173. //空间需要补充分类名称属性
  174. ensureSpecialValues(context, (List<AdmSpace>) response.getContent());
  175. return response;
  176. }
  177. @Override
  178. public void ensureSpecialValues(InstanceUrlParam context, List<AdmSpace> vos) {
  179. if(CollUtil.isEmpty(vos)) {
  180. return;
  181. }
  182. // 1. 收集空间的设备对象,扩展设备对象赋值
  183. List<AdmEquipment> equipmentList = new ArrayList<>();
  184. for(AdmSpace vo:vos) {
  185. // 收集设备对象
  186. if(CollUtil.isEmpty(vo.getEquipment()) && CollUtil.isEmpty(vo.getEquipmentFor())) {
  187. continue;
  188. }
  189. if(CollUtil.isNotEmpty(vo.getEquipment())) {
  190. equipmentList.addAll(vo.getEquipment());
  191. }
  192. if(CollUtil.isEmpty(vo.getEquipmentFor())) {
  193. equipmentList.addAll(vo.getEquipmentFor());
  194. }
  195. }
  196. // 对象扩展属性赋值
  197. equipmentService.ensureSpecialValues(context, equipmentList);
  198. //2.收集空间的系统 扩展系统、及系统下设备对象赋值
  199. List<AdmSystem> admSystemList = new ArrayList<>();
  200. for(AdmSpace vo:vos) {
  201. // 收集设备对象
  202. if(CollUtil.isEmpty(vo.getGeneralSystem())) {
  203. continue;
  204. }
  205. admSystemList.addAll(vo.getGeneralSystem());
  206. }
  207. // 对象扩展属性赋值
  208. systemService.ensureSpecialValues(context, admSystemList);
  209. }
  210. @Override
  211. public List<AdmSpace> querySpaceInShaft(InstanceUrlParam context, String shaftId, String spaceType) {
  212. AdmQueryCriteria request = new AdmQueryCriteria();
  213. request.setName(AdmSpace.OBJ_TYPE);
  214. request.resetRelationCond(false, "ArchSubset", "Sh2Sp", spaceType, shaftId, true);
  215. AdmResponse rs = doQuery(context, request, AdmSpace.class);
  216. return (List<AdmSpace>) rs.getContent();
  217. }
  218. @Override
  219. public List<AdmSpace> querySpaceInBuilding(InstanceUrlParam context, String buildingId, String spaceType) {
  220. AdmQueryCriteria request = new AdmQueryCriteria();
  221. request.setName(AdmSpace.OBJ_TYPE);
  222. request.resetRelationCond(false, "ArchSubset", "Bd2Sp", spaceType, buildingId, true);
  223. AdmResponse rs = doQuery(context, request, AdmSpace.class);
  224. return (List<AdmSpace>) rs.getContent();
  225. }
  226. @Override
  227. public Map<String, List<String>> querySpaceMapInFloors(InstanceUrlParam context, List<String> floorIds, String spaceType) {
  228. QueryCriteria queryRequest = new QueryCriteria();
  229. ObjectNode criteriaRel = JsonNodeFactory.instance.objectNode();
  230. criteriaRel.put("graphCode", "ArchSubset");
  231. criteriaRel.put("relCode", "Fl2Sp");
  232. criteriaRel.put("relValue", spaceType);
  233. ArrayNode objNode = criteriaRel.putArray("objFrom");
  234. floorIds.forEach(floorId -> objNode.add(floorId));
  235. queryRequest.setCriteria(criteriaRel);
  236. List<ObjectRelation> relations = relationService.queryByCondition(context, queryRequest);
  237. if(CollUtil.isEmpty(relations)) {
  238. return null;
  239. }
  240. return relations.stream().collect(Collectors.groupingBy(ObjectRelation::getObjFrom,
  241. Collectors.mapping(ObjectRelation::getObjTo, Collectors.toList())));
  242. }
  243. @Override
  244. public Map<String, Set<String>> queryTrafficSpaceMap(InstanceUrlParam context, Iterable<String> spaceIds, String spaceType) {
  245. Map<String, Set<String>> result = new HashMap<>();
  246. loadTrafficSpace(context, spaceIds, spaceType, true, result);
  247. loadTrafficSpace(context, spaceIds, spaceType, false, result);
  248. return result;
  249. }
  250. /**
  251. * 装载交通关系
  252. * @param context
  253. * @param spaceIds
  254. * @param spaceType
  255. * @param isFrom
  256. * @param result
  257. */
  258. private void loadTrafficSpace(InstanceUrlParam context, Iterable<String> spaceIds, String spaceType, boolean isFrom, Map<String, Set<String>> result) {
  259. // 关系查询条件
  260. QueryCriteria queryRequest = new QueryCriteria();
  261. ObjectNode criteriaRel = JsonNodeFactory.instance.objectNode();
  262. criteriaRel.put("graphCode", "TrafficNetwork");
  263. criteriaRel.put("relValue", spaceType);
  264. ArrayNode objNode = criteriaRel.putArray(isFrom?"objFrom":"objTo");
  265. spaceIds.forEach(spaceId -> objNode.add(spaceId));
  266. queryRequest.setCriteria(criteriaRel);
  267. List<ObjectRelation> relations = relationService.queryByCondition(context, queryRequest);
  268. if(CollUtil.isEmpty(relations)) {
  269. return;
  270. }
  271. // 存入Map
  272. for(ObjectRelation relation:relations) {
  273. String key = isFrom ? relation.getObjFrom() : relation.getObjTo();
  274. String value = isFrom ? relation.getObjTo() : relation.getObjFrom();
  275. Set<String> valueList = result.get(key);
  276. if(CollUtil.isEmpty(valueList)) {
  277. valueList = new HashSet<>();
  278. result.put(key, valueList);
  279. }
  280. valueList.add(value);
  281. }
  282. }
  283. /**
  284. * 空间的类型是根据项目自定义的
  285. * 参见中心
  286. * @Select("SELECT s.object_type as type,d.name FROM zone_space_base s
  287. * LEFT JOIN dict.dict_space_project d ON s.object_type = d.id WHERE s.project_id = #{projectId} AND d.project_id = #{projectId}
  288. * GROUP BY object_type, d.name")
  289. * @param context
  290. * @return
  291. */
  292. @Override
  293. public List<JSONObject> doQuerySpaceTypes(InstanceUrlParam context) {
  294. //根据项目id查询空间类型
  295. Map<String, String> codeNameMap = queryProjectDefineType(context);
  296. if(CollUtil.isNotEmpty(codeNameMap)){
  297. //查询object
  298. JSONArray demObjs = queryAndDealObjs(context, codeNameMap.keySet());
  299. if(CollUtil.isNotEmpty(demObjs)){
  300. List<AdmSpace> spaceList = JSONArray.parseArray(JSONUtil.toJsonStr(demObjs), AdmSpace.class);
  301. Set<String> classCodes = spaceList.stream().map(space -> space.getClassCode()).collect(Collectors.toSet());
  302. if(CollUtil.isNotEmpty(classCodes)){
  303. codeNameMap.keySet().removeIf(code -> !classCodes.contains(code));
  304. }
  305. }
  306. }
  307. if(CollUtil.isNotEmpty(codeNameMap)){
  308. List<JSONObject> list = new ArrayList<>(codeNameMap.size());
  309. codeNameMap.keySet().stream().forEach(code ->{
  310. JSONObject jsonObject = new JSONObject();
  311. jsonObject.put("name",codeNameMap.get(code));
  312. jsonObject.put("code",code);
  313. list.add(jsonObject);
  314. });
  315. return list;
  316. }else{
  317. return null;
  318. }
  319. }
  320. private DmpResult<JSONArray> queryObj(InstanceUrlParam context, QueryCriteria criteriaObject){
  321. JSONObject paraPe = null;
  322. try {
  323. String paraStr = objectMapper.writeValueAsString(criteriaObject);
  324. paraPe = JSONObject.parseObject(paraStr);
  325. } catch (JsonProcessingException e) {
  326. throw new RuntimeException("转换查询条件出错:" + e.getMessage());
  327. }
  328. return rwdClient.queryObject(context, paraPe);
  329. }
  330. private QueryCriteria ensureQueryCommParm(QueryCriteria dmpRequest, InstanceUrlParam context, Collection<String> classCodes, String objType){
  331. ObjectNode criteriaEq = null;
  332. if(JSONUtil.isNull(dmpRequest.getCriteria())){
  333. criteriaEq = JsonNodeFactory.instance.objectNode();
  334. }else{
  335. criteriaEq = dmpRequest.getCriteria();
  336. }
  337. if(CollUtil.isNotEmpty(classCodes)){
  338. ArrayNode array = criteriaEq.putObject("classCode").putArray("$in");
  339. classCodes.forEach(code -> array.add(code));
  340. }
  341. criteriaEq.put("objType", objType);
  342. dmpRequest.setCriteria(criteriaEq);
  343. return dmpRequest;
  344. }
  345. private JSONArray queryAndDealObjs(InstanceUrlParam context, Set<String> codes){
  346. QueryCriteria criteriaObject = ensureQueryCommParm(new QueryCriteria(), context, codes,AdmSpace.OBJ_TYPE);
  347. criteriaObject.setSize(50L);
  348. DmpResult<JSONArray> dmpResult = queryObj(context, criteriaObject);
  349. if(CollUtil.isEmpty(dmpResult.getData())){
  350. return null;
  351. }
  352. JSONArray demObjs = dmpResult.getData();
  353. if(dmpResult.getCount() > criteriaObject.getSize()){
  354. for(int page = 2; page < dmpResult.getCount()/criteriaObject.getSize()+1; page++){
  355. criteriaObject.setPage(page*1L);
  356. demObjs.addAll(queryObj(context, criteriaObject).getData());
  357. }
  358. }
  359. return demObjs;
  360. }
  361. private Map<String, String> queryProjectDefineType(InstanceUrlParam context){
  362. Map<String, String> codeNameMap = null;
  363. ObjectNode criteriaEq = JsonNodeFactory.instance.objectNode();
  364. criteriaEq.put("projectId", context.getProjectId());
  365. criteriaEq.put("objType", AdmSpace.OBJ_TYPE);
  366. DmpResult<JSONArray> resultDef = queryDefineType(context, criteriaEq);
  367. if(resultDef.getResult().equals(DmpResult.SUCCESS) && CollUtil.isNotEmpty(resultDef.getData())){
  368. List<ObjectTypeDefine> defineList = JSONUtil.toList(resultDef.getData().toString(), ObjectTypeDefine.class);
  369. codeNameMap = defineList.stream().collect(Collectors.toMap(ObjectTypeDefine::getCode, ObjectTypeDefine::getName, (k1,k2)->k1));
  370. }
  371. return codeNameMap;
  372. }
  373. /**
  374. * 查询空间flooId对应的infos中height存在的 flool
  375. * @param context
  376. * @param floolIds
  377. * @return
  378. */
  379. private DmpResult<JSONArray> queryFloorsByFloorIds(InstanceUrlParam context, Collection<String> floolIds){
  380. QueryCriteria queryCriteria = ensureQueryCommParm(new QueryCriteria(),context, CollUtil.newArrayList("floor"),"floor");
  381. ObjectNode criteriaPe = queryCriteria.getCriteria();
  382. if(CollUtil.isNotEmpty(floolIds)){
  383. ArrayNode array = criteriaPe.putObject("id").putArray("$in");
  384. floolIds.forEach(floorId -> array.add(floorId));
  385. }
  386. JSONObject height = new JSONObject();
  387. height.put("$gt", 0f);
  388. criteriaPe.putPOJO("height", height);
  389. queryCriteria.setCriteria(criteriaPe);
  390. queryCriteria.setWithoutCount(true);
  391. return queryObj(context, queryCriteria);
  392. }
  393. /**
  394. * space.height = floor.infos.height
  395. * @param spaceMapByFlId
  396. * @param floorList
  397. * @return
  398. */
  399. private List<AdmSpace> processHeight(Map<String, AdmSpace> spaceMapByFlId, List<AdmFloor> floorList){
  400. floorList.forEach(admFloor -> {
  401. if(CollUtil.isNotEmpty(admFloor.getInfos())){
  402. Float height = (Float) admFloor.getInfos().get("height");
  403. AdmSpace admSpace = spaceMapByFlId.get(admFloor.getId());
  404. admSpace.setHeight(height);
  405. }
  406. });
  407. return (List<AdmSpace>) spaceMapByFlId.values();
  408. }
  409. /**
  410. * 添加绑定关系
  411. */
  412. private void processRelations(List<AdmSpace> vosR){
  413. if(CollUtil.isNotEmpty(vosR)) {
  414. List<RelationDTO> relationDTOS = new ArrayList<>(vosR.size());
  415. vosR.forEach(admSpace -> {
  416. if (StrUtil.isNotEmpty(admSpace.getBuildingId())) {
  417. relationDTOS.add(new RelationDTO(null, "ArchSubset", "Bd2Sp", null, admSpace.getBuildingId(), admSpace.getId()));
  418. }
  419. if (StrUtil.isNotEmpty(admSpace.getFloorId())) {
  420. relationDTOS.add(new RelationDTO(null, "ArchSubset", "Fl2Sp", null, admSpace.getFloorId(), admSpace.getId()));
  421. }
  422. });
  423. if (CollUtil.isNotEmpty(relationDTOS)) {
  424. relationService.doSave(AdmContextUtil.toDmpContext(), relationDTOS);
  425. }
  426. }
  427. }
  428. /**
  429. * 转换为目标实体,并增加属性
  430. */
  431. private void assembleDefineTypes(InstanceUrlParam context, List<JSONObject> defineList, String floorId){
  432. boolean needCount = true;
  433. defineList.stream().forEach(defineType -> {
  434. //查询空间数量
  435. int count = needCount ? countSpace(context, defineType.getString("code"), floorId) : 0;
  436. //处理单个对象
  437. assembleDefineObj(defineType, count, context.getProjectId());
  438. });
  439. }
  440. /**
  441. * 单个转换为目标实体,并增加属性
  442. */
  443. private void assembleDefineObj(JSONObject defineType, int count, String projectId){
  444. defineType.put("count", count);
  445. defineType.put("projectId", projectId);
  446. defineType.remove("aliasName");
  447. defineType.remove("aliasCode");
  448. defineType.remove("valid");
  449. defineType.remove("objType");
  450. defineType.remove("groupCode");
  451. defineType.remove("parentCode");
  452. defineType.remove("id");
  453. }
  454. private void addRelationTo(QueryCriteria rwdQueryCriteria, String objFrom, String relCode){
  455. ObjectNode relationTO = JsonNodeFactory.instance.objectNode();
  456. relationTO.put("relCode", relCode);
  457. relationTO.put("graphCode", "ArchSubset");
  458. relationTO.put("objFrom", objFrom);
  459. ObjectNode criteria = rwdQueryCriteria.getCriteria();
  460. criteria.putPOJO("relationTo", relationTO);
  461. rwdQueryCriteria.setCriteria(criteria);
  462. }
  463. private void addRelationFrom(QueryCriteria rwdQueryCriteria, String objTo, String relCode){
  464. ObjectNode relationFrom = JsonNodeFactory.instance.objectNode();
  465. if(StrUtil.isNotEmpty(objTo)){
  466. relationFrom.put("objTo", objTo);
  467. }
  468. ObjectNode criteria = rwdQueryCriteria.getCriteria();
  469. criteria.putPOJO("relationFrom", relationFrom);
  470. rwdQueryCriteria.setCriteria(criteria);
  471. }
  472. /**
  473. * 查询存在空间的数量
  474. * @param context
  475. * @param classCode
  476. * @param objFrom
  477. * @return
  478. */
  479. private int countSpace(InstanceUrlParam context, String classCode, String objFrom){
  480. QueryCriteria queryCriteria = ensureQueryCommParm(new QueryCriteria(), context, CollUtil.newArrayList(classCode),"space");
  481. addRelationTo(queryCriteria, objFrom,"Fl2Sp");
  482. queryCriteria.setOnlyCount(true);
  483. DmpResult<JSONArray> dmpResult = queryObj(context, queryCriteria);
  484. return dmpResult.getCount();
  485. }
  486. private DmpResult<JSONArray> queryDefineType(InstanceUrlParam context, ObjectNode criteriaEq){
  487. QueryCriteria criteriaObject = new QueryCriteria();
  488. criteriaObject.setCriteria(criteriaEq);
  489. JSONObject paraPe = null;
  490. try {
  491. String paraStr = objectMapper.writeValueAsString(criteriaObject);
  492. paraPe = JSONObject.parseObject(paraStr);
  493. } catch (JsonProcessingException e) {
  494. throw new RuntimeException("查询公共定义的空间类型,转换查询条件出错:" + e.getMessage());
  495. }
  496. return rwdClient.queryDefClass(context, paraPe);
  497. }
  498. }