JAVA socket编程之NIO技术_JAVA_编程开发_程序员俱乐部

class="java">
package nio.socket;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
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;

import org.junit.Test;

public class Server {

	@Test
	public void server() throws IOException {
		//selector就是您注册对各种io事件的兴趣的地方 而且当那些事件发生时 就是这个对象告诉您所发生的事情
		Selector selector = Selector.open();
		//开启一个ServerSocketChannel通道
		ServerSocketChannel ssc = ServerSocketChannel.open();
		//绑定地址和端口
		ssc.bind(new InetSocketAddress("localhost", 8080));
		// 设置为非阻塞式的IO 
		ssc.configureBlocking(false);
		//注册感兴趣的
		ssc.register(selector, SelectionKey.OP_ACCEPT);
		//设置一个缓冲
		ByteBuffer buffer = ByteBuffer.allocate(1024);
		//接收客户端过来的通道
		SocketChannel sc = null;
		while (true) {
			selector.select();  //查询事件，监听一旦有我们注册的感兴趣的事件发生的时候便会激活
			Set<SelectionKey> selectedKeys = selector.selectedKeys(); //获取到触发的事件集合
			SelectionKey key = null;
			Iterator<SelectionKey> iterator = selectedKeys.iterator();
			for (;iterator.hasNext();) {//循环
				key = iterator.next();
				if(key.isAcceptable()){ //表示触发的事件是新连接
					ServerSocketChannel newssc = (ServerSocketChannel) key.channel();
					 sc = newssc.accept(); //获取到客户端过来的通道
					 sc.configureBlocking(false); //设置为非阻塞式IO
					 sc.register(selector, SelectionKey.OP_READ);//注册读取事件
					 iterator.remove(); //事件集合中删除
					 System.out.println("有新的连接");
				}else if(key.isReadable()){ //表示读取事件 在连接时候注册了读取事件 在这里处理
					sc = (SocketChannel) key.channel(); //得到客户端的通道
					while(true){
						buffer.clear(); //如果信息量超过预设的缓存就会便会批量读取 重复使用 这里必须清理一下
						int read = sc.read(buffer);
						if(read == -1 || read == 0)break; //如果读取完了 则退出循环
						if(read > 0){
							byte[] array = buffer.array();
							System.out.println("接受到的数据"+new String(array));
						}
					}
					buffer.clear();
					buffer.put("这是我的测试".getBytes());
					buffer.flip();//flip方法将Buffer从写模式切换到读模式
					sc.write(buffer);
					sc.close();//使用完毕关闭客户端的通道
					System.out.println("操作结束==============");
					iterator.remove();
				}
				
			}
		}

	}

}