java nio

视频

1package cn.zvc.nio;
2
3import java.io.IOException;
4import java.net.InetSocketAddress;
5import java.net.ServerSocket;
6import java.nio.ByteBuffer;
7import java.nio.channels.ClosedChannelException;
8import java.nio.channels.SelectionKey;
9import java.nio.channels.Selector;
10import java.nio.channels.ServerSocketChannel;
11import java.nio.channels.SocketChannel;
12import java.util.Iterator;
13import java.util.Set;
14
15public class NIOServer {
16	private int blockSize = 4096;
17	private ByteBuffer sendBuffer = ByteBuffer.allocate(blockSize);
18	
19	private ByteBuffer receiveBuffer = ByteBuffer.allocate(blockSize);
20	private Selector selector;
21	
22	private int flag = 1;
23	public NIOServer(int port) throws IOException {
24		ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
25		//设置是否阻塞
26		serverSocketChannel.configureBlocking(false);
27		ServerSocket serverSocket = serverSocketChannel.socket();
28		serverSocket.bind(new InetSocketAddress(port));
29		selector = Selector.open();
30		
31		serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
32		
33		System.out.println("server start:" + port);
34	}
35	
36	
37	public void linsten() throws IOException{
38		while(true) {
39			selector.select();
40			Set<SelectionKey> selectionKeys = selector.selectedKeys();
41			Iterator<SelectionKey> it = selectionKeys.iterator();
42			while(it.hasNext()) {
43				SelectionKey sk = it.next();
44				it.remove();
45				//业务逻辑
46				handleKey(sk);
47			}
48		}
49	}
50	
51	public void handleKey(SelectionKey sk) throws IOException{
52		ServerSocketChannel server = null;
53		SocketChannel client = null;
54		String reviceText;
55		String sendText;
56		int count = 0;
57		if(sk.isAcceptable()){
58			server = (ServerSocketChannel) sk.channel();
59			client = server.accept();
60			client.configureBlocking(false);
61			client.register(selector, sk.OP_READ);
62			
63		} else if(sk.isReadable()) {
64			client = (SocketChannel) sk.channel();
65			count = client.read(receiveBuffer);
66			if(count>0){
67				reviceText = new String(receiveBuffer.array(),0,count);
68				System.out.println("服务端接受到客户端的信息："+ reviceText);
69				client.register(selector, SelectionKey.OP_WRITE);
70			} 
71		} else if(sk.isWritable()){
72			receiveBuffer.clear();
73			client = (SocketChannel) sk.channel();
74			sendText = "msg send to client:"+ flag;
75			sendBuffer.put(sendText.getBytes());
76			sendBuffer.flip();
77			client.write(sendBuffer);
78			System.out.println("服务端发送给客户端："+ sendText);
79		}
80				
81	}
82	public static void main(String[] args) {
83		
84	}
85}