forked from kidgrow-microservices-platform
blame | history | patch | commit | commitdiff | ignore whitespace
侯瑞军
2020-04-10 6eb8cb8d7b18732394b0d0ee1fcea23173cfc9f3 
 kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java


@@ -3,6 +3,7 @@


import cn.hutool.core.collection.CollectionUtil;


import com.kidgrow.common.constant.CommonConstant;


import com.kidgrow.common.constant.SecurityConstants;


import com.kidgrow.common.model.SysOrganization;


import com.kidgrow.common.model.SysUser;


import com.kidgrow.common.utils.AddrUtil;


import com.netflix.zuul.ZuulFilter;


@@ -17,6 +18,7 @@


import org.springframework.stereotype.Component;




import javax.servlet.http.HttpServletRequest;


import java.util.List;




import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.FORM_BODY_WRAPPER_FILTER_ORDER;




@@ -60,7 +62,16 @@


                SysUser user = (SysUser)authentication.getPrincipal();


                ctx.addZuulRequestHeader(SecurityConstants.USER_ID_HEADER, String.valueOf(user.getId()));


                ctx.addZuulRequestHeader(SecurityConstants.USER_HEADER, user.getUsername());


                tenantId=String.valueOf(user.getId());


                ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME, "");//待完善


                List<SysOrganization> organizations = (List<SysOrganization>)user.getOrganizations();


                //如果有组织架构


                if(organizations!=null && 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());


                }


//                log.error("用户ID写入Header成功，用户ID为："+String.valueOf(user.getId()));


            }