Netty架构探索之NIO架构演变

Netty初探

NIO 的类库和 API 繁杂, 使用麻烦: 需要熟练掌握Selector、 ServerSocketChannel、 SocketChannel、 ByteBuffer等。 开发工作量和难度都非常大: 例如客户端面临断线重连、 网络闪断、心跳处理、半包读写、 网络拥塞和异常流的处理等等。Netty 对 JDK 自带的 NIO 的 API 进行了良好的封装,解决了上述问题。且Netty拥有高性能、 吞吐量更高,延迟更低,减少资源消耗,最小化不必要的内存复制等优点。Netty 现在都在用的是4.x,5.x版本已经废弃,Netty 4.x 需要JDK 6以上版本支持。

Netty的使用场景

1)互联网行业:在分布式系统中,各个节点之间需要远程服务调用,高性能的 RPC 框架必不可少,Netty 作为异步高性能的通信框架,往往作为基础通信组件被这些 RPC 框架使用。典型的应用有:阿里分布式服务框架 Dubbo 的 RPC 框架使用 Dubbo 协议进行节点间通信,Dubbo 协议默认使用 Netty 作为基础通信组件,用于实现。各进程节点之间的内部通信。Rocketmq底层也是用的Netty作为基础通信组件。
2)游戏行业:无论是手游服务端还是大型的网络游戏,Java 语言得到了越来越广泛的应用。Netty 作为高性能的基础通信组件,它本身提供了 TCP/UDP 和 HTTP 协议栈。
3)大数据领域:经典的 Hadoop 的高性能通信和序列化组件 Avro 的 RPC 框架,默认采用 Netty 进行跨界点通信,它的 Netty Service 基于 Netty 框架二次封装实现。
netty相关开源项目:https://netty.io/wiki/related-projects.html

Netty通讯示例

Netty的maven依赖:

<dependency>
    <groupId>io.netty</groupId>
    <artifactId>netty-all</artifactId>
    <version>4.1.35.Final</version>
</dependency>

NettyServer(Netty服务端)

package com.io.netty.demo;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;

public class NettyServer {

    public static void main(String[] args) {
        EventLoopGroup bossGroup = new NioEventLoopGroup(1);
        EventLoopGroup workGroup = new NioEventLoopGroup();
        try {

            ServerBootstrap bootstrap = new ServerBootstrap();
            bootstrap.group(bossGroup, workGroup)
                    .channel(NioServerSocketChannel.class)
                    // 初始化服务器连接队列大小,服务端处理客户端连接请求是顺序处理的,所以同一时间只能处理一个客户端连接。
                    // 多个客户端同时来的时候,服务端将不能处理的客户端连接请求放在队列中等待处理
                    .option(ChannelOption.SO_BACKLOG, 1024)
                    //创建通道初始化对象,设置初始化参数
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ch.pipeline().addLast(new NettyServerHandler());
                        }
                    });
            System.out.println("netty server start success  ");
            //绑定一个端口并且同步, 生成了一个ChannelFuture异步对象,通过isDone()等方法可以判断异步事件的执行情况
            // 启动服务器(并绑定端口),bind是异步操作,sync方法是等待异步操作执行完毕
            ChannelFuture channelFuture = bootstrap.bind(8000).sync();
            //对通道关闭进行监听,closeFuture是异步操作,监听通道关闭
            // 通过sync方法同步等待通道关闭处理完毕,这里会阻塞等待通道关闭完成
            channelFuture.channel().closeFuture().sync();
        } catch (Exception ce) {
            ce.printStackTrace();
        } finally {
            bossGroup.shutdownGracefully();
            workGroup.shutdownGracefully();
        }
    }
}

NettyClient(Netty 客户端)

package com.io.netty.demo;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;

public class NettyClient {


    public static void main(String[] args) {

        EventLoopGroup clientGoup = new NioEventLoopGroup();

        try {
            Bootstrap bootstrap = new Bootstrap();
            bootstrap.group(clientGoup)
                    .channel(NioSocketChannel.class)// 使用 NioSocketChannel 作为客户端的通道实现
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ch.pipeline().addLast(new NettyClientHandler());
                        }
                    });

            //启动客户端去连接服务器端
            ChannelFuture cf = bootstrap.connect("127.0.0.1", 8000).sync();
            //对关闭通道进行监听
            cf.channel().closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            clientGoup.shutdownGracefully();
        }
    }


}

NettyClientHandler(netty客户端处理器)

package com.io.netty.demo;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.CharsetUtil;

public class NettyClientHandler extends ChannelInboundHandlerAdapter {


    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
      ctx.writeAndFlush(Unpooled.copiedBuffer("hello server", CharsetUtil.UTF_8));

    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        ByteBuf buf = (ByteBuf) msg;
        System.out.println("接受到服务端的消息:"+buf.toString(CharsetUtil.UTF_8));
    }
    }

NettyServerHandler(netty服务端处理器)

package com.io.netty.demo;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;

import java.nio.charset.Charset;

public class NettyServerHandler extends ChannelInboundHandlerAdapter {

    private Charset charset;

    public NettyServerHandler() {
        charset = Charset.forName("UTF-8");
    }

    /**
     * 当客户端发送数据到通道,这里是通道读取
     *
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        System.out.println(Thread.currentThread().getName() + " 接受请求");
        ByteBuf buf = (ByteBuf) msg;

        System.out.println("服务端接受到的数据:" + buf.toString(charset));
    }

    /**
     * 通道数据读取完毕调用
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
        System.out.println("服务端数据读取完毕");
        ByteBuf buf = Unpooled.copiedBuffer("hello client", charset);
        ctx.writeAndFlush(buf);
    }

    /**
     * 当客户端建立了连接过后,服务端会调用channelActive方法
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("通道激活");
        //建立连接可以给客户端发送消息
        ctx.writeAndFlush(Unpooled.copiedBuffer("hello", charset));

    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println("11111");
    }

    /**
     * 当客户端建立了连接过后,要注册到通道上会调用
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
        System.out.println("注册...");
    }

    /**
     * 当客户端失去连接过后,会调用
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
        System.out.println("客户端通道移除");
    }

    /**
     * 出现异常调用
     *
     * @param ctx
     * @param cause
     * @throws Exception
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        System.out.println("出现异常");
    }
}

值得一说的是如果不编写客户端,只要服务端启动了,那么也是可以通过telnet 发送数据的;我们发现Netty框架的目标就是让你的业务逻辑从网络基础应用编码中分离出来,让你可以专注业务的开发,而不需写一大堆类似NIO的网络处理操作。

NIO架构演变

Netty的线程模型在《Scalable IO in Java》一书中对io的架构演变做了一个介绍,可以先理解理解下里面所说的io模型的演变,就能够更清楚的理解Netty为什么会出现

传统的BIO模型

在这里插入图片描述
传统的BIO模型在之前的笔记中已经说的很多了,BIO的模型面临的问题就是需要开启大量的线程,从上图也可以知道, 每个client请求到server过后,server都需要开启一个线程去处理client的请求,然后经过读取、解码、处理业务,最后发送请求处理结果到client,但是如果这个环节上的任何一个地方出现缓慢,那么这个线程是不会释放的,也就是会一直占用这个线程,直到send完毕,而如果大量的连接进来过后,这种模型的弊端就显而易见,要开启大量的线程,如果已经开启的线程处理不能及时完成,那么最终的结果就是后面的连接会一直阻塞在队列中而不是及时响应客户端从而线程超时。

基本的 Reactor模型

在这里插入图片描述
这种模型就是NIO的模型,这种模型采用的是linxu的epoll模型,之前的笔记中已经用了大量的文字来说明了NIO和epoll、poll、select了,这里就不多说了,这种模型采用的是事件响应机制,就是一个reactor就是一个io多路复用器Selector,所以Netty就是采用了Reactor的模型,响应式编程;这种模型在之前的笔记中已经通过对代码来说明了,就是客户端开启一个线程,客户端的这个serversocketChannel也注册到多路复用器Selector上,然后开启select(调用了linxu底层的epoll_wait),如果在某一个时刻有客户端连接过来了或者已经有了的连接发生了事件,比如读写事件 ,那么Reactor(selector)会通知事件响应的机制接受到客户端的 socketChannel,然后进行处理,处理的事件有两种,一种acceptor(连接事件),一种数据读取事件,也就是业务处理事件;那么这种模型就是一个线程在轮询,当轮询到select的时候,如果没有事件发生,就会一直进行阻塞,如果有事件发生,那么通过selectKeys获取到发生了事件的key(socketChannel),然后进行处理,比如有 10万个连接过来了,那么selectKey中就有10万个连接请求,那么这个线程会把这10万个连接依次的注册到多路复用器selector上,之前的代码截图如下:
在这里插入图片描述
上图上之前笔记中的代码片段,这种reactor的单线程模型,可能大家想都想到了,肯定是有问题的,有什么问题呢?代码执行到1的时候,如果这个时候没有连接事件或者其他事件过来,那么会一直阻塞到1处,但是如果这个时候有10万个连接过来了,那么通过selctedKeys得到这10万个连接,并且把这个10万个连接注册到多路复用器上,但是有个问题,如果你在注册这些连接到多路复用器 上的时候,又有连接过来了,那么这个时候你还在注册之前的连接,那么这个时候其他过来的连接或者读写事件都能等着,什么事儿也做不了,再比如,如果 这10万个事件中有1万个是读事件,其他是连接,但是在2和3处都需要去处理,都需要事件,所以上面的这个写法肯定是有问题的,所以 这也就是单线程的NIO所带来的一些问题,虽然它可以将一个线程的性能压榨到极限,但是在处理的过程中也会产生一些问题,并发较高的情况下,一个线程的处理能力也是有限的,而且如果其中有些读取事件处理缓慢的话,那么和BIO一样,新过来的连接会一直处于等待状态,无法得到即使的处理响应。

多线程的Reactor模型

在这里插入图片描述
从上图可以看出在基本的模型的基础上进行了优化,acceptor是接受连接事件,而read是读取client发送数据的事件,这种优化后的模型的意思就是将上面的代码片段的3处全部 丢到一个线程池Thread Pool中去执行,意思就是读取事件交给线程池去做,而接受连接事件还是在原来的线程中去处理,因为接受连接注册到多路复用器上是很快的,分发线程处理也是很快的,处理都很快,这样在一定程度上可以解决之前的单线程模型由于事件过多而导致的后续事件的等待问题;但是这种模式还是有缺陷,有什么缺陷呢?就是比并发较高的情况下,过来的读取事件太多,连接事件也较多或者读取事件很多的情况下,分发线程还是需要一定的时间的,而且如果并发很高的情况下,线程池满了,后续的读取时间也需要阻塞的,所以在一定高并发的场景下还是有缺陷的。

主从Reactor模型

在这里插入图片描述
上面的这个模型就是典型的Netty模型,就是一主一从模型,什么意思呢?从多线程版本的Reactaor模型中可以知道,只有一个Reactor,那么不管你是否使用线程池,那么接受连接事件和抽离读取事件都是在一个Reactor中处理的,这里要明白的是一个Reactor就是一个多路复用器,现在的这种模型就是多个Reactor,也就是多个多路复用器Selector;这种多Reactor模型从上图也可以知道将处理连接事件和处理读取事件分开了,这样的好处就是主Reactor只需要处理连接时间,比如有了1000个连接过来了,主Reactor就可以直接注册到从Reactor的多路复用器上,简单来举例就是一个公司中部门经理就是比喻成一个主Reactor,他的工作就是去承接所有的项目,因为他是部门经理,人脉广,他可以一个人谈很多的项目,他谈好的项目他就可以直接转给部门员工去完成,因为项目的完成过程是很久的,所以主reactor只需要接受连接,然后注册到从Reactor的多路 复用器上,当有 读写事件过来了,直接在从reactor上进行处理,而从Reactor也是采用线程池的分发来处理的,所以这种一主一从的Reactor模型就是Netty采用的一种模型。我们来看下netty的使用的时候要创建两个线程组:

EventLoopGroup bossGroup = new NioEventLoopGroup(1);
EventLoopGroup workGroup = new NioEventLoopGroup();

上面的bossGoup就可以看成是mainReactor,而wokGroup就是subReactor,一般mainReactor是 需要分配一个线程即可,如果你的系统并发实在是高的很,那么可以多分配几个,因为接受连接事件处理来肯定是很快的,所以一般一个线程也就够了,不需要分配太多,而如果NioEventLoopGroup构造方法没有传入具体的线程数,那么netty底层默认分配的是机器的 CPU核数 * 2;如果mainReactor分配了多个线程,那么在打开监听端口的时候就需要监听多个端口;所以这个就可以是一主一从,或者多主多从,如下图:
在这里插入图片描述

Netty的线程模型

理解了上面的NIO的线程模型的演变过后,其实Netty就是对NIO的线程模型进行了封装,成了异步非阻塞的线程模型,因为AIO目前是操作系统支持不是太好,所以在原来的NIO基础上做了一层封装,支持异步非阻塞的线程模型,而底层是基于EPOLL模型,在处理高并发的场景下是非常好的一种线程模型选择,包括阿里的dubbo底层的rpc协议也是使用netty,所以我这里从网上找了一张图来很好的说明了netty的工作架构图
在这里插入图片描述
模型解释:

  1. Netty 抽象出两组线程池BossGroup和WorkerGroup,BossGroup专门负责接收客户端的连接, WorkerGroup专门负责网络的读写
  2. BossGroup和WorkerGroup类型都是NioEventLoopGroup
  3. NioEventLoopGroup 相当于一个事件循环线程组, 这个组中含有多个事件循环线程 , 每一个事件循环线程是NioEventLoop
  4. 每个NioEventLoop都有一个selector , 用于监听注册在其上的socketChannel的网络通讯
  5. 每个Boss NioEventLoop线程内部循环执行的步骤有 3 步
    1.处理accept事件 , 与client 建立连接 , 生成 NioSocketChannel ;
    2.将NioSocketChannel注册到某个worker NIOEventLoop上的selector;
    3.处理任务队列的任务 , 即runAllTasks
  6. 每个worker NIOEventLoop线程循环执行的步骤
    1.轮询注册到自己selector上的所有NioSocketChannel 的read, write事件
    2.处理 I/O 事件, 即read , write 事件, 在对应NioSocketChannel 处理业务
    3.runAllTasks处理任务队列TaskQueue的任务 ,一些耗时的业务处理一般可以放入TaskQueue中慢慢处理,这样不影响数据在 pipeline 中的流动处理
  7. 每个worker NIOEventLoop处理NioSocketChannel业务时,会使用 pipeline (管道),管道中维护了很多 handler 处理器用来处理 channel 中的数据。

Netty模块组件

【Bootstrap、ServerBootstrap】:
Bootstrap 意思是引导,一个 Netty 应用通常由一个 Bootstrap 开始,主要作用是配置整个 Netty 程序,串联各个组件,Netty 中 Bootstrap 类是客户端程序的启动引导类,ServerBootstrap 是服务端启动引导类。
【Future、ChannelFuture】:
正如前面介绍,在 Netty 中所有的 IO 操作都是异步的,不能立刻得知消息是否被正确处理。
但是可以过一会等它执行完成或者直接注册一个监听,具体的实现就是通过 Future 和 ChannelFutures,他们可以注册一个监听,当操作执行成功或失败时监听会自动触发注册的监听事件。
【Channel】:
Netty 网络通信的组件,能够用于执行网络 I/O 操作。Channel 为用户提供:
1)当前网络连接的通道的状态(例如是否打开?是否已连接?)
2)网络连接的配置参数 (例如接收缓冲区大小)
3)提供异步的网络 I/O 操作(如建立连接,读写,绑定端口),异步调用意味着任何 I/O 调用都将立即返回,并且不保证在调用结束时所请求的 I/O 操作已完成。
4)调用立即返回一个 ChannelFuture 实例,通过注册监听器到 ChannelFuture 上,可以 I/O 操作成功、失败或取消时回调通知调用方。
5)支持关联 I/O 操作与对应的处理程序。
不同协议、不同的阻塞类型的连接都有不同的 Channel 类型与之对应。
下面是一些常用的 Channel 类型:

NioSocketChannel,异步的客户端 TCP Socket 连接。
NioServerSocketChannel,异步的服务器端 TCP Socket 连接。
NioDatagramChannel,异步的 UDP 连接。
NioSctpChannel,异步的客户端 Sctp 连接。
NioSctpServerChannel,异步的 Sctp 服务器端连接。
这些通道涵盖了 UDP 和 TCP 网络 IO 以及文件 IO。

【Selector】:
Netty 基于 Selector 对象实现 I/O 多路复用,通过 Selector 一个线程可以监听多个连接的 Channel 事件。
当向一个 Selector 中注册 Channel 后,Selector 内部的机制就可以自动不断地查询(Select) 这些注册的 Channel 是否有已就绪的 I/O 事件(例如可读,可写,网络连接完成等),这样程序就可以很简单地使用一个线程高效地管理多个 Channel 。
【NioEventLoop】:
NioEventLoop 中维护了一个线程和任务队列,支持异步提交执行任务,线程启动时会调用 NioEventLoop 的 run 方法,执行 I/O 任务和非 I/O 任务:
I/O 任务,即 selectionKey 中 ready 的事件,如 accept、connect、read、write 等,由 processSelectedKeys 方法触发。
非 IO 任务,添加到 taskQueue 中的任务,如 register0、bind0 等任务,由 runAllTasks 方法触发。
【NioEventLoopGroup】:
NioEventLoopGroup,主要管理 eventLoop 的生命周期,可以理解为一个线程池,内部维护了一组线程,每个线程(NioEventLoop)负责处理多个 Channel 上的事件,而一个 Channel 只对应于一个线程。
【ChannelHandler】:
ChannelHandler 是一个接口,处理 I/O 事件或拦截 I/O 操作,并将其转发到其 ChannelPipeline(业务处理链)中的下一个处理程序。
ChannelHandler 本身并没有提供很多方法,因为这个接口有许多的方法需要实现,方便使用期间,可以继承它的子类:

ChannelInboundHandler 用于处理入站 I/O 事件。
ChannelOutboundHandler 用于处理出站 I/O 操作。

或者使用以下适配器类:

ChannelInboundHandlerAdapter 用于处理入站 I/O 事件。
ChannelOutboundHandlerAdapter 用于处理出站 I/O 操作。

【ChannelHandlerContext】:
保存 Channel 相关的所有上下文信息,同时关联一个 ChannelHandler 对象。
【ChannelPipline】:
保存 ChannelHandler 的 List,用于处理或拦截 Channel 的入站事件和出站操作。
ChannelPipeline 实现了一种高级形式的拦截过滤器模式,使用户可以完全控制事件的处理方式,以及 Channel 中各个的 ChannelHandler 如何相互交互。
在 Netty 中每个 Channel 都有且仅有一个 ChannelPipeline 与之对应,它们的组成关系如下:
在这里插入图片描述
一个 Channel 包含了一个 ChannelPipeline,而 ChannelPipeline 中又维护了一个由 ChannelHandlerContext 组成的双向链表,并且每个 ChannelHandlerContext 中又关联着一个 ChannelHandler。
read事件(入站事件)和write事件(出站事件)在一个双向链表中,入站事件会从链表 head 往后传递到最后一个入站的 handler,出站事件会从链表 tail 往前传递到最前一个出站的 handler,两种类型的 handler 互不干扰。

ByteBuf详解

从结构上来说,ByteBuf 由一串字节数组构成。数组中每个字节用来存放信息。
ByteBuf 提供了两个索引,一个用于读取数据,一个用于写入数据。这两个索引通过在字节数组中移动,来定位需要读或者写信息的位置。当从 ByteBuf 读取时,它的 readerIndex(读索引)将会根据读取的字节数递增。同样,当写 ByteBuf 时,它的 writerIndex 也会根据写入的字节数进行递增。
在这里插入图片描述
需要注意的是极限的情况是 readerIndex 刚好读到了 writerIndex 写入的地方。
如果 readerIndex 超过了 writerIndex 的时候,Netty 会抛出 IndexOutOf-BoundsException 异常。
示例代码:

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.util.CharsetUtil;

public class NettyByteBuf {
    public static void main(String[] args) {
        // 创建byteBuf对象,该对象内部包含一个字节数组byte[10]
        // 通过readerindex和writerIndex和capacity,将buffer分成三个区域
        // 已经读取的区域:[0,readerindex)
        // 可读取的区域:[readerindex,writerIndex)
        // 可写的区域: [writerIndex,capacity)
        ByteBuf byteBuf = Unpooled.buffer(10);
        System.out.println("byteBuf=" + byteBuf);

        for (int i = 0; i < 8; i++) {
            byteBuf.writeByte(i);
        }
        System.out.println("byteBuf=" + byteBuf);

        for (int i = 0; i < 5; i++) {
            System.out.println(byteBuf.getByte(i));
        }
        System.out.println("byteBuf=" + byteBuf);

        for (int i = 0; i < 5; i++) {
            System.out.println(byteBuf.readByte());
        }
        System.out.println("byteBuf=" + byteBuf);


        //用Unpooled工具类创建ByteBuf
        ByteBuf byteBuf2 = Unpooled.copiedBuffer("hello,zhuge!", CharsetUtil.UTF_8);
        //使用相关的方法
        if (byteBuf2.hasArray()) {
            byte[] content = byteBuf2.array();
            //将 content 转成字符串
            System.out.println(new String(content, CharsetUtil.UTF_8));
            System.out.println("byteBuf2=" + byteBuf2);

            System.out.println(byteBuf2.getByte(0)); // 获取数组0这个位置的字符h的ascii码,h=104

            int len = byteBuf2.readableBytes(); //可读的字节数  12
            System.out.println("len=" + len);

            //使用for取出各个字节
            for (int i = 0; i < len; i++) {
                System.out.println((char) byteBuf2.getByte(i));
            }

            //范围读取
            System.out.println(byteBuf2.getCharSequence(0, 6, CharsetUtil.UTF_8));
            System.out.println(byteBuf2.getCharSequence(6, 6, CharsetUtil.UTF_8));
        }
    }
}

Netty实战聊天室系统

ChatServer(服务端):


import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;

public class ChatServer {


    public static void main(String[] args) {
        EventLoopGroup bossGroup = new NioEventLoopGroup(1);
        EventLoopGroup workGroup = new NioEventLoopGroup();

        try {
            ServerBootstrap bootstrap = new ServerBootstrap();
            bootstrap.group(bossGroup, workGroup)
                    .channel(NioServerSocketChannel.class)
                    .option(ChannelOption.SO_BACKLOG, 1024)
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ChannelPipeline pipeline = ch.pipeline();
                            pipeline.addLast("encoder", new StringEncoder());
                            pipeline.addLast("decoder", new StringDecoder());
                            pipeline.addLast(new ChatServerHandler());
                        }
                    });
            ChannelFuture channel = bootstrap.bind(8000).sync();
            System.out.println("chat service staring");
            channel.channel().closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            bossGroup.shutdownGracefully();
            workGroup.shutdownGracefully();
        }
    }
}

ChatServerHandler (服务端处理器)


import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.text.SimpleDateFormat;
import java.util.Date;

public class ChatServerHandler extends SimpleChannelInboundHandler<String> {

    private SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    static  ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    /**
     * 发送消息读取
     * @param ctx
     * @param msg
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.forEach(ch ->{
            if(ch != channel){
                ch.writeAndFlush("[客户端] "+ ch.remoteAddress()  +" "+ dateFormat.format(new Date()) +" 发送了消息:" + msg +"\n");
            }else{
                ch.writeAndFlush("[自己] "+ch.remoteAddress() +" "+ dateFormat.format(new Date()) +" 发送了消息:"+msg +" \n");
            }
        });

    }

    /**
     *下线了提醒
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("[客户端] "+channel.remoteAddress() +" 下线了 "+ dateFormat.format(new Date()) +"\n");
        System.out.println("[客户端]"+channel.remoteAddress()+"下线了 \n");
        System.out.println("下线了,size="+channelGroup.size());
    }

    /**
     * 通道激活
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("[客户端] "+channel.remoteAddress()+" 上线了 "+ dateFormat.format(new Date())+"\n");
        channelGroup.add(channel);
        System.out.println(channelGroup.size());
        System.out.println("[客户端] "+channel.remoteAddress()+" 上线了 \n");
    }

}

ChatClient (客户端):


import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;

import java.util.Scanner;

public class ChatClient {

    public static void main(String[] args) {
        EventLoopGroup workGroup = new NioEventLoopGroup();

        try {
            Bootstrap bootstrap = new Bootstrap();
            bootstrap.group(workGroup)
                    .channel(NioSocketChannel.class)
                    .handler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            ChannelPipeline pipeline = ch.pipeline();
                            pipeline.addLast(new StringEncoder());
                            pipeline.addLast(new StringDecoder());
                            pipeline.addLast(new ChatClientHandler());
                        }
                    });
            ChannelFuture future = bootstrap.connect("127.0.0.1", 8000).sync();

            Channel channel = future.channel();
            Scanner scanner = new Scanner(System.in);
            while (scanner.hasNext()){
                String s = scanner.nextLine();
                channel.writeAndFlush(s);
            }
            channel.closeFuture().sync();
        } catch (InterruptedException e) {
            e.printStackTrace();
        } finally {
            workGroup.shutdownGracefully();
        }
    }
}

ChatClientHandler (客户端处理器)


import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;

public class ChatClientHandler extends SimpleChannelInboundHandler<String> {
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        System.out.println(msg);
    }
}

我在码云上之前写了一个netty的简单rpc通讯框架,就是学习使用的
https://gitee.com/scjava/bml-rpc.git

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值