向Netflix Zuul前置过滤器的请求正文添加新字段

问题描述

我正在尝试在Zuul前置过滤器中向请求的正文添加新字段.

I'm trying to add a new field to request's body, in a Zuul Pre-filter.

我正在使用来自此处，我的过滤器的实现与该示例中的UppercaseRequestEntityFilter非常相似.

I'm using one of the Neflix's Zuul sample projects from here, and my filter's implementation is very similar to UppercaseRequestEntityFilter from this sample.

我能够应用大写形式的转换，甚至可以完全修改请求，唯一的不便之处是，我无法修改长度超过原始请求长度的正文请求内容机构的要求.

I was able to apply a transformation such as uppercase, or even to completely modify the request, the only inconvenient is that I'm not able to modify the content of body's request that has a length more than the original length of the body's request.

这是我的过滤器的实现:

This is my filter's implementation:

@Component
public class MyRequestEntityFilter extends ZuulFilter {
    public String filterType() {
        return "pre";
    }

    public int filterOrder() {
        return 10;
    }

    public boolean shouldFilter() {
        RequestContext context = getCurrentContext();
        return true;
    }

    public Object run() {
        try {
            RequestContext context = getCurrentContext();
            InputStream in = (InputStream) context.get("requestEntity");
            if (in == null) {
                in = context.getRequest().getInputStream();
            }

            String body = StreamUtils.copyToString(in, Charset.forName("UTF-8"));

            body = body.replaceFirst("qqq", "qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq");

            // body = body.toUpperCase();

            context.set("requestEntity", new ServletInputStreamWrapper(body.getBytes("UTF-8")));
        }
        catch (IOException e) {
            rethrowRuntimeException(e);
        }
        return null;
    }
} 

这是我正在做的请求:

这是我收到的回复:

推荐答案

使用PrefixRequestEntityFilter的实现，我能够从 sample-zuul-examples :

I was able to obtain what I wanted, using the implementation of PrefixRequestEntityFilter, from sample-zuul-examples:

@Component
public class MyRequestEntityFilter extends ZuulFilter {
    public String filterType() {
        return "pre";
    }

    public int filterOrder() {
        return 10;
    }

    public boolean shouldFilter() {
        RequestContext context = getCurrentContext();
        return true;
    }

    public Object run() {
        try {
            RequestContext context = getCurrentContext();
            InputStream in = (InputStream) context.get("requestEntity");
            if (in == null) {
                in = context.getRequest().getInputStream();
            }

            String body = StreamUtils.copyToString(in, Charset.forName("UTF-8"));

            body = body.replaceFirst("qqq", "qqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqqq");

            byte[] bytes = body.getBytes("UTF-8");

            context.setRequest(new HttpServletRequestWrapper(getCurrentContext().getRequest()) {
                @Override
                public ServletInputStream getInputStream() throws IOException {
                    return new ServletInputStreamWrapper(bytes);
                }

                @Override
                public int getContentLength() {
                    return bytes.length;
                }

                @Override
                public long getContentLengthLong() {
                    return bytes.length;
                }
            });

        }
        catch (IOException e) {
            rethrowRuntimeException(e);
        }
        return null;
    }
}

这篇关于向Netflix Zuul前置过滤器的请求正文添加新字段的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！