Zuul 修改请求头响应头（死磕）

Posted 2021-03-23 crazymakercircle

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Zuul 修改请求头响应头（死磕）相关的知识，希望对你有一定的参考价值。

疯狂创客圈 Java 高并发【亿级流量聊天室实战】实战系列【博客园总入口】

架构师成长+面试必备之高并发基础书籍【Netty Zookeeper Redis 高并发实战】

疯狂创客圈高并发环境视频，陆续上线：

Windows Redis 安装（带视频）
Linux Redis 安装（带视频）
Windows Zookeeper 安装（带视频）
Linux Zookeeper 安装（带视频）
RabbitMQ 离线安装（带视频）
Nacos 安装（带视频）
ElasticSearch 安装, 带视频**

小视频以及所需工具的百度网盘链接，请参见疯狂创客圈高并发社群博客

前言

由于 SpingSecurity + SpringSession 整合场景，涉及到修改Zuul请求头的问题。

故梳理一个比较全面的Zuul 修改请求头、响应头的文章。

1 SpingSecurity + SpringSession 整合场景二

本场景为统一网关 + 微服务场景：网关 SpingSecurity 实现 token认证，转成sessionID后，路由到单体微服务，单体微服务的 SpringSession 实现Session 共享。SessionID 保持在 token中。

技术图片

2 功能介绍：

Zuul网关 SpingSecurity 实现 token认证，从token中提取sessionID，放入header头部，再传给后台微服务

技术图片

3 参考文献：Zuul 使用filter 修改请求头和响应头

一：修改请求参数，参见博文

RequestContext ctx = RequestContext.getCurrentContext();
HttpServletRequest request = ctx.getRequest();
     Map<String, List<String>> requestQueryParams = ctx.getRequestQueryParams();

        if (requestQueryParams==null) {
            requestQueryParams=new HashMap<>();
        }

        //将要新增的参数添加进去,被调用的微服务可以直接 去取,就想普通的一样,框架会直接注入进去
        ArrayList<String> arrayList = new ArrayList<>();
        arrayList.add("1");
        requestQueryParams.put("test", arrayList);

        ctx.setRequestQueryParams(requestQueryParams);

二：修改请求头，参见博文

RequestContext ctx = RequestContext.getCurrentContext();  
ctx.addZuulRequestHeader("original_requestURL",request.getRequestURL().toString());

三：修改响应头，参见博文

RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletResponse response = ctx.getResponse();
        String info = response.getHeader("info");
        String info_size = response.getHeader("info_size");
        /**
         * 设置响应头，使请求变为文件下载
         */
        ctx.addZuulResponseHeader("Content-Type", "application/octet-stream");
        ctx.addZuulResponseHeader("Content-Disposition", "attachment;fileName=" + info);
        ctx.addZuulResponseHeader("Content-Length", ""+info_size);

4 ZuulFilter 修改请求头

1 对含有token头的request 进行拦截

2 springsecurity 负责将 sessionid 放在 request 的attribute中

3 filter 负责修改请求头，将session 加到向微服务的请求head中

package com.crazymaker.springcloud.cloud.center.zuul.config;

import com.crazymaker.springcloud.common.constants.SessionConstants;
import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import com.netflix.zuul.exception.ZuulException;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Component
@Slf4j
public class ModifyRequestHeaderFilter extends ZuulFilter {

    @Override
    public boolean shouldFilter() {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();
        String uri = request.getRequestURI();
        /**
         * 根据条件去判断是否需要路由，是否需要执行该过滤器
         */
        String token = request.getHeader(SessionConstants.AUTHORIZATION);
        log.info("token=" + token);
        if (!StringUtils.isEmpty(token) ) {
            return true;
        }
        return false;
    }

    /**
     * 修改请求头
     *
     * @return
     * @throws ZuulException
     */
    @Override
    public Object run() throws ZuulException {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();

        String sessionSeed = (String) request.getAttribute(SessionConstants.SESSION_SEED);
        log.info("sessionSeed=" + sessionSeed);

        ctx.addZuulRequestHeader(SessionConstants.SESSION_SEED,sessionSeed);

//        response.addHeader(SessionConstants.SESSION_SEED, sessionId);
        return null;

    }

    @Override
    public String filterType() {
        return FilterConstants.PRE_TYPE;
    }

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

}