2012-08-26 37 views
17

我在我的Spring Servlet中获得了一个HttpServletRequest请求,我希望将AS-IS(即POST或GET内容)转发到不同的服务器。转发HttpServletRequest到不同的服务器

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

我是否需要获取所有信息并构建新的HTTPUrlConnection?还是有更简单的方法?

回答

4

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

  • 正确的HTTP方法
  • 请求参数
  • 请求报头(HTTPUrlConnection不允许任意设置用户代理“ Java/1.*”总是附加,你需要的HttpClient)

这是一个很大的工作,更何况,因为每个这样的代理调用它不会规模将占据一个线程在你的米achine。

我的建议是:使用原始套接字或并拦截最低级别的HTTP协议,只是在运行时替换一些值(如Host标头)。你能提供更多的背景吗,为什么你需要这个?无论你应该做一边转发这样的

+0

我有一个客户端,一个中间服务器和几台主服务器。 客户端只会与调度其服务器的中间服务器通话。服务器向中间服务器返回一个响应,然后将其处理,然后将响应返回给客户端。 – user1144031

+0

不应该复制请求标头(3)和主体(4)来弥补复制请求参数(2)(因为post参数是请求主体的一部分,因为它们会成为url的一部分)?如果我执行这两个步骤,它会减少(在http请求中) – mickeymoon

10

讨论,这里是我是如何做的:

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

2

我还需要做同样的事情,在使用Spring控制器和RestTemplate进行一些非最优化之后,我发现了一个更好的解决方案:Smiley's HTTP Proxy Servlet。好处是,它确实是AS-IS代理,就像Apache的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> 

或者,当然,你可以用annotation config去。

如果您在使用Spring启动,它是更容易:你只需要创建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进行配置的教程可以在this link上找到。