使用Netty构建http代理服务器的简单方法?

时间:2013-03-28 22:17:24

标签: java netty

我是Netty的新手,我正在使用它来创建一个简单的http代理服务器来接收来自客户端的请求,将请求转发到另一台服务器,然后将响应复制回原始请求的响应。一个额外的要求是我能够支持超时,因此如果代理服务器花费太长时间来响应,代理将自行响应并关闭与代理服务器的连接。

我已经使用Jetty实现了这样的应用程序,但是使用Jetty我需要使用太多的线程来阻止入站请求被阻止(这是一个使用非常少的内存或CPU的轻量级应用程序,但是代理服务器足够高,流量突发导致代理服务器排队,或者需要太多线程。)

根据我的理解,我可以使用Netty构建一个管道,其中每个阶段执行少量计算,然后释放它的线程并等待数据准备好进行管道中的下一个阶段。

我的问题是,这样的应用程序有一个简单的例子吗?到目前为止,我对基本的Netty教程的服务器代码进行了简单的修改,但它缺乏对客户端的所有支持。我看到了netty客户端教程,但我不确定如何混合两者中的代码来创建一个简单的代理应用程序。

public static void main(String[] args) throws Exception {
    ChannelFactory factory =
            new NioServerSocketChannelFactory(
                    Executors.newCachedThreadPool(),
                    Executors.newCachedThreadPool());

    ServerBootstrap bootstrap = new ServerBootstrap(factory);

    bootstrap.setPipelineFactory(new ChannelPipelineFactory() {
        public ChannelPipeline getPipeline() {
            return Channels.pipeline(
                    new HttpRequestDecoder(),
                    new HttpResponseEncoder(),
                    /* 
                     * Is there something I can put here to make a
                     * request to another server asynchronously and
                     * copy the result to the response inside
                     * MySimpleChannelHandler?
                     */
                    new MySimpleChannelHandler()
                    );
        }
    });

    bootstrap.setOption("child.tcpNoDelay", true);
    bootstrap.setOption("child.keepAlive", true);

    bootstrap.bind(new InetSocketAddress(8080));
}

private static class MySimpleChannelHandler extends SimpleChannelHandler {

    @Override
    public void messageReceived(ChannelHandlerContext ctx, MessageEvent e) {
        HttpRequest request = (HttpRequest) e.getMessage();
        HttpResponse response = new DefaultHttpResponse(HttpVersion.HTTP_1_1, HttpResponseStatus.OK);
        response.setContent(request.getContent());

        Channel ch = e.getChannel();
        ChannelFuture f = ch.write(response);
        f.addListener(new ChannelFutureListener() {
            public void operationComplete(ChannelFuture future) {
                Channel ch = future.getChannel();
                ch.close();
            }
        });
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, ExceptionEvent e) {
        e.getCause().printStackTrace();

        Channel ch = e.getChannel();
        ch.close();
    }
}

1 个答案:

答案 0 :(得分:3)

你必须看看LittleProxy,看看他们是如何做到的,因为它写在Netty之上。