访问 PUT 或 POST 请求的原始正文

Question

我正在 Grails 中实现一个 RESTful API，并使用一个自定义身份验证方案，该方案涉及对请求正文进行签名（以类似于 Amazon 的 S3 身份验证方案的方式）。因此，要对请求进行身份验证，我需要访问原始 POST 或 PUT 正文内容来计算和验证数字签名。

我正在控制器的 beforeInterceptor 中进行身份验证。所以我希望在拦截器中可以访问 request.body 之类的东西，并且仍然能够在实际操作中使用 request.JSON。恐怕如果我使用 getInputStream 或 getReader（ServletRequest 提供的方法）读取拦截器中的正文，当我尝试通过 request.JSON 访问它时，正文将在操作中显示为空。

我正在从 Django 迁移到 Grails，一年前我在 Django 中遇到了完全相同的问题，但很快就修复了。 Django 提供了一个 request.raw_post_data 属性，您可以将其用于此目的。

最后，为了美观和 RESTful，我希望它适用于 POST 和 PUT 请求。

任何建议或指示将不胜感激。如果它不存在，我更喜欢关于如何实现优雅解决方案的指针，而不是快速和肮脏的黑客的想法。 =) 在 Django 中，我编辑了一些中间件请求处理程序以向请求添加一些属性。我对 Groovy 和 Grails 很陌生，所以我不知道该代码在哪里，但如果有必要我不介意这样做。

Answer 1

可以通过覆盖 Servlet 过滤器中的 HttpServletRequest 来实现。

您需要实现一个存储请求正文的 HttpServletRequestWrapper： src/java/grails/util/http/MultiReadHttpServletRequest.java

package grails.util.http;

import org.apache.commons.io.IOUtils;

import javax.servlet.http.HttpServletRequestWrapper;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.ServletInputStream;
import java.io.*;
import java.util.concurrent.atomic.AtomicBoolean;

public class MultiReadHttpServletRequest extends HttpServletRequestWrapper {

    private byte[] body;

    public MultiReadHttpServletRequest(HttpServletRequest httpServletRequest) {
        super(httpServletRequest);
        // Read the request body and save it as a byte array
        InputStream is = super.getInputStream();
        body = IOUtils.toByteArray(is);
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        return new ServletInputStreamImpl(new ByteArrayInputStream(body));
    }

    @Override
    public BufferedReader getReader() throws IOException {
        String enc = getCharacterEncoding();
        if(enc == null) enc = "UTF-8";
        return new BufferedReader(new InputStreamReader(getInputStream(), enc));
    }

    private class ServletInputStreamImpl extends ServletInputStream {

        private InputStream is;

        public ServletInputStreamImpl(InputStream is) {
            this.is = is;
        }

        public int read() throws IOException {
            return is.read();
        }

        public boolean markSupported() {
            return false;
        }

        public synchronized void mark(int i) {
            throw new RuntimeException(new IOException("mark/reset not supported"));
        }

        public synchronized void reset() throws IOException {
            throw new IOException("mark/reset not supported");
        }
    }

}

一个覆盖当前 servletRequest 的 Servlet 过滤器：src/java/grails/util/http/MultiReadServletFilter.java

package grails.util.http;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.Set;
import java.util.TreeSet;

public class MultiReadServletFilter implements Filter {

    private static final Set<String> MULTI_READ_HTTP_METHODS = new TreeSet<String>(String.CASE_INSENSITIVE_ORDER) {{
        // Enable Multi-Read for PUT and POST requests
            add("PUT");
            add("POST");
    }};

    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        if(servletRequest instanceof HttpServletRequest) {
            HttpServletRequest request = (HttpServletRequest) servletRequest;
            // Check wether the current request needs to be able to support the body to be read multiple times
            if(MULTI_READ_HTTP_METHODS.contains(request.getMethod())) {
                // Override current HttpServletRequest with custom implementation
                filterChain.doFilter(new MultiReadHttpServletRequest(request), servletResponse);
                return;
            }
        }
        filterChain.doFilter(servletRequest, servletResponse);
    }

    public void init(FilterConfig filterConfig) throws ServletException {
    }

    public void destroy() {
    }
}

然后你需要运行 grails install-templates 并编辑 src/templates/war 中的 web.xml 并将其添加到 charEncodingFilter 定义之后：

<filter>
    <filter-name>multireadFilter</filter-name>
    <filter-class>grails.util.http.MultiReadServletFilter</filter-class>
</filter>

<filter-mapping>
    <filter-name>multireadFilter</filter-name>
    <url-pattern>/*</url-pattern>
</filter-mapping>

您应该可以根据需要随时拨打request.inputStream。

我没有测试过这个具体的代码/程序，但我过去做过类似的事情，所以它应该可以工作 ;-)

注意：请注意，巨大的请求可能会杀死您的应用程序（OutOfMemory...）

Answer 2

这里可以看到

http://jira.codehaus.org/browse/GRAILS-2017

只需关闭 grails 对 XML 的自动处理，就可以在控制器中访问文本。像这样

class EventsController {   

static allowedMethods = [add:'POST']

def add = {
    log.info("Got request " + request.reader.text)      
    render "OK"
}}

最好， 安德斯

Answer 3

似乎能够继续访问 POST 请求的流和请求参数的唯一方法是编写一个覆盖流读取以及参数访问的包装器。这是一个很好的例子：

Modify HttpServletRequest body