123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303 |
- package com.persagy.proxy.adm.controller;
- import cn.hutool.core.collection.CollUtil;
- import cn.hutool.core.util.ObjectUtil;
- import cn.hutool.core.util.StrUtil;
- import com.alibaba.fastjson.JSONArray;
- import com.alibaba.fastjson.JSONObject;
- import com.fasterxml.jackson.databind.node.JsonNodeFactory;
- import com.fasterxml.jackson.databind.node.ObjectNode;
- import com.persagy.dmp.basic.dto.RequestData;
- import com.persagy.dmp.basic.model.QueryCriteria;
- import com.persagy.dmp.digital.client.DigitalObjectFacade;
- import com.persagy.dmp.digital.entity.ObjectDigital;
- import com.persagy.dmp.digital.entity.ObjectRelation;
- import com.persagy.proxy.adm.constant.AdmCommonConstant;
- import com.persagy.proxy.adm.constant.enums.AdmDictCategoryEnum;
- import com.persagy.proxy.adm.request.AdmResponse;
- import com.persagy.proxy.adm.service.IAdmRelationService;
- import com.persagy.proxy.adm.utils.AdmContextUtil;
- import com.persagy.proxy.common.client.DmpRwdClient;
- import com.persagy.proxy.common.entity.InstanceUrlParam;
- import com.persagy.proxy.common.entity.RelationDTO;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.beans.factory.annotation.Value;
- import org.springframework.web.bind.annotation.*;
- import java.util.ArrayList;
- import java.util.List;
- import java.util.Set;
- import java.util.stream.Collectors;
- /**
- * @description:023、关系-竖井下的业务空间
- * @author:lgy
- * @data:2021/9/13 14:49
- */
- @RestController
- @RequestMapping("/rel/sh-contain-sp-base")
- public class RelationShContainSpBaseController {
- @Autowired
- private IAdmRelationService service;
- @Autowired
- DmpRwdClient rwdClient;
- @Value("${middleware.group.code}")
- private String groupCode;
- /**
- * 默认图类型
- */
- public static String DEFAULT_GRAPH_CODE = "ArchSubset";
- /**
- * 默认边类型
- */
- public static String DEFAULT_REL_CODE = "Sh2Sp";
- /**
- * 创建竖井下的业务空间关系
- * @param param 竖井和业务空间的关系对象
- * @return
- * @throws Exception
- */
- @PostMapping(value = {"/link"})
- public AdmResponse create(@RequestBody JSONObject param) throws Exception {
- String shaftId = param.getString("shaftId");
- String spaceId = param.getString("spaceId");
- String type = param.getString("objectType");
- if(!StrUtil.isAllNotEmpty(shaftId,spaceId)) {
- return AdmResponse.failure("必填项:ShaftID(竖井id)、SpaceID(业务空间id)");
- }
- // 创建关系对象
- List<RelationDTO> voList = new ArrayList<>();
- voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId , spaceId));
- service.doSave(AdmContextUtil.toDmpContext(), voList);
- return AdmResponse.success();
- }
- /**
- * 根据对象删除关系
- *根据对象删除竖井下的业务空间关系,只针对一个对象
- * @param param 对象
- * @param type 空间类型
- * @return 删除的结果信息
- */
- @PostMapping("/unlink")
- public AdmResponse delete(@RequestParam(value = "type", required = false) String type,
- @RequestBody JSONObject param) throws Exception {
- if(CollUtil.isEmpty(param)) {
- return AdmResponse.success();
- }
- // 组装上下文条件
- InstanceUrlParam context = AdmContextUtil.toDmpContext();
- // 组装条件
- QueryCriteria criteria = new QueryCriteria();
- String shaftId = param.getString("shaftId");
- String spaceId = param.getString("spaceId");
- if(!StrUtil.isAllNotEmpty(shaftId,spaceId)) {
- return AdmResponse.failure("必填项:ShaftID(竖井id)、SpaceID(业务空间id)");
- }
- ObjectNode node = JsonNodeFactory.instance.objectNode();
- node.put("graphCode", DEFAULT_GRAPH_CODE);
- node.put("relCode", DEFAULT_REL_CODE);
- node.put("relValue", type);
- node.put("objFrom", shaftId);
- node.put("objTo",spaceId);
- criteria.setCriteria(node);
- service.doDelete(context, criteria);
- return AdmResponse.success();
- }
- /**
- * 创建竖井下的业务空间关系
- *
- * @param param 竖井和业务空间的关系对象
- * @param type 空间类型
- * @return 创建的结果信息
- */
- @PostMapping("/link-shsps")
- public AdmResponse createShsps(@RequestParam(value = "type", required = false) String type,
- @RequestBody JSONObject param) {
- String shaftId = param.getString("shaftId");
- JSONArray spaceIdList = param.getJSONArray("spaceIdList");
- if(StrUtil.isBlank(shaftId) || CollUtil.isEmpty(spaceIdList)) {
- return AdmResponse.failure("必填项:SpaceIdList(业务空间id列表)、ShaftID(竖井id),此方法不会覆盖以前的记录");
- }
- // 创建关系对象.先删除,后添加
- InstanceUrlParam context = AdmContextUtil.toDmpContext();
- List<RelationDTO> voList = new ArrayList<>();
- for(int i = 0;i < spaceIdList.size();i++) {
- String spaceId = spaceIdList.getString(i);
- if(StrUtil.isBlank(spaceId)) {
- continue;
- }
- QueryCriteria queryRequest = new QueryCriteria();
- ObjectNode node = JsonNodeFactory.instance.objectNode();
- node.put("graphCode", DEFAULT_GRAPH_CODE);
- node.put("relCode", DEFAULT_REL_CODE);
- node.put("relValue", type);
- node.put("objFrom", shaftId);
- node.put("objTo", spaceId);
- queryRequest.setCriteria(node);
- service.doDelete(context,queryRequest);
- // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
- voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
- }
- // 组装上下文条件
- service.doSave(context, voList);
- return AdmResponse.success();
- }
- /**
- * 竖井下的业务空间,空间一对多
- * @param param 竖井和业务空间的关系对象
- * @param type 空间类型
- * @return 创建的结果信息
- */
- @PostMapping("/link-spshs")
- public AdmResponse createSpshs(@RequestParam(value = "type", required = false) String type,
- @RequestBody JSONObject param) {
- String spaceId = param.getString("spaceId");
- JSONArray shaftIdList = param.getJSONArray("shaftIdList");
- if(StrUtil.isBlank(spaceId) || CollUtil.isEmpty(shaftIdList)) {
- return AdmResponse.failure("必填项:SpaceId(业务空间id)、ShaftIDList(竖井id列表)");
- }
- // 创建关系对象.先删除,后添加
- InstanceUrlParam context = AdmContextUtil.toDmpContext();
- List<RelationDTO> voList = new ArrayList<>();
- for(int i = 0;i < shaftIdList.size();i++) {
- String shaftId = shaftIdList.getString(i);
- if(StrUtil.isBlank(shaftId)) {
- continue;
- }
- QueryCriteria queryRequest = new QueryCriteria();
- ObjectNode node = JsonNodeFactory.instance.objectNode();
- node.put("graphCode", DEFAULT_GRAPH_CODE);
- node.put("relCode", DEFAULT_REL_CODE);
- node.put("relValue", type);
- node.put("objFrom", shaftId);
- node.put("objTo", spaceId);
- queryRequest.setCriteria(node);
- service.doDelete(context,queryRequest);
- // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
- voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
- }
- // 组装上下文条件
- service.doSave(context, voList);
- return AdmResponse.success();
- }
- /**
- * 创建竖井下的业务空间,竖井一对多,此方法会覆盖以前的记录
- *
- * @param param 竖井和业务空间的关系对象
- * @param type 空间类型
- * @return 创建的结果信息
- */
- @PostMapping("/link-shsps-replace")
- public AdmResponse createShSpsReplace(@RequestParam(value = "Type", required = false) String type,
- @RequestParam(value = "BuildingId", required = false) String buildingId,
- @RequestParam(value = "FloorId", required = false) String floorId,
- @RequestBody JSONObject param) throws Exception {
- String shaftId = param.getString("shaftId");
- JSONArray spaceIdList = param.getJSONArray("spaceIdList");
- if(StrUtil.isBlank(shaftId) || CollUtil.isEmpty(spaceIdList)) {
- return AdmResponse.failure("必填项:SpaceIdList(业务空间id列表)、ShaftID(竖井id),此方法会覆盖以前的记录");
- }
- // 创建关系对象.先删除,后添加
- InstanceUrlParam context = AdmContextUtil.toDmpContext();
- RequestData requestData = RequestData.builder().buildingId(buildingId).classCode(type)
- .floorId(floorId).projectId(context.getProjectId()).objType(AdmDictCategoryEnum.SPACE.getValue()).build();
- QueryCriteria queryRequestTmp = new QueryCriteria();
- ObjectNode nodeTmp = JsonNodeFactory.instance.objectNode();
- nodeTmp.put("graphCode", DEFAULT_GRAPH_CODE);
- nodeTmp.put("relCode", DEFAULT_REL_CODE);
- nodeTmp.put("relValue", type);
- nodeTmp.put("objFrom", shaftId);
- queryRequestTmp.setCriteria(nodeTmp);
- //关系查询空间id
- Set<String> oldSpaceIds = service.queryByCondition(context,queryRequestTmp).stream().map(ObjectRelation::getObjTo).collect(Collectors.toSet());
- List<ObjectDigital> objectDigitals = DigitalObjectFacade.queryObjectListSuperiorId(context.getGroupCode(),
- context.getProjectId(), AdmCommonConstant.APP_ID, context.getUserId(), requestData);
- //楼层、建筑、对象类查询的空间id
- Set<String> spaceIdSet = objectDigitals.stream().map(ObjectDigital::getId).collect(Collectors.toSet());
- //空间id取交集
- if(ObjectUtil.isNotEmpty(oldSpaceIds)){
- spaceIdSet.retainAll(oldSpaceIds);
- }
- //删除空间关系
- for (String oldSpaceId : spaceIdSet) {
- QueryCriteria queryRequest = new QueryCriteria();
- ObjectNode node = JsonNodeFactory.instance.objectNode();
- node.put("graphCode", DEFAULT_GRAPH_CODE);
- node.put("relCode", DEFAULT_REL_CODE);
- node.put("relValue", type);
- node.put("objFrom", shaftId);
- node.put("objTo", oldSpaceId);
- queryRequest.setCriteria(node);
- service.doDelete(context,queryRequest);
- }
- //替换空间关系
- List<RelationDTO> voList = new ArrayList<>();
- for(int i = 0;i < spaceIdList.size();i++) {
- String spaceId = spaceIdList.getString(i);
- if(StrUtil.isBlank(spaceId)) {
- continue;
- }
- QueryCriteria queryRequest = new QueryCriteria();
- ObjectNode node = JsonNodeFactory.instance.objectNode();
- node.put("graphCode", DEFAULT_GRAPH_CODE);
- node.put("relCode", DEFAULT_REL_CODE);
- node.put("relValue", type);
- node.put("objFrom", shaftId);
- node.put("objTo", spaceId);
- queryRequest.setCriteria(node);
- service.doDelete(context,queryRequest);
- // 创建关系对象 - 参考(datasyn中DataCenterSync.kt)
- voList.add(new RelationDTO(null, DEFAULT_GRAPH_CODE, DEFAULT_REL_CODE, type, shaftId, spaceId));
- }
- // 组装上下文条件
- service.doSave(context, voList);
- return AdmResponse.success();
- }
- /**
- * 查询空间类型和名称
- * @param projectId projectId
- * @param shaftId shaftId 竖井id
- * @param param param
- * @return {@link AdmResponse}
- * @author lgy
- * @date 2021/9/23 18:37
- */
- @PostMapping("/space-type-query")
- public AdmResponse spaceTypeQuery(@RequestParam(value = "shaftId", required = false) String shaftId,
- @RequestBody JSONObject param) throws Exception {
- // 组装上下文条件
- InstanceUrlParam context = AdmContextUtil.toDmpContext();
- // 组装条件
- QueryCriteria queryRequestTmp = new QueryCriteria();
- ObjectNode nodeTmp = JsonNodeFactory.instance.objectNode();
- nodeTmp.put("graphCode", DEFAULT_GRAPH_CODE);
- nodeTmp.put("relCode", DEFAULT_REL_CODE);
- nodeTmp.put("objFrom", shaftId);
- queryRequestTmp.setCriteria(nodeTmp);
- //关系查询空间id
- Set<String> relValue = service.queryByCondition(context,queryRequestTmp).stream().map(ObjectRelation::getRelValue).collect(Collectors.toSet());
- JSONObject queryCriteria = new JSONObject();
- JSONObject criteria = new JSONObject();
- criteria.put("code",relValue);
- queryCriteria.put("criteria",criteria);
- JSONArray dmpResult = rwdClient.queryDefClass(context, queryCriteria).getData(JSONArray.class);
- JSONArray result = new JSONArray();
- for (int i = 0; i < dmpResult.size(); i++) {
- JSONObject defClass = dmpResult.getJSONObject(i);
- JSONObject temp = new JSONObject();
- temp.put("objectType",defClass.getString("code"));
- temp.put("objectTypeName",defClass.getString("name"));
- result.add(temp);
- }
- return AdmResponse.success(result);
- }
- }
|