java nio 实现客户端与服务端读写

刚开始写，有待后续完善

服务端代码：

package com.geng.test3;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectableChannel;
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 Server {
	private static Selector selector ;

	public  static void init() throws IOException{
		selector = Selector.open();
		ServerSocketChannel ssc = ServerSocketChannel.open();
		ssc.configureBlocking(false);
		ssc.bind(new InetSocketAddress(InetAddress.getLocalHost(), 8099));
		ssc.register(selector, SelectionKey.OP_ACCEPT);
	}
	
	public static void listen() throws IOException{
		while(true){
			int select = selector.select();
			if(select == 0){
				continue;
			}
			Set<SelectionKey> selectedKeys = selector.selectedKeys();
			Iterator<SelectionKey> it = selectedKeys.iterator();
			while(it.hasNext()){
				SelectionKey key = it.next();
				// 由于select操作只管对selectedKeys进行添加，所以key处理后我们需要从里面把key去掉  
				it.remove();
				if(key.isAcceptable()){
					ServerSocketChannel serverChannel = (ServerSocketChannel) key.channel();
					// 得到与客户端的套接字通道  
					SocketChannel channel = serverChannel.accept();
					channel.configureBlocking(false);
					// 同样将于客户端的通道在selector上注册，,可以通过key获取关联的选择器
					channel.register(selector, SelectionKey.OP_READ|SelectionKey.OP_WRITE);
				}else if(key.isReadable()){
					read(key);
					 // 改变自身关注事件，可以用位或操作|组合时间  
					key.interestOps(SelectionKey.OP_READ|SelectionKey.OP_WRITE);
				}else if(key.isWritable()){
					write(key);
					 // 改变自身关注事件，可以用位或操作|组合时间  
					key.interestOps(SelectionKey.OP_READ|SelectionKey.OP_WRITE);
				}
			}
		}
	}

	private static void write(SelectionKey key) throws IOException {
		SocketChannel channel = (SocketChannel) key.channel();
		channel.write(ByteBuffer.wrap("server write".getBytes()));
		
	}

	private static void read(SelectionKey key) throws IOException {
		SocketChannel channel = (SocketChannel) key.channel();
		ByteBuffer buffer = ByteBuffer.allocate(1024);
		int count = channel.read(buffer);
		byte[] bytes = new byte[count];
		buffer.flip();
		buffer.get(bytes);
		//System.out.println(new String(bytes));
		
	}
	
	public static void main(String[] args) throws IOException {
		init();
		listen();
	}
}

客户端代码：

package com.geng.test3;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectableChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.util.Iterator;
import java.util.Set;

public class Client {

	private static Selector selector;
	
	public static void init() throws IOException{
		selector = Selector.open();
		SocketChannel sc = SocketChannel.open();
		sc.configureBlocking(false);
		sc.connect(new InetSocketAddress(InetAddress.getLocalHost(), 8099));
		
		sc.register(selector, SelectionKey.OP_CONNECT);
	}
	
	public static void listen() throws IOException{
		while(true){
			selector.select();
			Set<SelectionKey> selectedKeys = selector.selectedKeys();
			Iterator<SelectionKey> it = selectedKeys.iterator();
			while(it.hasNext()){
				SelectionKey key = it.next();
				it.remove();
				
				if(key.isConnectable()){
					SocketChannel channel = (SocketChannel) key.channel();
					if(channel.finishConnect()){
						channel.configureBlocking(false);
						channel.register(selector, SelectionKey.OP_READ|SelectionKey.OP_WRITE);
					}
				}else if(key.isReadable()){
					read(key);
					 // 改变自身关注事件，可以用位或操作|组合时间  
					key.interestOps(SelectionKey.OP_READ|SelectionKey.OP_WRITE);
				}else if(key.isWritable()){
					write(key);
					 // 改变自身关注事件，可以用位或操作|组合时间  
					key.interestOps(SelectionKey.OP_READ|SelectionKey.OP_WRITE);
				}
				
			}
		}
	}
	
	private static void write(SelectionKey key) throws IOException {
		SocketChannel channel = (SocketChannel) key.channel();
		channel.write(ByteBuffer.wrap("server write".getBytes()));
		
	}

	private static void read(SelectionKey key) throws IOException {
		SocketChannel channel = (SocketChannel) key.channel();
		ByteBuffer buffer = ByteBuffer.allocate(1024);
		int count = channel.read(buffer);
		byte[] bytes = new byte[count];
		buffer.flip();
		buffer.get(bytes);
		System.out.println(new String(bytes));
		
	}
}