在同一个Netty程序中支持多个不同协议，打造一站式服务

2023-11-06 21:55:02

简介
在之前的文章中，我们介绍了在同一个netty程序中支持多个不同的服务，它的逻辑很简单，就是在一个主程序中启动多个子程序，每个子程序通过一个BootStrap来绑定不同的端口，从而达到访问不同端口的目的。这种方式非常简单，但也有一个缺点，就是需要创建多个子程序，这会增加程序的复杂性，并且在资源受限的情况下，可能无法创建过多的子程序。

为了解决这个问题，我们可以使用Netty的ChannelFactory来创建一个可以同时支持多个协议的Channel。这样，我们就可以在一个主程序中创建一个ChannelFactory，然后通过这个ChannelFactory创建多个Channel，每个Channel可以绑定不同的端口和协议。这种方式更加简单和高效，并且可以减少程序的复杂性。

实现原理

Netty的ChannelFactory是一个接口，它定义了创建Channel的方法。Netty提供了多种不同的ChannelFactory，包括BootstrapChannelFactory和ServerChannelFactory。BootstrapChannelFactory用于创建客户端Channel，ServerChannelFactory用于创建服务端Channel。

为了在一个主程序中支持多个不同的协议，我们需要创建一个自定义的ChannelFactory，这个ChannelFactory可以同时创建客户端Channel和服务端Channel。我们可以通过继承BootstrapChannelFactory和ServerChannelFactory来实现这个自定义的ChannelFactory。

在自定义的ChannelFactory中，我们需要实现createChannel()方法。这个方法负责创建Channel。在createChannel()方法中，我们需要首先创建一个ChannelConfig对象，然后创建一个ChannelPipeline对象。ChannelConfig对象用于配置Channel，ChannelPipeline对象用于管理Channel的事件处理器。

在ChannelPipeline对象中，我们需要添加不同的事件处理器，这些事件处理器可以处理不同的协议。例如，我们可以添加一个HTTP协议的事件处理器和一个TCP协议的事件处理器。这样，当Channel接收到HTTP请求时，HTTP协议的事件处理器就会被触发，当Channel接收到TCP请求时，TCP协议的事件处理器就会被触发。

代码示例

public class MultiProtocolChannelFactory extends BootstrapChannelFactory implements ServerChannelFactory {

    @Override
    public Channel createChannel() {
        ChannelConfig channelConfig = new DefaultChannelConfig();
        ChannelPipeline channelPipeline = new DefaultChannelPipeline();

        // 添加HTTP协议的事件处理器
        channelPipeline.addLast("http", new HttpServerCodec());
        channelPipeline.addLast("httpHandler", new HttpServerHandler());

        // 添加TCP协议的事件处理器
        channelPipeline.addLast("tcp", new TcpServerCodec());
        channelPipeline.addLast("tcpHandler", new TcpServerHandler());

        Channel channel = new NioSocketChannel(channelConfig, channelPipeline);

        return channel;
    }
}

在主程序中，我们可以使用这个自定义的ChannelFactory来创建多个Channel，每个Channel可以绑定不同的端口和协议。

public class Main {

    public static void main(String[] args) {
        // 创建一个自定义的ChannelFactory
        ChannelFactory channelFactory = new MultiProtocolChannelFactory();

        // 创建一个Channel并绑定80端口，用于监听HTTP请求
        Channel httpChannel = channelFactory.newChannel();
        httpChannel.bind(80);

        // 创建一个Channel并绑定9000端口，用于监听TCP请求
        Channel tcpChannel = channelFactory.newChannel();
        tcpChannel.bind(9000);
    }
}