Переслать HttpServletRequest на другой сервер

Я получил запрос HttpServletRequest в своем Spring Servlet, который я хотел бы переслать как есть (т.е. содержимое GET или POST) на другой сервер.

Что было бы лучшим способом сделать это с помощью Spring Framework?

Нужно ли мне HTTPUrlConnection всю информацию и создавать новый HTTPUrlConnection? Или есть более простой способ?

Ответы

Ответ 1

К сожалению, нет простого способа сделать это. В основном вам придется восстановить запрос, в том числе:

  • правильный метод HTTP
  • параметры запроса
  • заголовки запросов (HTTPUrlConnection не позволяет установить произвольный пользовательский агент, всегда добавляется "Java/1.*", вам нужен HttpClient)
  • Тело

Это большая работа, не говоря уже о том, что она не будет масштабироваться, поскольку каждый такой прокси-вызов будет занимать один поток на вашем компьютере.

Мой совет: используйте сырые сокеты или netty и перехватить HTTP-протокол на самом низком уровне, просто заменив некоторые значения (например, Host header) на лету. Можете ли вы предоставить больше контекста, зачем вам это нужно?

Ответ 2

Обсуждения о том, следует ли вам переадресовывать этот путь в сторону, вот как я это сделал:

package com.example.servlets;

import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Enumeration;

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

import com.example.servlets.GlobalConstants;

@SuppressWarnings("serial")
public class ForwardServlet extends HttpServlet {

    @Override
    public void doGet(HttpServletRequest req, HttpServletResponse resp) {
        forwardRequest("GET", req, resp);
    }

    @Override
    public void doPost(HttpServletRequest req, HttpServletResponse resp) {
        forwardRequest("POST", req, resp);
    }

    private void forwardRequest(String method, HttpServletRequest req, HttpServletResponse resp) {
        final boolean hasoutbody = (method.equals("POST"));

        try {
            final URL url = new URL(GlobalConstants.CLIENT_BACKEND_HTTPS  // no trailing slash
                    + req.getRequestURI()
                    + (req.getQueryString() != null ? "?" + req.getQueryString() : ""));
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            conn.setRequestMethod(method);

            final Enumeration<String> headers = req.getHeaderNames();
            while (headers.hasMoreElements()) {
                final String header = headers.nextElement();
                final Enumeration<String> values = req.getHeaders(header);
                while (values.hasMoreElements()) {
                    final String value = values.nextElement();
                    conn.addRequestProperty(header, value);
                }
            }

          //conn.setFollowRedirects(false);  // throws AccessDenied exception
            conn.setUseCaches(false);
            conn.setDoInput(true);
            conn.setDoOutput(hasoutbody);
            conn.connect();

            final byte[] buffer = new byte[16384];
            while (hasoutbody) {
                final int read = req.getInputStream().read(buffer);
                if (read <= 0) break;
                conn.getOutputStream().write(buffer, 0, read);
            }

            resp.setStatus(conn.getResponseCode());
            for (int i = 0; ; ++i) {
                final String header = conn.getHeaderFieldKey(i);
                if (header == null) break;
                final String value = conn.getHeaderField(i);
                resp.setHeader(header, value);
            }

            while (true) {
                final int read = conn.getInputStream().read(buffer);
                if (read <= 0) break;
                resp.getOutputStream().write(buffer, 0, read);
            }
        } catch (Exception e) {
            e.printStackTrace();
            // pass
        }
    }
}

Очевидно, что это могло бы использовать небольшую работу по обработке ошибок и тому подобное, но оно было функциональным. Однако я перестал использовать его, потому что в моем случае было проще делать вызовы непосредственно в CLIENT_BACKEND, чем обрабатывать файлы cookie, auth и т.д. Через два разных домена.

Ответ 3

Мне также нужно было сделать то же самое, и после некоторого неоптимального использования Spring-контроллеров и RestTemplate я нашел лучшее решение: Smiley HTTP Proxy Servlet. Преимущество в том, что он действительно выполняет проксирование AS-IS, как и Apache mod_proxy, и делает это потоковым способом, не кэшируя полный запрос/ответ в памяти.

Просто вы регистрируете новый сервлет по пути, который вы хотите передать на другой сервер, и назначаете этому сервлету целевой хост в качестве параметра init. Если вы используете традиционное веб-приложение с web.xml, вы можете настроить его следующим образом:

<servlet>
    <servlet-name>proxy</servlet-name>
    <servlet-class>org.mitre.dsmiley.httpproxy.ProxyServlet</servlet-class>
    <init-param>
      <param-name>targetUri</param-name>
      <param-value>http://target.uri/target.path</param-value>
    </init-param>
</servlet>
<servlet-mapping>
  <servlet-name>proxy</servlet-name>
  <url-pattern>/mapping-path/*</url-pattern>
</servlet-mapping>

или, конечно, вы можете пойти с конфигурацией аннотации.

Если вы используете Spring Boot, это еще проще: вам нужно только создать bean-компонент типа ServletRegistrationBean с необходимой конфигурацией:

@Bean
public ServletRegistrationBean proxyServletRegistrationBean() {
    ServletRegistrationBean bean = new ServletRegistrationBean(
            new ProxyServlet(), "/mapping-path/*");
    bean.addInitParameter("targetUri", "http://target.uri/target.path");
    return bean;
}

Таким образом, вы также можете использовать свойства Spring, доступные в данной среде.

Вы даже можете расширить класс ProxyServlet и переопределить его методы для настройки заголовков запросов/ответов и т.д., В случае необходимости.

Обновление: после использования прокси-сервлета Smiley в течение некоторого времени у нас возникли проблемы с тайм-аутом, он не работал надежно. Перешел на Zuul с Netflix, проблем не было после этого. Учебное пособие по настройке Spring Boot можно найти по этой ссылке.

Ответ 4

@RequestMapping(value = "/**")
public ResponseEntity route(HttpServletRequest request) throws IOException {
    String body = IOUtils.toString(request.getInputStream(), Charset.forName(request.getCharacterEncoding()));
    try {
        ResponseEntity<Object> exchange = restTemplate.exchange(firstUrl + request.getRequestURI(),
                HttpMethod.valueOf(request.getMethod()),
                new HttpEntity<>(body),
                Object.class,
                request.getParameterMap());
        return exchange;
    } catch (final HttpClientErrorException e) {
        return new ResponseEntity<>(e.getResponseBodyAsByteArray(), e.getResponseHeaders(), e.getStatusCode());
    }
}