From 6ec40d5a0c145ccffd1a28841f1c732f764e3e44 Mon Sep 17 00:00:00 2001 From: zhaoxiaohao <913652501@qq.com> Date: Mon, 20 Apr 2020 17:32:00 +0800 Subject: [PATCH] 添加 部门的拦截的功能 --- kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java | 40 +++++++++++++++------------------------- 1 files changed, 15 insertions(+), 25 deletions(-) diff --git a/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java b/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java index 287b8d5..79a5510 100644 --- a/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java +++ b/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java @@ -1,6 +1,7 @@ package com.kidgrow.zuul.filter; import cn.hutool.core.collection.CollectionUtil; +import com.alibaba.fastjson.JSON; import com.kidgrow.common.constant.CommonConstant; import com.kidgrow.common.constant.SecurityConstants; import com.kidgrow.common.model.ResultBody; @@ -20,15 +21,17 @@ import org.springframework.stereotype.Component; import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; import java.net.URLEncoder; import java.util.List; import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.FORM_BODY_WRAPPER_FILTER_ORDER; +import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.PRE_DECORATION_FILTER_ORDER; /** * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br> * - * @Description: 将认证用户的相关信息放入header中, 后端服务可以直接读取使用<br> + * @Description: 将认证用户的相关信息放入header中, 后端服务可以直接读取使用 包含了----组织的拦截---<br> * @Project: <br> * @CreateDate: Created in 2020/2/21 10:12 <br> * @Author: <a href="4345453@kidgrow.com">liuke</a> @@ -60,47 +63,34 @@ RequestContext ctx = RequestContext.getCurrentContext(); HttpServletRequest req = ctx.getRequest(); UserAgent userAgent = UserAgent.parseUserAgentString(req.getHeader("User-Agent")); - String tenantId=""; + String tenantId = ""; //客户端模式只返回一个clientId if (principal instanceof SysUser) { - SysUser user = (SysUser)authentication.getPrincipal(); + SysUser user = (SysUser) authentication.getPrincipal(); ctx.addZuulRequestHeader(SecurityConstants.USER_ID_HEADER, String.valueOf(user.getId())); ctx.addZuulRequestHeader(SecurityConstants.USER_HEADER, user.getUsername()); ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME, "");//待完善 - List<SysOrganization> organizations = (List<SysOrganization>)user.getOrganizations(); - //如果有组织架构 - if(organizations!=null && organizations.size()==2){ - if(organizations.get(0)!=null){ - ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_ID_HEADER,String.valueOf(organizations.get(0).getId())); - ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_NAME_HEADER, URLEncoder.encode(organizations.get(0).getOrgName(),"UTF-8")); - tenantId=String.valueOf(organizations.get(0).getId()); - }else{ - return ResultBody.failed("您的组织,已经被禁用,请联系管理员"); - } - if(organizations.get(1)!=null){ - ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_ID_HEADER,String.valueOf(organizations.get(1).getId())); - ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_NAME_HEADER,URLEncoder.encode(organizations.get(1).getOrgName(),"UTF-8")); - }else { - return ResultBody.failed("您组织部门或者科室,已经被禁用,请联系管理员"); - } - } + List<SysOrganization> organizations = (List<SysOrganization>) user.getOrganizations(); + //如果没有有组织架构 + //将组织数据 存到header 里面 + ctx.addZuulRequestHeader(SecurityConstants.USER_ORGS_HEADER, JSON.toJSONString(organizations)); // log.error("用户ID写入Header成功,用户ID为:"+String.valueOf(user.getId())); } - OAuth2Authentication oauth2Authentication = (OAuth2Authentication)authentication; + OAuth2Authentication oauth2Authentication = (OAuth2Authentication) authentication; String clientId = oauth2Authentication.getOAuth2Request().getClientId(); //保存租户id,租户id根据业务尽进行替换 - switch (clientId){ + switch (clientId) { case "hospital": - tenantId=tenantId; + tenantId = tenantId; break; case "webApp": - tenantId="webApp"; + tenantId = "webApp"; break; default: - tenantId=clientId; + tenantId = clientId; break; } ctx.addZuulRequestHeader(CommonConstant.USER_AGENT_BROWSER, userAgent.getBrowser().toString()); -- Gitblit v1.8.0