提交代码

amos-boot-module/amos-boot-module-biz/amos-boot-module-common-biz/src/main/java/com/yeejoin/amos/boot/module/common/biz/controller/OrgUsrController.java

 package com.yeejoin.amos.boot.module.common.biz.controller;
 
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 
 import javax.servlet.http.HttpServletRequest;
 
+import com.yeejoin.amos.boot.biz.common.controller.DataDictionaryController;
+import com.yeejoin.amos.boot.biz.common.entity.BaseEntity;
+import com.yeejoin.amos.boot.biz.common.entity.DataDictionary;
+import com.yeejoin.amos.boot.biz.common.service.impl.DataDictionaryServiceImpl;
+import com.yeejoin.amos.boot.biz.common.utils.Menu;
+import com.yeejoin.amos.boot.biz.common.utils.TreeParser;
+import com.yeejoin.amos.boot.module.common.api.mapper.OrgUsrMapper;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -71,6 +75,9 @@ import io.swagger.annotations.ApiOperation;
 public class OrgUsrController extends BaseController {
 
 	@Autowired
+	DataDictionaryServiceImpl iDataDictionaryService;
+
+	@Autowired
 	OrgUsrServiceImpl iOrgUsrService;
 
 	@Autowired
@@ -83,6 +90,9 @@ public class OrgUsrController extends BaseController {
 	FireTeamMapper fireTeamMapper;
 
 	@Autowired
+	OrgUsrMapper orgUsrMapper;
+
+	@Autowired
 	EmqKeeper emqKeeper;
 	@Value("${jcs.company.topic.delete:jcs/company/topic/delete}")
 	private String airportDeleteTopic;
@@ -90,6 +100,12 @@ public class OrgUsrController extends BaseController {
 	@Autowired
 	EquipFeignClient equipFeignClient;
 
+	@Autowired
+	DataDictionaryController dataDictionaryController;
+	@Value("special-dept")
+	String deptKey;
+	@Value("special-deptType")
+	String deptTypeKey;
 	/**
 	 * 新增单位信息
 	 *
@@ -742,15 +758,35 @@ public class OrgUsrController extends BaseController {
 	@ApiOperation(httpMethod = "GET", value = "根据机构类型(逗号分割),机构编码获取列表不分页", notes = "根据机构类型(逗号分割),机构编码获取列表不分页")
 	@GetMapping(value = "/{authKey}/listWithAuth")
 	public ResponseModel<List<OrgUsr>> getListWithAuth(@RequestParam(required = false) String orgTypes,
-			@PathVariable String authKey) {
+			@PathVariable String authKey) throws Exception {
 		ReginParams reginParams = getSelectedOrgInfo();
 		ReginParams.PersonIdentity personIdentity = reginParams.getPersonIdentity();
 		String bizOrgCode = personIdentity.getBizOrgCode();
+
+
+		//拿到数据字典中存储的 特殊类型 特殊部门
+		QueryWrapper<DataDictionary> queryWrapper = new QueryWrapper<>();
+		queryWrapper.eq("type", "01").or().eq("type", "02");
+		Collection<DataDictionary> list = iDataDictionaryService.list(queryWrapper);
+		ArrayList<String> objects = new ArrayList<>();
+		list.stream().forEach(e->objects.add(e.getName()));
+
+		if (objects.contains(authKey)){
+				if (objects.contains(personIdentity.getCompanyId())) {
+					LambdaQueryWrapper<OrgUsr> wrapper = new LambdaQueryWrapper<>();
+					wrapper.eq(OrgUsr::getParentId,personIdentity.getCompanyId());
+					wrapper.eq(BaseEntity::getIsDelete,false);
+					OrgUsr orgUsr = orgUsrMapper.selectOne(wrapper);
+					personIdentity.setBizOrgCode(orgUsr.getBizOrgCode() != null? orgUsr.getBizOrgCode() : personIdentity.getBizOrgCode());
+				}
+		}
 		// 权限处理
 		PermissionInterceptorContext.setDataAuthRule(authKey);
 		return ResponseHelper.buildResponse(iOrgUsrService.getListByBizOrgTypeCode(orgTypes, bizOrgCode));
 	}
 
+
+
 	@PersonIdentify
 	@TycloudOperation(ApiLevel = UserType.AGENCY)
 	@RequestMapping(value = "/{authKey}/treeWithAuth", method = RequestMethod.GET)