非阻塞模式ServerSocketChannel 聊天室服务器端

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Iterator;
import java.util.Set;

/**
 * 非阻塞的Echoserver
 *  在这个非阻塞的模式下，Echoserver只用启动一个主线程就能同事处理其他三件事：
 * 1.接收客户连接
 * 2.接收客户发送的数据
 * 3.向客户发回响应数据
 *
 * @author Administrator
 *
 */
public class NZSEchoServer {
    private Selector selector = null;
    private ServerSocketChannel serverSocketChannel = null;
    private int port = 8000;

    private Charset charset = Charset.forName("GBK");

    /**
     * 构造方法负责启动服务器，绑定端口
     *
     * @throws IOException
     */
    private NZSEchoServer() throws IOException {
        // 创建一个selector 对象
        selector = Selector.open();
        // 创建serverSocketChannel对象
        serverSocketChannel = ServerSocketChannel.open();
        // 设置参数 使得在同一个主机上关闭了服务器程序，紧接着再启动该服务器程序时可以顺利绑定到相同的端口
        serverSocketChannel.socket().setReuseAddress(true);
        // 使erverSocketChannel工作处于非阻塞模式
        serverSocketChannel.configureBlocking(false);
        // 把服务器进程与一个本地端口绑定
        serverSocketChannel.socket().bind(new InetSocketAddress(port));
        System.out.println("服务器启动！");
    }

    /**
     * 负责开头说的三件事 * 1.接收客户连接 2.接收客户发送的数据 3.向客户发回响应数据
     *
     * @throws IOException
     */
    public void service() throws IOException {
        // serverSocketChannel向Selector注册接收连接就绪事件
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        while (selector.select() > 0) {
            // 获得Selector的selected-keys集合
            Set readyKeys = selector.selectedKeys();
            Iterator it = readyKeys.iterator();

            //从集合中依次取出SelectionKey对象,判断是那种事件发生，然后进行处理
            while (it.hasNext()) {
                SelectionKey key = null;
                try {
                    // 处理selectionkey 取出第一个selectionkey
                    key = (SelectionKey) it.next();
                    // 把selectionKey从selected-key集合中删除
                    it.remove();

                    // 这个key 标识连接就绪事件 处理
                    if (key.isAcceptable()) {
                        //获得与SelectionKey相连的ServerSocketChannel
                        ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
                        //获得与客户端连接的SocketChannel
                        SocketChannel sockChannel = ssc.accept();
                        System.out.println("接收到客户端连接，来自：" + sockChannel.socket().getInetAddress() + ":"
                                + sockChannel.socket().getPort());

                        //设置SocketChannel为非阻塞模式
                        sockChannel.configureBlocking(false);
                        //创建一个用于存放用户发送来的数据的缓冲区
                        ByteBuffer buffer = ByteBuffer.allocate(1024);
                        //SocketChannel向Selector注册读就绪事件和写就绪事件   关联了一个buffer
                        //这个byteBuffer将作为附件与新建的selectionKey对象关联
                        sockChannel.register(selector, SelectionKey.OP_READ|SelectionKey.OP_WRITE,buffer);

                    }
                    // 这个key 标识读就绪事件 处理
                    if (key.isReadable()) {
                        receive(key);
                    }
                    // 这个key 标识写就绪事件 处理
                    if (key.isWritable()) {
                        send(key);
                    }

                } catch (Exception e) {
                    e.printStackTrace();
                    try {
                    //使得这个selctionkey失效
                    //使得selectory不再监控这个selectionkey感兴趣的事件
                        if (key != null) {
                            key.cancel();
                            key.channel().close();
                        }
                    } catch (Exception e2) {
                        e.printStackTrace();
                    }

                }

            }

        }

    }
    /**
     * 处理写就绪事件
     * @param key
     * @throws IOException
     */
    private void send(SelectionKey key) throws IOException {
        //获取与selectionKey相关联的附件
        ByteBuffer buffer = (ByteBuffer) key.attachment();
        SocketChannel socketChannel = (SocketChannel) key.channel();
//        flip的作用有两个：
//        1. 把limit设置为当前的position值
//        2. 把position设置为0
//        然后处理的数据就是从position到limit直接的数据，也就是你刚刚读取过来的数据
        buffer.flip();
        //按照gbk编码，把buffer中的字节转换成字符串
        String data = decode(buffer);
        //如果还没有读到一行数据，之间返回
        if (data.indexOf("\r\n") == -1)
            return;

        //已经有一行以上数据，截取一行数据
        String outputData = data.substring(0, data.indexOf("\n") + 1);
        System.out.println(outputData);
        //把输出的字符串按照gbk编码，转换成字节，把他反正outputBuffer中
        ByteBuffer outputBuffer = encode("echo" + outputData);

        //outputBuffer.hasRemaining()判断是否还有未处理的字节
        //非阻塞模式下不确保write方法一次就把outputBuffer所有字节发送完，只是奉行能发多少就发送多少的原则，所以我们要采用循环
        while (outputBuffer.hasRemaining()) {
            socketChannel.write(outputBuffer);
        }

            //我觉得  相当于移动 栈指针 然后删除没有指向的数据段
            ByteBuffer temp = encode(outputData);
            //设置buffer的位置为temp的极限
            buffer.position(temp.limit());
            //删除buffer中已经处理的数据
            buffer.compact();
            //如果已经输出了字符串  "bye\r\n" ,就使selectionKet失效，并关闭SocketChannel
            if (outputData.equals("bye\r\n")) {
                key.cancel();
                socketChannel.close();
                System.out.println("关闭与客户端的连接！");
            }

    }

    /**
     *
     * 处理读就绪事件
     * 把收到的数据放入buffer
     *
     * @param key
     * @throws IOException
     */
    public void receive(SelectionKey key) throws IOException {
        //获得与SelectionKey关联的附件
        ByteBuffer buffer = (ByteBuffer) key.attachment();
        //获得与SelectionKey关联的Sockethannel
        SocketChannel socketChannel = (SocketChannel) key.channel();
        //创建一个byteBuffer,用于存放读到的数据
        ByteBuffer readBuffer = ByteBuffer.allocate(32);
        socketChannel.read(readBuffer);
        readBuffer.flip();

        //把buffer的极限设为容量
        buffer.limit(buffer.capacity());
        //把readBuffer中的内容拷贝到buffer中
        //假定buffer的容量足够大，不会出现缓冲区溢出异常
        buffer.put(readBuffer); // 把读到的数据放到buffer中
    }

    /**
     * 编码  把字符串转换成自己序列
     *
     * @param string
     * @return
     */
    private ByteBuffer encode(String string) {
        return charset.encode(string);
    }

    /**
     * 解码  把字节序列转换为字符串
     *
     * @param buffer
     * @return
     */
    private String decode(ByteBuffer buffer) {
        CharBuffer charBuffer = charset.decode(buffer);
        return charBuffer.toString();
    }
    /**
     * 主程序
     * @param args
     * @throws IOException
     */
    public static void main(String[] args) throws IOException {
        NZSEchoServer server=new NZSEchoServer();
        server.service();
    }
}