NIO代码实例分享

 

(1)server主要是读取client发过来的信息，并返回一条信息

 

 

/**
 * @author Jeff
 */
public class HelloWorldServer {

    static int               BLOCK        = 1024;
    static String            name         = "";
    protected Selector       selector;
    protected ByteBuffer     clientBuffer = ByteBuffer.allocate(BLOCK);
    protected CharsetDecoder decoder;
    static CharsetEncoder    encoder      = Charset.forName("GB2312").newEncoder();

    public HelloWorldServer(int port) throws IOException{
        selector = this.getSelector(port);
        Charset charset = Charset.forName("GB2312");
        decoder = charset.newDecoder();
    }

    // 获取Selector
    protected Selector getSelector(int port) throws IOException {
        ServerSocketChannel server = ServerSocketChannel.open();
        Selector sel = Selector.open();
        server.socket().bind(new InetSocketAddress(port));
        server.configureBlocking(false);
        server.register(sel, SelectionKey.OP_ACCEPT);
        return sel;
    }

    // 监听端口
    public void listen() {
        try {
            for (;;) {
                selector.select();
                Iterator iter = selector.selectedKeys().iterator();
                while (iter.hasNext()) {
                    SelectionKey key = (SelectionKey) iter.next();
                    iter.remove();
                    process(key);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 处理事件
    protected void process(SelectionKey key) throws IOException {
        if (key.isAcceptable()) { // 接收请求
            ServerSocketChannel server = (ServerSocketChannel) key.channel();
            SocketChannel channel = server.accept();
            // 设置非阻塞模式
            channel.configureBlocking(false);
            channel.register(selector, SelectionKey.OP_READ);
        } else if (key.isReadable()) { // 读信息
            SocketChannel channel = (SocketChannel) key.channel();
            int count = channel.read(clientBuffer);
            if (count > 0) {
                clientBuffer.flip();
                CharBuffer charBuffer = decoder.decode(clientBuffer);
                name = charBuffer.toString();
                // System.out.println(name);
                SelectionKey sKey = channel.register(selector, SelectionKey.OP_WRITE);
                sKey.attach(name);
            } else {
                channel.close();
            }

            clientBuffer.clear();
        } else if (key.isWritable()) { // 写事件
            SocketChannel channel = (SocketChannel) key.channel();
            String name = (String) key.attachment();

            ByteBuffer block = encoder.encode(CharBuffer.wrap("Hello !" + name));

            channel.write(block);

            // channel.close();

        }
    }

    public static void main(String[] args) {
        int port = 8888;
        try {
            HelloWorldServer server = new HelloWorldServer(port);
            System.out.println("listening on " + port);

            server.listen();

        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

 

 

(2)client端代码

 

 

 /** 
  *  
  * @author Jeff 
  * 
  */  
 public class HelloWorldClient {  
   
     static int SIZE = 10;  
     static InetSocketAddress ip = new InetSocketAddress("localhost", 8888);  
     static CharsetEncoder encoder = Charset.forName("GB2312").newEncoder();  
   
     static class Message implements Runnable {  
         protected String name;  
         String msg = "";  
   
         public Message(String index) {  
             this.name = index;  
         }  
   
         public void run() {  
             try {  
                 long start = System.currentTimeMillis();  
                 //打开Socket通道  
                 SocketChannel client = SocketChannel.open();  
                 //设置为非阻塞模式  
                 client.configureBlocking(false);  
                 //打开选择器  
                 Selector selector = Selector.open();  
                 //注册连接服务端socket动作  
                 client.register(selector, SelectionKey.OP_CONNECT);  
                 //连接  
                 client.connect(ip);  
                 //分配内存  
                 ByteBuffer buffer = ByteBuffer.allocate(8 * 1024);  
                 int total = 0;  
   
                 _FOR: for (;;) {  
                     selector.select();  
                     Iterator iter = selector.selectedKeys().iterator();  
   
                     while (iter.hasNext()) {  
                         SelectionKey key = (SelectionKey) iter.next();  
                         iter.remove();  
                         if (key.isConnectable()) {  
                             SocketChannel channel = (SocketChannel) key  
                                     .channel();  
                             if (channel.isConnectionPending())  
                                 channel.finishConnect();  
                             channel  
                                     .write(encoder  
                                             .encode(CharBuffer.wrap(name)));  
   
                             channel.register(selector, SelectionKey.OP_READ);  
                         } else if (key.isReadable()) {  
                             SocketChannel channel = (SocketChannel) key  
                                     .channel();  
                             int count = channel.read(buffer);  
                             if (count > 0) {  
                                 total += count;  
                                 buffer.flip();  
   
                                 while (buffer.remaining() > 0) {  
                                     byte b = buffer.get();  
                                     msg += (char) b;  
                                       
                                 }  
   
                                 buffer.clear();  
                             } else {  
                                 client.close();  
                                 break _FOR;  
                             }  
                         }  
                     }  
                 }  
                 double last = (System.currentTimeMillis() - start) * 1.0 / 1000;  
                 System.out.println(msg + "used time :" + last + "s.");  
                 msg = "";  
             } catch (IOException e) {  
                 e.printStackTrace();  
             }  
         }  
     }  
   
     public static void main(String[] args) throws IOException {  
       
         String names[] = new String[SIZE];  
   
         for (int index = 0; index < SIZE; index++) {  
             names[index] = "jeff[" + index + "]";  
             new Thread(new Message(names[index])).start();  
         }  
       
     }  
 } 

 

 

转载地址：http://ecjtu05.blog.sohu.com/119571227.html