From d95667d16f96dfbfa9ecf34d9600a453e89d2071 Mon Sep 17 00:00:00 2001 From: 侯瑞军 <411269194@kidgrow.com> Date: Mon, 20 Apr 2020 09:47:56 +0800 Subject: [PATCH] 修改业务日志的save控制器 --- kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java | 25 +++++++++++++++++++------ 1 files changed, 19 insertions(+), 6 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 dade34c..287b8d5 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 @@ -3,12 +3,14 @@ import cn.hutool.core.collection.CollectionUtil; import com.kidgrow.common.constant.CommonConstant; import com.kidgrow.common.constant.SecurityConstants; +import com.kidgrow.common.model.ResultBody; import com.kidgrow.common.model.SysOrganization; import com.kidgrow.common.model.SysUser; import com.kidgrow.common.utils.AddrUtil; import com.netflix.zuul.ZuulFilter; import com.netflix.zuul.context.RequestContext; import eu.bitwalker.useragentutils.UserAgent; +import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants; import org.springframework.security.authentication.AnonymousAuthenticationToken; @@ -18,6 +20,7 @@ import org.springframework.stereotype.Component; import javax.servlet.http.HttpServletRequest; +import java.net.URLEncoder; import java.util.List; import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.FORM_BODY_WRAPPER_FILTER_ORDER; @@ -48,6 +51,7 @@ return true; } + @SneakyThrows @Override public Object run() { Authentication authentication = SecurityContextHolder.getContext().getAuthentication(); @@ -62,14 +66,23 @@ 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.size()==2){ - ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_ID_HEADER,String.valueOf(organizations.get(0).getId())); - ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_NAME_HEADER,organizations.get(0).getOrgName()); - ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_ID_HEADER,String.valueOf(organizations.get(1).getId())); - ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_NAME_HEADER,organizations.get(1).getOrgName()); - tenantId=String.valueOf(organizations.get(0).getId()); + 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("您组织部门或者科室,已经被禁用,请联系管理员"); + } } // log.error("用户ID写入Header成功,用户ID为:"+String.valueOf(user.getId())); } -- Gitblit v1.8.0