forked from kidgrow-microservices-platform
blame | history | patch | commit | commitdiff | ignore whitespace
zhaoxiaohao
2021-03-18 deb1110ca94cb0ac7bcdc51b4e8dd00407792a94 
 kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java


@@ -19,6 +19,7 @@


import org.springframework.security.authentication.AnonymousAuthenticationToken;


import org.springframework.security.core.Authentication;


import org.springframework.security.core.context.SecurityContextHolder;


import org.springframework.security.core.context.SecurityContextImpl;


import org.springframework.security.oauth2.provider.OAuth2Authentication;


import org.springframework.stereotype.Component;




@@ -64,7 +65,15 @@


    @Override


    public Object run() {


        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();




        if (authentication != null && !(authentication instanceof AnonymousAuthenticationToken)) {


                    RequestContext currentContext = RequestContext.getCurrentContext();


        //获取request对象


        HttpServletRequest request = currentContext.getRequest();


        SecurityContextImpl securityContextImpl = (SecurityContextImpl) request


                .getSession().getAttribute("SPRING_SECURITY_CONTEXT");




        authentication = securityContextImpl.getAuthentication();


            Object principal = authentication.getPrincipal();


            RequestContext ctx = RequestContext.getCurrentContext();


            HttpServletRequest req = ctx.getRequest();


@@ -90,22 +99,21 @@


                    doctorId=sysDoctors.get(0).getId();


                    departmentId=sysDoctors.get(0).getDepartmentId();


                    departmentName=sysDoctors.get(0).getDepartmentName();


                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_NAME_HEADER, URLEncoder.encode(sysDoctors.get(0).getHospitalName(),"UTF-8"));


                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_ID_HEADER,String.valueOf(departmentId));


                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_NAME_HEADER,URLEncoder.encode(departmentName,"UTF-8"));


                }


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


                ctx.addZuulRequestHeader(SecurityConstants.DOCTOR_ID_HEADER, String.valueOf(doctorId));


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




                ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME,URLEncoder.encode(str,"UTF-8"));//待完善


                ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME,URLEncoder.encode(str,"UTF-8"));


                ctx.addZuulRequestHeader(SecurityConstants.USER_HOSPITAL_ID_HEADER,hospital.toString());


                tenantId=hospital.toString();


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


                //将组织数据 存到header 里面


                organizations.sort((e1,e2)->e1.getOrgLevel().compareTo(e2.getOrgLevel()));


                if (organizations != null&&organizations.size()>0) {


                    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"));


                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_ID_HEADER,String.valueOf(departmentId));


                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_NAME_HEADER,(departmentName.equals("")?URLEncoder.encode(organizations.get(organizations.size()-1).getOrgName(),"UTF-8"):departmentName));


                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORGS_HEADER,JSON.toJSONString(organizations));


                }


                //将角色放到header