forked from kidgrow-microservices-platform
blame | history | patch | commit | commitdiff | ignore whitespace
zhaoxiaohao
2020-11-18 4130e1d3fc667fad3349e421e40a0ffa145c563c 
 kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java


@@ -18,7 +18,7 @@


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


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;




@@ -63,7 +63,16 @@


    @SneakyThrows


    @Override


    public Object run() {


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


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




        RequestContext currentContext = RequestContext.getCurrentContext();


        //获取request对象


        HttpServletRequest request = currentContext.getRequest();




        SecurityContextImpl securityContextImpl = (SecurityContextImpl) request


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


        Authentication authentication = securityContextImpl.getAuthentication();




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


            Object principal = authentication.getPrincipal();


            RequestContext ctx = RequestContext.getCurrentContext();


@@ -104,6 +113,14 @@


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


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


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


                    System.out.println(organizations.size());


                    System.out.println("111111111111111111111");


                    for (SysOrganization organization : organizations) {


                        if(organization!=null){


                            System.out.println(organization.getOrgLevel());


                        }


                    }


                    System.out.println("22222222222222");


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


                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_ID_HEADER,String.valueOf(organizations.get(0).getId()));


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