java nio 实现最精简代理服务器(代理tomcat服务器),没有select多线程,只有ServerSocketChannel与socketChannel

简介:使用java的nio技术,实现简单的代理服务器,访问本地1234端口,把请求转发的tomcat8080端口,实现代理

使用过程:
1. 开启tomcat,确保访问127.0.0.1:8080端口有显示
2. 启动App.java的main运行
3. 然后访问127.0.0.1:1234端口

有三个java文件
App.java 启动main进程
Server.java 服务类,监听1234端口,开启代理
Proxy.java 代理类,实现请求转发,代理访问8080端口,返回请求结果

App.java

public class App {

    public static void main(String[] args) throws Exception {

        Server server = new Server(1234);
        server.connection();
    }
}

Server.java

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;

public class Server {

    private int port;
    private SocketChannel remoteSocketChannel;

    public Server(int port) {
        this.port = port;
    }

    public void connection() throws Exception
    {
        System.out.println("start server listen...........");
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.socket().bind(new InetSocketAddress(this.port));

        while(true){
            SocketChannel socketChannel = serverSocketChannel.accept();
            this.remoteSocketChannel=socketChannel;
            System.out.println("get one conn:"+socketChannel);

            ByteBuffer buf = ByteBuffer.allocate(10240);
            buf.clear();
            int len = socketChannel.read(buf);

            System.out.println("read from client:"+len+" ..............");

            Proxy client = new Proxy("localhost",8080,this);
            ByteBuffer request = buf.duplicate();
            request.rewind();
            client.request(request);

            buf.clear();
        }
    }

    public void output(ByteBuffer buf) throws Exception
    {
        remoteSocketChannel.write(buf);
    }

    public void done() throws Exception
    {
        remoteSocketChannel.close();
    }
}

Proxy.java

import java.io.Serializable;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;

class Proxy {

    private String proxyIp;
    private int port;
    private Server server;

    public Proxy(String proxyIp, int port,Server server) {
        this.proxyIp = proxyIp;
        this.port = port;
        this.server=server;
    }

    public void request(ByteBuffer buf) throws Exception {

        System.out.println("start proxy conn........");
        SocketChannel socketChannel = SocketChannel.open();
        socketChannel.connect(new InetSocketAddress(this.proxyIp, this.port));

        while (true) {
            System.out.println("in proxy while.................");
            if (socketChannel.isConnected()) {
                System.out.println("proxy isConnected.............");
                //发送http

                int bytesWrite = socketChannel.write(buf);

                buf.clear();

                int read = 0;
                ByteBuffer buffer = ByteBuffer.allocate(1024);
                while ((read = socketChannel.read(buffer)) != -1) {
                    {
                        System.out.println("read from proxy:" + read + ".....................");
                        buffer.flip();
                        while (buffer.position() < buffer.limit()) {
                            //不能解析中文
//                            System.out.print((char) buffer.get());
                            //解析中文,不知道是否有个别乱码,缓存刚好截断utf8中文(三个字节)?
//                            System.out.println(Charset.forName("UTF-8").newDecoder().decode(buffer).toString());
                            server.output(buffer);
                        }
                        //#todo 需要计算body长度,缓存没满不一定读完数据,html文档没读完
                        if (buffer.limit() < buffer.capacity()) {
                            break;
                        }
                        buffer.clear();
                    }
                }
            }
            socketChannel.close();
            server.done();
            break;
        }
    }
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Netty是一款基于NIO的网络编程框架,提供了高效、稳定、灵活的网络编程能力。使用Netty实现代理服务器可以简化开发过程,提高性能和可维护性。 以下是使用Netty实现代理服务器的示例代码: ``` import io.netty.bootstrap.Bootstrap; import io.netty.buffer.ByteBuf; import io.netty.channel.*; import io.netty.channel.socket.SocketChannel; import io.netty.channel.socket.nio.NioSocketChannel; import io.netty.handler.codec.http.*; import io.netty.handler.logging.LogLevel; import io.netty.handler.logging.LoggingHandler; import io.netty.handler.stream.ChunkedWriteHandler; public class ProxyServer { public static void main(String[] args) throws Exception { EventLoopGroup workerGroup = new NioEventLoopGroup(); try { Bootstrap bootstrap = new Bootstrap(); bootstrap.group(workerGroup) .channel(NioSocketChannel.class) .handler(new LoggingHandler(LogLevel.INFO)) .option(ChannelOption.AUTO_READ, false) .handler(new ChannelInitializer<SocketChannel>() { @Override protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline().addLast(new HttpClientCodec()); ch.pipeline().addLast(new HttpObjectAggregator(65536)); ch.pipeline().addLast(new ChunkedWriteHandler()); ch.pipeline().addLast(new ProxyServerHandler()); } }); ChannelFuture future = bootstrap.connect("www.example.com", 80).sync(); future.channel().closeFuture().sync(); } finally { workerGroup.shutdownGracefully(); } } private static class ProxyServerHandler extends ChannelInboundHandlerAdapter { private Channel remoteChannel; @Override public void channelActive(ChannelHandlerContext ctx) throws Exception { remoteChannel = ctx.channel(); ctx.read(); } @Override public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof HttpRequest) { HttpRequest request = (HttpRequest) msg; String host = request.headers().get("Host"); ChannelFuture future = new Bootstrap() .group(ctx.channel().eventLoop()) .channel(ctx.channel().getClass()) .handler(new LoggingHandler(LogLevel.INFO)) .option(ChannelOption.AUTO_READ, false) .option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 10000) .handler(new ChannelInitializer<SocketChannel>() { @Override protected void initChannel(SocketChannel ch) throws Exception { ch.pipeline().addLast(new HttpResponseDecoder()); ch.pipeline().addLast(new HttpObjectAggregator(65536)); ch.pipeline().addLast(new ChunkedWriteHandler()); ch.pipeline().addLast(new ChannelInboundHandlerAdapter() { @Override public void channelActive(ChannelHandlerContext ctx) throws Exception { ctx.writeAndFlush(request); ctx.read(); } @Override public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception { if (msg instanceof HttpResponse) { HttpResponse response = (HttpResponse) msg; response.headers().remove("Transfer-Encoding"); response.headers().remove("Content-Length"); remoteChannel.writeAndFlush(response); remoteChannel.writeAndFlush(new ChunkedNioStream((ByteBuf) msg)); } else if (msg instanceof HttpContent) { remoteChannel.writeAndFlush(new ChunkedNioStream((ByteBuf) msg)); if (msg instanceof LastHttpContent) { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } } } @Override public void channelInactive(ChannelHandlerContext ctx) throws Exception { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } @Override public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception { cause.printStackTrace(); remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } }); } }) .connect(host, 80); remoteChannel.config().setAutoRead(false); future.addListener((ChannelFutureListener) future1 -> { if (future1.isSuccess()) { remoteChannel.config().setAutoRead(true); ctx.channel().config().setAutoRead(true); } else { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); ctx.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } }); } else if (msg instanceof HttpContent) { remoteChannel.writeAndFlush(new ChunkedNioStream((ByteBuf) msg)); if (msg instanceof LastHttpContent) { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT); } } } @Override public void channelInactive(ChannelHandlerContext ctx) throws Exception { if (remoteChannel != null) { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } } @Override public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception { cause.printStackTrace(); if (remoteChannel != null) { remoteChannel.writeAndFlush(LastHttpContent.EMPTY_LAST_CONTENT) .addListener(ChannelFutureListener.CLOSE); } ctx.close(); } } } ``` 以上代码中,代理服务器连接到目标服务器的IP地址和端口号是硬编码的,你需要根据实际情况进行修改。在启动代理服务器之后,当客户端发送HTTP请求时,会在一个新的线程中处理请求,解析请求并连接到目标服务器,将请求转发给目标服务器。接收到目标服务器的响应后,将响应转发给客户端。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值