Zuul 如何在pre Filter内为request setAttribute

发布于 2022-09-07 03:58:04 字数 3861 浏览 6 评论 0

我希望能在zuul的filter对request进行setAttribute("key",value)操作,然后在下游的服务能够通过request.getAttribute("key")的方式获取到数据;但结果未如预期发生,在filter内为request添加了attribute并未被传递到下游服务中;

/**
 * Created by zhaoyuening on 2018/1/26.
 */
@Component
public class TokenCheckFilter extends ZuulFilter {

    private final String JWT_HEADER_NAME = "token";
    private final Logger logger = LoggerFactory.getLogger(TokenCheckFilter.class);

    @Value("${token.filter}")
    private boolean IS_FILTER;

    @Autowired
    private RoleResourceManager roleResourceManager;

    @Override
    public String filterType() {
        return "pre";
    }

    @Override
    public int filterOrder() {
        return 5;
    }

    @Override
    public boolean shouldFilter() {
        if (IS_FILTER){
            logger.info("开启token过滤");
        }else{
            logger.info("未开启token过滤");
        }
        return IS_FILTER;
    }

    //超级管理员角色
    @Value("${root.role}")
    private String ROOT_ROLE;


    @Override
    //TODO 目前每次修改鉴权权限等信息 需要重启
    //后续版本修改后采用mq 自动更新缓存
    public Object run() {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();
        String url = request.getRequestURI();
        String method = request.getMethod();

        //检测资源状态 是否为白名单 或 被禁用
        // 0 白名单 1 被监控权限 2 禁用
        Integer resourceStatus = roleResourceManager.checkStatusForResource(method, url);
        if (resourceStatus == null || resourceStatus == -1||resourceStatus == 2){
            createResponce("request fail!",ctx);
            return null;
        } else if (resourceStatus == 0){
            //白名单 放行
            return null;
        }

        //使用jwtBean解码装载jwt信息
        CodecUtils.JwtBean jwtBean = CodecUtils.JwtBean.getJwtBean(request.getHeader(JWT_HEADER_NAME));
        //将信息存储到request
        addRequestAttribute(jwtBean,request);
        if (jwtBean != null){
            //jwt 用户角色信息
            List<String> roleIdList = (List<String>) jwtBean.getPlayload(JwtFieldEnum.ROLES.getField());
            //指定id为1 的角色为超级管理员
            if (roleIdList != null && roleIdList.contains(ROOT_ROLE)){
                return null;
            }

            if (roleIdList != null && roleResourceManager.checkAuthority(method, url, roleIdList)) {
                //该检验通过 该用户具备访问资源的权限
                return null;
            }
        }
        //游客身份
        //查看是否为白名单Resource
        createResponce("request fail",ctx);
        return null;
    }

    /**
     * 请求失败 调用
     * @param msg
     * @param ctx
     */
    private void createResponce(String msg,RequestContext ctx){
        // 过滤该请求,不对其进行路由
        ctx.setSendZuulResponse(false);
        // 返回无权限错误码
        ctx.setResponseStatusCode(50008);
        // 返回错误内容
        ctx.setResponseBody(msg);
        ctx.set("isSuccess", false);
    }

    /**
     * 从jwtBean获取相应信息,并加装到request内
     * @param jwtBean
     * @param request
     */
    private void addRequestAttribute(CodecUtils.JwtBean jwtBean,HttpServletRequest request) {
        if (jwtBean == null) {
            return;
        }
        //roles
        Object roles = jwtBean.getPlayload(JwtFieldEnum.ROLES.getField());
        Object username = jwtBean.getPlayload(JwtFieldEnum.USERNAME.getField());
        Object admin = jwtBean.getPlayload(JwtFieldEnum.ADMIN.getField());
        //save info to request
        request.setAttribute(RequestAttributeKeyEnum.ACCOUNT_ROLES.getKey(),roles);
        request.setAttribute(RequestAttributeKeyEnum.ACCOUNT_USERNAME.getKey(),username);
        request.setAttribute(RequestAttributeKeyEnum.ACCOUNT_ADMIN.getKey(),admin);

        RequestContext.getCurrentContext().setRequest(request);
    }
}

这是我的自定义pre filter 对request进行操作发生在最后的addRequestAttribute方法内,应该如何操作让下游接收到我在filter内对request添加的数据;

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(2

潦草背影 2022-09-14 03:58:04

RequestContext#addZuulRequestHeader,传递的内容,放在头部。这样下游应用,可以从头部获取。

烟沫凡尘 2022-09-14 03:58:04

问题解决了 setAttribute 只是以request对象为载体在单个程序内传输的数据,并在http规范中,不会被转发到其他地方。只能采用requestHeader的方式来传输

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文