TCP/NIO示例代码

package rpc;

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.SocketChannel;
import java.nio.charset.Charset;
import java.util.Scanner;

public class NioClient {
	private static Selector selector = null;
	private static Charset charset = Charset.forName("UTF-8");
	private ClientThread thread = new ClientThread();

	public static void main(String[] args) throws IOException {
		selector = Selector.open();
		InetSocketAddress address = new InetSocketAddress("127.0.0.1",30000);
		SocketChannel channel = SocketChannel.open(address);
		channel.configureBlocking(false);
		channel.register(selector, SelectionKey.OP_READ);
		new NioClient().thread.start();
		Scanner scan = new Scanner(System.in);
		while(scan.hasNextLine()){
			String line = scan.nextLine();
			channel.write(charset.encode(line));
		}
	}
	private class ClientThread extends Thread{
		public void run(){
			try {
				while(selector.select()>0){
					for(SelectionKey key : selector.selectedKeys()){
						selector.selectedKeys().remove(key);
						if(key.isReadable()){
							SocketChannel sc = (SocketChannel) key.channel();
							System.out.println(" the remote port to which this socket is connected"+sc.socket().getPort());
							System.out.println(" the Local port to which this socket is connected"+sc.socket().getLocalPort());
							ByteBuffer buf = ByteBuffer.allocate(1024);		
							String content ="";
							while(sc.read(buf)>0){
								sc.read(buf);
								buf.flip();
								content += charset.decode(buf);
							}
							System.out.println("ClientThread:"+content);
							key.interestOps(SelectionKey.OP_READ);
									
						}
					}
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

}

package rpc;

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

public class NioServer {

	public static void main(String[] args) throws IOException {
		//A selectable channel for stream-oriented listening sockets.
		ServerSocketChannel server = ServerSocketChannel.open();
		//implements an IP Socket Address (IP address + port number) 
		InetSocketAddress address = new InetSocketAddress("127.0.0.1", 30000);
		//Binds the ServerSocket to a specific address
		server.socket().bind(address);
		//Adjusts this channel's blocking mode
		server.configureBlocking(false);
		//Opens a selector
		Selector selector = Selector.open();
		//Registers this channel with the given selector, returning a selection key
		server.register(selector, SelectionKey.OP_ACCEPT);
		Charset charset = Charset.forName("UTF-8");
		while(selector.select()>0){
			for(SelectionKey sk : selector.selectedKeys()){
				//Removes the specified element from this set if it is present (optional operation). 
				selector.selectedKeys().remove(sk);
				if(sk.isAcceptable()){
					SocketChannel socketChannel = server.accept();
					socketChannel.configureBlocking(false);
					socketChannel.register(selector, SelectionKey.OP_READ);
					System.out.println(" the remote port to which this socket is connected"+socketChannel.socket().getPort());
					System.out.println(" the Local port to which this socket is connected"+socketChannel.socket().getLocalPort());
					sk.interestOps(SelectionKey.OP_ACCEPT);
				}
				if(sk.isReadable()){
					SocketChannel socketChannel = (SocketChannel) sk.channel();
					System.out.println(" the remote port to which this socket is connected"+socketChannel.socket().getPort());
					System.out.println(" the Local port to which this socket is connected"+socketChannel.socket().getLocalPort());
					ByteBuffer buff = ByteBuffer.allocate(1024);
					String content ="";
					try {
						while(socketChannel.read(buff)>0){
							buff.flip();				
							content += charset.decode(buff);
						}
						System.out.println("server receive:"+content);
						sk.interestOps(SelectionKey.OP_READ);
					} catch (Exception e) {
						sk.cancel();
						if(sk.channel()!=null)
							sk.channel().close();
					}
					//写回
					if(content.length()>0){
						for(SelectionKey key :selector.keys()){
							Channel targetChannel = key.channel();
							if(targetChannel instanceof SocketChannel){
								SocketChannel channel = (SocketChannel) targetChannel;
								channel.write(charset.encode(content+"server写入"));
							}
						}
					}
				}
			}
		}

	}

}