javanio2

package com.lanhuigu.nio.selector;
 
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

public class NIOServer {
    private Selector selector;
    public void initServer(int port) throws Exception {
        ServerSocketChannel serverChannel = ServerSocketChannel.open();
        // 设置通道为非阻塞
        serverChannel.configureBlocking(false);
        serverChannel.socket().bind(new InetSocketAddress(port));
        this.selector = Selector.open();
        /*
         * 当事件到达的时候,selector.select()会返回,
         * 如果事件没有到达selector.select()会一直阻塞
         */
        SelectionKey selectionKey1 = serverChannel.register(selector, SelectionKey.OP_ACCEPT);//感兴趣事件是16,准备好的事件是0。
        
        int interestSet = selectionKey1.interestOps();//感兴趣事件是16
        boolean isInterestedInAccept  = (interestSet & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT;//true
        int isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT;//0
        int isInterestedInRead    = interestSet & SelectionKey.OP_READ;//0
        int isInterestedInWrite   = interestSet & SelectionKey.OP_WRITE;//0
        
        int readySet = selectionKey1.readyOps();//0
    }
 
    /**
     * 采用轮训的方式监听selector上是否有需要处理的事件,如果有,进行处理
     */
    public void listen() throws Exception {
        System.out.println("start server");
        while (true) {
            // 当注册事件到达时,方法返回,否则该方法会一直阻塞
            selector.select();//客户端条用connect方法接触阻塞。这个selectorchannelArray数组里面绑定了一个ServerSocketChannelImpl和 n 个SocketChannel。接收客户端的事件,有连接事件传递数据过来事件,就会解除阻塞分别调用Accept和read方法。
            // 获得selector中选中的相的迭代器,选中的相为注册的事件
            Iterator ite = this.selector.selectedKeys().iterator();
            while (ite.hasNext()) {
                SelectionKey key = (SelectionKey) ite.next();//客户端条用connect方法,key的感兴趣是16,准备好是16,不准备好select()就会一直阻塞,
                // 删除已选的key以防重负处理
                ite.remove();
                // 客户端请求连接事件
                if (key.isAcceptable()) {
                    ServerSocketChannel server = (ServerSocketChannel) key.channel();//接收连接的channel
                    SocketChannel channel = server.accept();//处理客户端数据的channel
                    // 设置成非阻塞
                    channel.configureBlocking(false);
                    // 在这里可以发送消息给客户端
                    channel.write(ByteBuffer.wrap(new String("hello client").getBytes()));
                    // 在客户端连接成功之后,为了可以接收到客户端的信息,需要给通道设置读的权限
                    SelectionKey selectionKey2 = channel.register(this.selector, SelectionKey.OP_READ);
                    //这个selector绑定了2个channel,分别是服务端和客户端channel,分别关心OP_ACCEPT和OP_READ事件。
                    //这个客户端channel感兴趣是1准备好是0,
                    int interestSet = selectionKey2.interestOps();
                    boolean isInterestedInAccept  = (interestSet & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT;
                    int isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT;
                    int isInterestedInRead    = interestSet & SelectionKey.OP_READ;
                    int isInterestedInWrite   = interestSet & SelectionKey.OP_WRITE;

                    int readySet = selectionKey2.readyOps();
                    // 获得了可读的事件
                } else if (key.isReadable()) {
                    read(key);
                }
            }
        }
    }
 
    private void read(SelectionKey key) throws Exception {
        // 服务器可读消息,得到事件发生的socket通道
        SocketChannel channel = (SocketChannel) key.channel();
        // 读取的缓冲区
        ByteBuffer buffer = ByteBuffer.allocate(10);
        channel.read(buffer);
        byte[] data = buffer.array();
        String msg = new String(data).trim();
        System.out.println("server receive from client: " + msg);
        ByteBuffer outBuffer = ByteBuffer.wrap(msg.getBytes());
        channel.write(outBuffer);
    }
 
    public static void main(String[] args) throws Throwable {
        NIOServer server = new NIOServer();
        server.initServer(8989);
        server.listen();
    }
}
package com.lanhuigu.nio.selector;
 
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.util.Iterator;
 
public class NIOClient {
    private Selector selector;
 
    public void initClient(String ip, int port) throws IOException {
        SocketChannel channel = SocketChannel.open();
        // 设置通道为非阻塞
        channel.configureBlocking(false);
        this.selector = Selector.open();
        // 客户端连接服务器,其实方法执行并没有实现连接,需要在listen()方法中调用channel.finishConnect();才能完成连接
        channel.connect(new InetSocketAddress(ip, port));   //channel连接通了,传统IO是连接上了之后就通过IO流来传输数据。服务端selector就会得到OP_ACCEPT事件select方法就会返回。
        // 将通道管理器和该通道绑定,并为该通道注册SelectionKey.OP_CONNECT事件
        SelectionKey selectionKey1 = channel.register(selector, SelectionKey.OP_CONNECT);  // channel和selector绑定,这个selector关心这个channel的OP_CONNECT事件,selector的通道数组里面只有一个。
        int interestSet = selectionKey1.interestOps();//感兴趣的是8,准备好的是0,
        boolean isInterestedInAccept  = (interestSet & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT;//false
        int isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT;//8
        int isInterestedInRead    = interestSet & SelectionKey.OP_READ;//0
        int isInterestedInWrite   = interestSet & SelectionKey.OP_WRITE;//0
        
        int readySet = selectionKey1.readyOps();//0

    }
 
    @SuppressWarnings("unchecked")
    public void listen() throws Exception {
        while (true) {
            /*
             * 选择一组可以进行I/O操作的事件,放在selector中,客户端的该方法不会阻塞,
             * selector的wakeup方法被调用,方法返回,而对于客户端来说,通道一直是被选中的
             * 这里和服务端的方法不一样,查看api注释可以知道,当至少一个通道被选中时。
            */
            int I = selector.select();//OP_CONNECT事件已有,不会阻塞
            // 获得selector中选中的项的迭代器
            Iterator ite = this.selector.selectedKeys().iterator();
            while (ite.hasNext()) {
                SelectionKey key = (SelectionKey) ite.next();
                // 删除已选的key,以防重复处理
                ite.remove();
                // 连接事件发生
                if (key.isConnectable()) {
                    // 如果正在连接,则完成连接
                    SocketChannel channel = (SocketChannel) key.channel();
                    if (channel.isConnectionPending()) {//该方法返回true表示连接成功
                        channel.finishConnect();
                    }
                    // 设置成非阻塞
                    channel.configureBlocking(false);
                    // 在这里可以给服务端发送信息哦
                    channel.write(ByteBuffer.wrap(new String("hello server!").getBytes()));
                    // 在和服务端连接成功之后,为了可以接收到服务端的信息,需要给通道设置读的权限。
                    SelectionKey selectionKey2 = channel.register(this.selector, SelectionKey.OP_READ); // 获得了可读的事件,此时感兴趣是1准备好是8,selector数组里面还是只有一个SelectionKeyImpl
                    //这个selctor关心这个channel的OP_READ和OP_CONNECT事件,并且只有这一个channel.
                
                    int interestSet = selectionKey2.interestOps();//1
                    boolean isInterestedInAccept  = (interestSet & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT;//fasle
                    int isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT;//0
                    int isInterestedInRead    = interestSet & SelectionKey.OP_READ;//1
                    int isInterestedInWrite   = interestSet & SelectionKey.OP_WRITE;//0
                    
                    int readySet = selectionKey2.readyOps();
                } else if (key.isReadable()) {
                    read(key);
                }
            }
        }
    }
 
    private void read(SelectionKey key) throws Exception {
        SocketChannel channel = (SocketChannel) key.channel();
        // 分配缓冲区
        ByteBuffer buffer = ByteBuffer.allocate(10);
        channel.read(buffer);
        byte[] data = buffer.array();
        String msg = new String(data).trim();
        System.out.println("client receive msg from server:" + msg);
        ByteBuffer outBuffer = ByteBuffer.wrap(msg.getBytes());
        channel.write(outBuffer);
    }
 
    public static void main(String[] args) throws Exception {
        NIOClient client = new NIOClient();
        client.initClient("localhost", 8989);
        client.listen();
    }
}

 

netty的服务端是有2个NioEventLoop,2个线程,2个selector的。  这里服务端只有一个selector既处理服务端channel也处理客户端channel。

 

 

posted @ 2018-08-22 17:59  无天666  阅读(296)  评论(0编辑  收藏  举报