java nio

视频

package cn.zvc.nio;

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

public class NIOServer {
	private int blockSize = 4096;
	private ByteBuffer sendBuffer = ByteBuffer.allocate(blockSize);
	
	private ByteBuffer receiveBuffer = ByteBuffer.allocate(blockSize);
	private Selector selector;
	
	private int flag = 1;
	public NIOServer(int port) throws IOException {
		ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
		//设置是否阻塞
		serverSocketChannel.configureBlocking(false);
		ServerSocket serverSocket = serverSocketChannel.socket();
		serverSocket.bind(new InetSocketAddress(port));
		selector = Selector.open();
		
		serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
		
		System.out.println("server start:" + port);
	}
	
	
	public void linsten() throws IOException{
		while(true) {
			selector.select();
			Set<SelectionKey> selectionKeys = selector.selectedKeys();
			Iterator<SelectionKey> it = selectionKeys.iterator();
			while(it.hasNext()) {
				SelectionKey sk = it.next();
				it.remove();
				//业务逻辑
				handleKey(sk);
			}
		}
	}
	
	public void handleKey(SelectionKey sk) throws IOException{
		ServerSocketChannel server = null;
		SocketChannel client = null;
		String reviceText;
		String sendText;
		int count = 0;
		if(sk.isAcceptable()){
			server = (ServerSocketChannel) sk.channel();
			client = server.accept();
			client.configureBlocking(false);
			client.register(selector, sk.OP_READ);
			
		} else if(sk.isReadable()) {
			client = (SocketChannel) sk.channel();
			count = client.read(receiveBuffer);
			if(count>0){
				reviceText = new String(receiveBuffer.array(),0,count);
				System.out.println("服务端接受到客户端的信息："+ reviceText);
				client.register(selector, SelectionKey.OP_WRITE);
			} 
		} else if(sk.isWritable()){
			receiveBuffer.clear();
			client = (SocketChannel) sk.channel();
			sendText = "msg send to client:"+ flag;
			sendBuffer.put(sendText.getBytes());
			sendBuffer.flip();
			client.write(sendBuffer);
			System.out.println("服务端发送给客户端："+ sendText);
		}
				
	}
	public static void main(String[] args) {
		
	}
}