我在我的 Spring Servlet 中收到一个HttpServletRequest
请求,我想将其按原样(即 GET 或 POST 内容)转发到不同的服务器。
使用 Spring Framework 最好的方法是什么?
我是否需要获取所有信息并构建一个新的HTTPUrlConnection
? 还是有更简单的方法?
我在我的 Spring Servlet 中收到一个HttpServletRequest
请求,我想将其按原样(即 GET 或 POST 内容)转发到不同的服务器。
使用 Spring Framework 最好的方法是什么?
我是否需要获取所有信息并构建一个新的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、auth 等更容易。
我也需要做同样的事情,在使用 Spring 控制器和 RestTemplate 进行一些非优化之后,我找到了一个更好的解决方案:Smiley 的 HTTP Proxy Servlet。好处是,它确实像 Apache 一样进行 AS-IS 代理,mod_proxy
并且它以流式方式进行,无需在内存中缓存完整的请求/响应。
简单地说,您将一个新的 servlet 注册到您要代理到另一台服务器的路径,并将此 servlet 指定为目标主机作为 init 参数。如果您使用带有 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>
或者,当然,您可以使用注释 config。
如果你使用的是 Spring Boot,那就更简单了:你只需要创建一个 type 的 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 的代理 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()));
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());
}
}
如果您被迫使用 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();
}