nio server

来源：互联网发布：mac os 查看本机ip 编辑：程序博客网时间：2024/06/05 19:53

import java.net.InetSocketAddress;
import java.net.Socket;
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.nio.charset.CharsetDecoder;
import java.nio.charset.CharsetEncoder;
import java.util.Iterator;

public class X {

/**
* @param args
*/
public static void main(String[] args) {
  // TODO Auto-generated method stub
    String[] strs = null;
    Selector selector = null;
    try {
     selector = Selector.open();
     ServerSocketChannel server = ServerSocketChannel.open();
     server.socket().bind(new InetSocketAddress(8080));
     server.configureBlocking(false);
     server.register(selector, SelectionKey.OP_ACCEPT);
    } catch (Exception e) {
     e.printStackTrace();
     System.exit(1);
    }

    int index = 0;
    SelectionKey key = null;

    Charset charset = Charset.forName("UTF-8");
    CharsetEncoder encoder = charset.newEncoder();
    CharsetDecoder decoder = charset.newDecoder();
    CharBuffer charBuffer = CharBuffer.allocate(1024);
    ByteBuffer buffer = ByteBuffer.allocate(1024 * 10);
    StringBuffer sBuffer = new StringBuffer();

    while(true) {
     try {
      index = selector.select();
      if(index > 0) {
       Iterator it = selector.selectedKeys().iterator();

       while(it.hasNext()) {
        key = (SelectionKey) it.next();
        it.remove();

        if(key.isValid()) {
         if(key.isAcceptable()) {
          System.out.println("accept ... ");
          Socket socket = ((ServerSocketChannel)key.channel()).accept().socket();
          SocketChannel sc = socket.getChannel();
          sc.configureBlocking(false);
          sc.register(key.selector(), SelectionKey.OP_READ);
         } else if(key.isReadable()) {
          System.out.println("read ... ");
          SocketChannel sc = (SocketChannel) key.channel();

          while(sc.read(buffer) > 0) {
           buffer.flip();
           decoder.decode(buffer, charBuffer, false);
           charBuffer.flip();
           sBuffer.append(charBuffer).append("/r/n");
           //System.out.println(charBuffer);
           buffer.clear();
           charBuffer.clear();
          }
          buffer.flip();
          strs = sBuffer.toString().split("/r/n");
          sc.configureBlocking(false);
          sc.register(key.selector(), SelectionKey.OP_WRITE);
         } else if(key.isWritable()) {
          System.out.println("write ... ");
          SocketChannel sc = (SocketChannel) key.channel();

          if(strs != null) {
           charBuffer.clear();
           charBuffer.flip();
           System.out.println("length:"+strs.length);
           for(int i=0;i<strs.length;i++) {
            sc.write(encoder.encode(charBuffer.wrap(strs[i]+"/r/n")));
            //sc.write(encoder.encode(charBuffer.wrap("Hello/r/n")));
            System.out.println(charBuffer);
            charBuffer.clear();
            charBuffer.flip();
           }
           //sc.write(encoder.encode(charBuffer.wrap("/r/n")));
           charBuffer.clear();
           charBuffer.flip();
          }
          sc.finishConnect();
          sc.close();
         }
        }
       }
      }
     } catch (Exception e) {
      e.printStackTrace();
      System.exit(1);
     }
    }
}

}