forked from kidgrow-microservices-platform

zhaoxiaohao
2020-11-18 aae97233d9af913104bc7c52a799f6a695c11ec5
添加日志输出
2 files modified
105 ■■■■■ changed files
kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/OrganizationFilter.java 87 ●●●●● patch | view | raw | blame | history
kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java 18 ●●●● patch | view | raw | blame | history
kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/OrganizationFilter.java
@@ -17,6 +17,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;
@@ -64,15 +65,16 @@
    @SneakyThrows
    @Override
    public Object run() {
        //Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        RequestContext currentContext = RequestContext.getCurrentContext();
        //获取request对象
        HttpServletRequest request = currentContext.getRequest();
        //security会把一个SecurityContextImpl对象存储到session中,此对象中有当前用户的各种资料
        SecurityContextImpl securityContextImpl = (SecurityContextImpl) request
                .getSession().getAttribute("SPRING_SECURITY_CONTEXT");
        Authentication authentication = securityContextImpl.getAuthentication();
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication != null && !(authentication instanceof AnonymousAuthenticationToken)) {
            RequestContext currentContext = RequestContext.getCurrentContext();
//            获取request对象
            HttpServletRequest request = currentContext.getRequest();
            //security会把一个SecurityContextImpl对象存储到session中,此对象中有当前用户的各种资料
            SecurityContextImpl securityContextImpl = (SecurityContextImpl) request
                    .getSession().getAttribute("SPRING_SECURITY_CONTEXT");
            authentication = securityContextImpl.getAuthentication();
            Object principal = authentication.getPrincipal();
            RequestContext ctx = RequestContext.getCurrentContext();
            if (principal instanceof SysUser) {
@@ -80,18 +82,17 @@
                OAuth2Authentication oauth2Authentication = (OAuth2Authentication) authentication;
                String clientId = oauth2Authentication.getOAuth2Request().getClientId();
//                if (CLIENTID.equals(clientId)) {
                    SysUser user = (SysUser) authentication.getPrincipal();
                    /**
                     * 将组织中为空的拦截
                     */
                    List<SysUserOrg> sysUserOrgs = getSysUserOrg(user.getId());
                    if (sysUserOrgs == null || sysUserOrgs.isEmpty()) {
                        ctx.setSendZuulResponse(false);
                        ctx.addZuulResponseHeader("Content-Type","application/json;charset=UTF-8");
                SysUser user = (SysUser) authentication.getPrincipal();
                /**
                 * 将组织中为空的拦截
                 */
                List<SysUserOrg> sysUserOrgs = this.getSysUserOrg(user.getId());
                if (sysUserOrgs == null || sysUserOrgs.isEmpty()) {
                    ctx.setSendZuulResponse(false);
                    ctx.addZuulResponseHeader("Content-Type", "application/json;charset=UTF-8");
//                        String str = new String("您的组织已经被禁用,请联系管理员".getBytes("utf-8"), "utf-8");
                        ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用,请联系管理员")));
                    }
                    else {
                    ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用1,请联系管理员")));
                } else {
//                        List<Long> collect = sysUserOrgs.stream().map(e -> e.getOrgId()).collect(Collectors.toList());
//                        List<SysOrganization> sysOrganizations = getSysOrganization();
//                        List<Long> orgIds = sysOrganizations.stream().filter(e -> e.getEnabled() == true && collect.contains(e.getId())).map(e -> e.getId()).collect(Collectors.toList());
@@ -99,28 +100,34 @@
//                            ctx.setSendZuulResponse(false);
//                            ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用,请联系管理员")));
//                        }
                        //根据fegin客户端查询状态
                        Map<String,Object> map;
                        for (SysUserOrg sysUserOrg : sysUserOrgs) {
                            map=new HashMap<>();
                            map.put("id",sysUserOrg.getOrgId());
                            List<SysOrganization> sysOrganizations = JSON.parseArray(JSON.toJSONString(sysOrganizationService.getListByMap(map).getData()), SysOrganization.class);
                            if(sysOrganizations==null||sysOrganizations.size()<=0){
                                ctx.setSendZuulResponse(false);
                                ctx.addZuulResponseHeader("Content-Type","application/json;charset=UTF-8");
                                ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用,请联系管理员")));
                            }else {
                                for (SysOrganization sysOrganization : sysOrganizations) {
                                    if(!sysOrganization.getEnabled()||sysOrganization.getIsDel()){
                                        ctx.setSendZuulResponse(false);
                                        ctx.addZuulResponseHeader("Content-Type","application/json;charset=UTF-8");
                                        ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用,请联系管理员")));
                                    }
                    //根据fegin客户端查询状态
                    Map<String, Object> map;
                    for (SysUserOrg sysUserOrg : sysUserOrgs) {
                        map = new HashMap<>();
                        map.put("id", sysUserOrg.getOrgId());
                        List<SysOrganization> sysOrganizations = JSON.parseArray(JSON.toJSONString(sysOrganizationService.getListByMap(map).getData()), SysOrganization.class);
                        System.out.println("5555555555555555");
                        System.out.println(sysOrganizations.size());
                        for (SysOrganization sysOrganization : sysOrganizations) {
                            System.out.println(sysOrganization.getOrgName());
                        }
                        System.out.println("66666666666");
                        if (sysOrganizations == null || sysOrganizations.size() <= 0) {
                            ctx.setSendZuulResponse(false);
                            ctx.addZuulResponseHeader("Content-Type", "application/json;charset=UTF-8");
                            ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用2,请联系管理员")));
                        } else {
                            for (SysOrganization sysOrganization : sysOrganizations) {
                                if (!sysOrganization.getEnabled() || sysOrganization.getIsDel()) {
                                    ctx.setSendZuulResponse(false);
                                    ctx.addZuulResponseHeader("Content-Type", "application/json;charset=UTF-8");
                                    ctx.setResponseBody(JSON.toJSONString(ResultBody.fail(1000, "您的组织已经被禁用3,请联系管理员")));
                                }
                            }
                        }
                    }
                }
//                }
            }
        }
@@ -136,6 +143,12 @@
    public List<SysUserOrg> getSysUserOrg(Long str) {
        //获取组织 Redis中获取
        List<SysUserOrg> sysUserOrgs = JSON.parseArray(JSON.toJSONString(redisUtils.hget(RedisConstant.USER_ORGANIZATION, str.toString())), SysUserOrg.class);
        System.out.println("3333333333333333333333");
        System.out.println(sysUserOrgs.size());
        for (SysUserOrg sysUserOrg : sysUserOrgs) {
            System.out.println(sysUserOrg.getFromLevel());
        }
        System.out.println("444444444444444444");
        if (sysUserOrgs == null || sysUserOrgs.size() <= 0) {
            //在数据库查询
            Map<String, Object> map = new HashMap<>();
kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java
@@ -18,6 +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,17 +64,16 @@
    @SneakyThrows
    @Override
    public Object run() {
       // 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();
        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();