nio实现一个简单的reactor模型server

nio涉及的类和方法

buffer

nio的buffer本质上是,一块内存区域。被封装到Buffer类里,并提供一组方法,供(channel)读写数据。

  1. 读写数据分如下4个步骤:
  • 写入数据到Buffer
  • 调用flip()方法
  • 从Buffer中读取数据
  • 调用clear()方法或者compact()方法

当向buffer写入数据时,buffer会记录下写了多少数据。一旦要读取数据,需要通过flip()方法将Buffer从写模式切换到读模式。在读模式下,可以读取之前写入到buffer的所有数据。

一旦读完了所有的数据,就需要清空缓冲区,让它可以再次被写入。有两种方式能清空缓冲区:调用clear()或compact()方法。clear()方法会清空整个缓冲区。compact()方法只会清除已经读过的数据。任何未读的数据都被移到缓冲区的起始处,新写入的数据将放到缓冲区未读数据的后面

  1. buffer的分配
ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
  1. buffer的类型
  • ByteBuffer
  • MappedByteBuffer
  • CharBuffer
  • DoubleBuffer
  • FloatBuffer
  • IntBuffer
  • LongBuffer
  • ShortBuffer
  1. buffer中写数据
    一种是调用buffer的put方法,一种是从channel中获取。
byteBuffer.put(new byte[127]);
//channel read是buffer的写
clientChannel.read(byteBuffer);
  1. buffer读数据
    读数据前要调用flip(),切换到读模式下。
    一种是调用buffer的get方法,一种是往channel写入
//保证读取的内容从头开始,并且读取的长度是已经写入的长度
byteBuffer.flip();
byteBuffer.get();
clientChannel.write(byteBuffer);
  1. clear()与compact()方法
    读完Buffer中的数据,如果需要让Buffer准备好再次被写入。可以通过clear()或compact()方法来完成。

如果Buffer中有一些未读的数据,调用clear()方法,数据将“被遗忘”,意味着不再有任何标记会告诉你哪些数据被读过,哪些还没有。

如果Buffer中仍有未读的数据,且后续还需要这些数据,但是此时想要先先写些数据,那么使用compact()方法。

select
  1. select的创建
Selector selector = Selector.open();
  1. select()方法
    select方法会阻塞

reactor模型在nio的基础上
(1)有个主线程(Accept)去接收读写事件。具体的处理线程(Process)去处理读写请求(使用线程池)。
(2)引入队列的概念,比如把写请求事件放到队列里,process去消费队列,去处理写处理。
*

还是以echo功能为例子。事件简单的reactor模型

package com.lxqn.jiapeng.reactorO;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Set;

/**
 * 主线程 Accept功能 接收器
 * Created by jiapeng on 2017/10/7.
 */
public class NIOServer {

    public static void main(String[] args) throws IOException {
        Selector selector = Selector.open();
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.configureBlocking(false);
        serverSocketChannel.bind(new InetSocketAddress(9998));
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        int coreNum = Runtime.getRuntime().availableProcessors();
        Processor[] processors = new Processor[coreNum];
        for (int i = 0; i < processors.length; i++) {
            processors[i] = new Processor();
        }
        int index = 0;
        while (true) {
            int n=selector.select(1000);
            if (n == 0) {
                System.out.print(".");
                continue;
            }
            System.out.println("n="+n);

            if(n>0){
                Set<SelectionKey> keys = selector.selectedKeys();
                for (SelectionKey key : keys) {
                    keys.remove(key);
                    if (key.isAcceptable()) {
                        ServerSocketChannel acceptServerSocketChannel = (ServerSocketChannel) key.channel();
                        SocketChannel socketChannel = acceptServerSocketChannel.accept();
                        socketChannel.configureBlocking(false);
                        System.out.println("Accept request from {}" + socketChannel.getRemoteAddress());
                        Processor processor = processors[(int) ((index++) % coreNum)];
                        processor.addChannel(socketChannel);
                        processor.wakeup();
                    }
                }
            }

        }
    }
}
package com.lxqn.jiapeng.reactorO;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.nio.channels.spi.SelectorProvider;
import java.util.Iterator;
import java.util.Set;
import java.util.concurrent.ConcurrentLinkedQueue;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 具体处理网络IO的类
 * Created by jiapeng on 2017/10/7.
 */
public class Processor {
    //使用线程池,大小为核数的2倍
    private static final ExecutorService service =
            Executors.newFixedThreadPool(2 * Runtime.getRuntime().availableProcessors());
    private Selector selector;

    //写操作队列,
    protected ConcurrentLinkedQueue<SelectionKey> toWrite;

    public Processor() throws IOException {
        this.selector = SelectorProvider.provider().openSelector();
        toWrite=new ConcurrentLinkedQueue<SelectionKey>();
        start();
    }
    public void addChannel(SocketChannel socketChannel) throws ClosedChannelException {
        socketChannel.register(this.selector, SelectionKey.OP_READ,ByteBuffer.allocate(1024));
    }
    public void wakeup() {
        this.selector.wakeup();
    }
    public void start() {
        service.submit(() -> {
            while (true) {
                int n=selector.select(1000);
                if (n == 0) {
                    continue;
                }
                if(n>0){
                    System.out.println("process n="+n);
                    Set<SelectionKey> keys = selector.selectedKeys();
                    Iterator<SelectionKey> iterator = keys.iterator();
                    while (iterator.hasNext()) {
                        SelectionKey key = iterator.next();
                        iterator.remove();
                        if (key.isReadable()) {
                            ByteBuffer buffer = (ByteBuffer) key.attachment();
                            SocketChannel socketChannel = (SocketChannel) key.channel();
                            int count = socketChannel.read(buffer);
                            System.out.println("count="+count);
                            if (count < 0) {
                                socketChannel.close();
                                key.cancel();
                                System.out.println("Read ended"+socketChannel);
                                continue;
                            } else if (count == 0) {
                                System.out.println("Message size is 0"+socketChannel);
                                continue;
                            } else {
                                System.out.println("Read message"+socketChannel+" message="+new String(buffer.array()));
                                key.interestOps(SelectionKey.OP_WRITE);
                            }
                        }
                        if (key.isWritable()) {
                            System.out.println("start write");
                            toWrite.offer(key);
                        }
                        //模拟队列处理的操作
                        SelectionKey writableKey=null;
                        while((writableKey=toWrite.poll())!=null) {
                            System.out.println("toWrite operation");
                            write(writableKey);
                        }
                    }
                }
            }
        });
    }

    private void write(SelectionKey key) throws IOException{
        ByteBuffer buf = (ByteBuffer) key.attachment();
        //make buffer ready for read
        buf.flip();
        SocketChannel clientChannel = (SocketChannel) key.channel();
        clientChannel.write(buf);
        if (!buf.hasRemaining()) {
            //交替注册读写操作,实现echo的功能
            key.interestOps(SelectionKey.OP_READ);
        }
        //清空buffer,清理已读取长度的buffer
        buf.compact();
        System.out.println("toWrite"+buf.toString()+" message="+new String(buf.array()));
    }

}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,219评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,363评论 1 293
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,933评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,020评论 0 206
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,400评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,640评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,896评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,597评论 0 199
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,327评论 1 244
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,581评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,072评论 1 261
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,399评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,054评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,083评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,849评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,672评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,585评论 2 270

推荐阅读更多精彩内容