zuulGateway 通过filter统一修改返回值

Love The Way You Lie 2022-05-27 00:05 307阅读 0赞

使用spring cloud有时候我们给客户端返回内容的时候，往往需要添加一些额外的东西。比如加密，多添加一个返回值等等。当然可以在方法里面处理，但如果方法很多，需要统一处理的，就很不方便了，这时候可以通过zuulGateway的filter来统一处理。

直接看代码，很简单：

import java.io.InputStream;
    import java.nio.charset.Charset;
    
    import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
    import org.springframework.util.StreamUtils;
    import org.springframework.util.StringUtils;
    import com.netflix.zuul.ZuulFilter;
    import com.netflix.zuul.context.RequestContext;
    import com.poly.zuul.common.Constants;
    import com.poly.zuul.enums.ServiceId;
    import com.poly.zuul.utils.AESJSEncryptUtils;
    import com.poly.zuul.utils.RsaEncryptUtil;
    
    
    /**
     * 返回值输出过滤器，这里用来处理返回值
     * 
     * @Title: ResponseFilter
     * @Description:
     * @author kokJuis
     * @date 上午9:52:42
     */
    public class ResponseFilter extends ZuulFilter
    {
    
        @Override
        public Object run()
        {
    
            RequestContext context = RequestContext.getCurrentContext();
    
            try
            {
                // 获取返回值内容，加以处理
                InputStream stream = context.getResponseDataStream();
                String body = StreamUtils.copyToString(stream, Charset.forName("UTF-8"));
                String returnStr = "";
    
    			
    			//你的处理逻辑，加密，添加新的返回值等等.....
    			
    
                // 内容重新写入
                context.setResponseBody(returnStr);
            }
            catch (Exception e)
            {
                e.printStackTrace();
            }
    
            return null;
        }
    
        @Override
        public boolean shouldFilter()
        {
            RequestContext ctx = RequestContext.getCurrentContext();
    
            String requestURI = String.valueOf(ctx.get("requestURI"));
    
            if (requestURI.contains(Constants.alipay))
            {
    			//不需要处理的URL请求，返回false
                return false;
            }
    
    
            return true;
        }
    
        @Override
        public int filterOrder()
        {
            return FilterConstants.SEND_RESPONSE_FILTER_ORDER - 3;
        }
    
        @Override
        public String filterType()
        {
            return FilterConstants.POST_TYPE;// 在请求被处理之后，会进入该过滤器
        }
    
    }

然后在@Configuration里面注册一下这个filter就可以了

@Bean
        public ResponseFilter responseFilter()
        {
            return new ResponseFilter();
        }