我在 Spring Servlet 中收到一个
HttpServletRequest
请求,我想将其按原样(即 GET 或 POST 内容)转发到不同的服务器。
使用 Spring 框架最好的方法是什么?
我需要获取所有信息并构建一个新的
HTTPUrlConnection
吗?或者有更简单的方法吗?
关于您是否应该以这种方式转发的讨论放在一边,我是这样做的:
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、身份验证等更容易。
我也需要做同样的事情,并且在使用 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 配置它的教程可以在此链接上找到。
不幸的是,没有简单的方法可以做到这一点。基本上你必须重建请求,包括:
HTTPUrlConnection
不允许设置任意用户代理,“Java/1.*
”始终附加,您需要HttpClient)这是一项艰巨的工作,更不用说它无法扩展,因为每个此类代理调用都会占用您计算机上的一个线程。
我的建议:使用原始套接字或 netty 并在最低级别拦截 HTTP 协议,只需即时替换一些值(如
Host
标头)。您能否提供更多背景信息,为什么需要这个?
@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());
}
}
如果你被迫使用spring,请检查rest模板方法交换将请求代理到第三方服务。
在这里您可以找到一个工作示例。
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();
}