不选择部门登陆自动添加部门

YeeAmosFireAutoSysService/src/main/java/com/yeejoin/amos/fas/config/PermissionAspect.java

@@ -3,6 +3,10 @@ package com.yeejoin.amos.fas.config;
 
 import javax.servlet.http.HttpServletRequest;
 
+import com.yeejoin.amos.component.feign.config.InnerInvokException;
+import com.yeejoin.amos.fas.business.feign.RemoteSecurityService;
+import com.yeejoin.amos.fas.business.vo.DepartmentBo;
+import com.yeejoin.amos.feign.privilege.model.DepartmentModel;
 import org.aspectj.lang.JoinPoint;
 import org.aspectj.lang.annotation.Aspect;
 import org.aspectj.lang.annotation.Before;
@@ -29,6 +33,11 @@ import com.yeejoin.amos.feign.privilege.Privilege;
 import com.yeejoin.amos.feign.privilege.model.AgencyUserModel;
 import com.yeejoin.amos.feign.privilege.model.CompanyModel;
 
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
 @Aspect
 @Component
 @ResponseBody
@@ -37,6 +46,9 @@ public class PermissionAspect {
 	
 	 @Autowired
 	 private RedisTemplate<String, String> redisTemplate;
+
+	@Autowired
+	private RemoteSecurityService remoteSecurityService;
 	
 	// 前置通知，在方法执行之前
 	@Before(value = "@annotation(Permission)")
@@ -46,16 +58,91 @@ public class PermissionAspect {
 
 		logger.info("======开始权限校验======");
 		// 用户token
-	    String token = (String) request.getHeader("token");
-	    token = ObjectUtils.isEmpty(token) ? (String) request.getHeader("X-Access-Token") : token;
-	    String product = (String) request.getHeader("product");
-	    String appKey = (String) request.getHeader("appKey");
-	    logger.info("用户token：" + token);
-	    RequestContext.setToken(token);
-	    RequestContext.setProduct(product);
-	    RequestContext.setAppKey(appKey);
-	    if (!TokenOperation.refresh(token)) {
-	      throw new PermissionException("登录信息失效，请重新登录");
-	    }
+		String token = (String) request.getHeader("token");
+		token = ObjectUtils.isEmpty(token) ? (String) request.getHeader("X-Access-Token") : token;
+		String product = (String) request.getHeader("product");
+		String appKey = (String) request.getHeader("appKey");
+		logger.info("用户token：" + token);
+		RequestContext.setToken(token);
+		RequestContext.setProduct(product);
+		RequestContext.setAppKey(appKey);
+		if (!TokenOperation.refresh(token)) {
+			throw new PermissionException("登录信息失效，请重新登录");
+		}
+
+		if (joinPoint.getSignature().getName().equals("saveCurCompany")) {
+			return;
+		}
+		FeignClientResult feignClientResult;
+		AgencyUserModel userModel=null;
+		try {
+			feignClientResult = Privilege.agencyUserClient.getme();
+			userModel = (AgencyUserModel) feignClientResult.getResult();
+
+		} catch (InnerInvokException e) {
+			e.printStackTrace();
+		}
+		String userId = null;
+		ReginParams regionParam = new ReginParams();
+		if(userModel != null){
+			userId = userModel.getUserId();
+			ReginParams reginParams = JSON.parseObject(redisTemplate.opsForValue().get(buildKey(userModel.getUserId(), token)), ReginParams.class);
+			if(reginParams == null && userModel.getCompanys().size() > 0){
+				CompanyModel companyModel = userModel.getCompanys().get(0);
+
+				List<DepartmentModel> deptList = remoteSecurityService.getDepartmentTreeByCompanyId(token, product, appKey, companyModel.getSequenceNbr().toString());
+				if(deptList.size() > 0){
+					CompanyBo companyBo = convertCompanyModelToBo(companyModel);
+					DepartmentBo departmentBo = convertDepartmentModelToBo(deptList.get(0));
+					regionParam.setCompany(companyBo);
+					regionParam.setDepartment(departmentBo);
+					regionParam.setRole(null);
+				}
+				redisTemplate.opsForValue().set(buildKey(userId, token), JSONObject.toJSONString(regionParam));
+			}
+		}
+	}
+
+	private DepartmentBo convertDepartmentModelToBo(DepartmentModel departmentModel){
+		DepartmentBo departmentBo = new DepartmentBo();
+		if(departmentModel != null) {
+			departmentBo.setCompanySeq(departmentModel.getCompanySeq().toString());
+			departmentBo.setDepartmentDesc(departmentModel.getDepartmentDesc());
+			departmentBo.setDepartmentName(departmentModel.getDepartmentName());
+			departmentBo.setLevel(departmentModel.getLevel());
+			departmentBo.setOrgCode(departmentModel.getOrgCode());
+			departmentBo.setParentId(departmentModel.getParentId().toString());
+			departmentBo.setDeptOrgCode(departmentModel.getDeptOrgCode().toString());
+			departmentBo.setSequenceNbr(departmentModel.getSequenceNbr().toString());
+		}
+		return departmentBo;
+	}
+
+	/**
+	 * 	Model 转 Bo
+ 	 */
+	private CompanyBo convertCompanyModelToBo(CompanyModel companyModel){
+
+		CompanyBo companyBo = new CompanyBo();
+		if(companyModel != null) {
+			companyBo.setAddress(companyModel.getAddress());
+			companyBo.setCompanyName(companyModel.getCompanyName());
+			companyBo.setCompanyOrgCode(Long.parseLong(companyModel.getCompanyOrgCode().toString()));
+			companyBo.setEmail(companyModel.getEmail());
+			companyBo.setLandlinePhone(companyModel.getLandlinePhone());
+			companyBo.setLongitude(companyModel.getLongitude());
+			companyBo.setLatitude(companyModel.getLatitude());
+			companyBo.setLevel(companyModel.getLevel());
+			companyBo.setOrgCode(companyModel.getOrgCode());
+			companyBo.setSequenceNbr(companyModel.getSequenceNbr().toString());
+			companyBo.setParentId(companyModel.getParentId().toString());
+		}
+		return companyBo;
+	}
+
+	//redi缓存选择的用户信息
+	private String buildKey(String userId, String token) {
+		return "region_" + userId + "_" + token;
+
 	}
 }