AdmSpaceServiceImpl.java 23 KB

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