将 HttpServletRequest 转发到不同的服务器

问题描述 投票:0回答:6

我在 Spring Servlet 中收到一个

HttpServletRequest
请求,我想将其按原样(即 GET 或 POST 内容)转发到不同的服务器。

使用 Spring 框架最好的方法是什么?

我需要获取所有信息并构建一个新的

HTTPUrlConnection
吗?或者有更简单的方法吗?

java spring http spring-mvc servlets
6个回答
27
投票

关于您是否应该以这种方式转发的讨论放在一边,我是这样做的:

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、身份验证等更容易。


15
投票

我也需要做同样的事情,并且在使用 Spring 控制器和 RestTemplate 进行一些非优化之后,我找到了一个更好的解决方案:Smiley 的 HTTP Proxy Servlet。好处是,它确实进行了 AS-IS 代理,就像 Apache 的

mod_proxy
一样,并且以流式传输的方式进行,无需在内存中缓存完整的请求/响应。

简单地说,您将一个新的 servlet 注册到您想要代理到另一台服务器的路径,并将目标主机作为初始化参数提供给该 servlet。如果您使用带有 web.xml 的传统 Web 应用程序,您可以按如下方式配置它:

<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,那就更简单了:您只需要创建一个类型为

ServletRegistrationBean
的 bean,并进行所需的配置:

@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 的代理 servlet 一段时间后,我们遇到了一些超时问题,它无法可靠地工作。从 Netflix 切换到 Zuul,之后没有任何问题。有关使用 Spring Boot 配置它的教程可以在此链接上找到。


13
投票

不幸的是,没有简单的方法可以做到这一点。基本上你必须重建请求,包括:

  • 正确的HTTP方法
  • 请求参数
  • 请求标头(
    HTTPUrlConnection
    不允许设置任意用户代理,“
    Java/1.*
    ”始终附加,您需要HttpClient)
  • 身体

这是一项艰巨的工作,更不用说它无法扩展,因为每个此类代理调用都会占用您计算机上的一个线程。

我的建议:使用原始套接字或 并在最低级别拦截 HTTP 协议,只需即时替换一些值(如

Host
标头)。您能否提供更多背景信息,为什么需要这个?


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

2
投票

如果你被迫使用spring,请检查rest模板方法交换将请求代理到第三方服务。

在这里您可以找到一个工作示例。


1
投票

使用Spring云网关

pom.xml

    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-gateway-mvc</artifactId>
    </dependency>  


<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-dependencies</artifactId>
            <version>2020.0.2</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

控制器

import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.gateway.mvc.ProxyExchange;
import org.springframework.http.ResponseEntity;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.util.UriComponents;
import org.springframework.web.util.UriComponentsBuilder;

@RestController
public class Proxy extends BaseController {

    private String prefix="/proxy";
    private String Base="localhost:8080/proxy";  //destination

    void setHeaders(ProxyExchange<?> proxy){
        proxy.header("header1", "val1"); //add additional headers
    }

    @GetMapping("/proxy/**")
    public ResponseEntity<?> proxyPath(@RequestParam MultiValueMap<String,String> allParams, ProxyExchange<?> proxy) throws Exception {
        String path = proxy.path(prefix); //extract sub path
        proxy.header("Cache-Control", "no-cache");
        setHeaders(proxy);

        UriComponents uriComponents =  UriComponentsBuilder.fromHttpUrl(Base + path).queryParams(allParams).build();
        return proxy.uri(uriComponents.toUri().toString()).get();
    }

    @PutMapping("/proxy/**")
    public ResponseEntity<?> proxyPathPut(ProxyExchange<?> proxy) throws Exception {
        String path = proxy.path(prefix);
        setHeaders(proxy);
        return proxy.uri(Base + path).put();
    }
© www.soinside.com 2019 - 2024. All rights reserved.