JAVA 网络编程(2) SOCKET 非阻塞NIO示例

server:

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.nio.charset.Charset;
import java.util.Iterator;
import java.util.logging.Logger;

/*
 * 传统的阻塞的NIO server
 * OP_READ	读操作
 * OP_WRITE	写操作
 * OP_CONNECT	链接操作
 * OP_ACCEPT	请求操作
 */
public class NIOServer {
	private static class NIOHandler {
		private int bufferSize = 1024;
		private String localCharset = "UTF-8";

		public NIOHandler() {
		}

		public NIOHandler(int bufferSize) {
			this(bufferSize, null);
		}

		public NIOHandler(String localCharset) {
			this(-1, localCharset);
		}

		public NIOHandler(int bufferSize, String localCharset) {
			if (0 < bufferSize) {
				this.bufferSize = bufferSize;
			}
			if (null != localCharset) {
				this.localCharset = localCharset;
			}
		}

		public void handlerAccept(SelectionKey key) throws Exception {
			SocketChannel channel = ((ServerSocketChannel) key.channel()).accept();
			channel.configureBlocking(false);
			channel.register(key.selector(), SelectionKey.OP_READ, ByteBuffer.allocate(bufferSize));
		}

		public void handleRead(SelectionKey key) throws Exception {
			// 获取channel
			SocketChannel channel = (SocketChannel) key.channel();
			ByteBuffer buffer = (ByteBuffer) key.attachment();
			buffer.clear();
			if (-1 == channel.read(buffer)) {// 如果没有读到则关闭
				channel.close();
			} else {
				// 将buffer转换为读状态
				buffer.flip();
				String rsvString = Charset.forName(localCharset).newDecoder().decode(buffer).toString();
				Logger.getGlobal().info("rsvString from client:" + rsvString);
				// 返回数据给客户端
				String sendString = "rsvd data:" + rsvString;
				buffer = ByteBuffer.wrap(sendString.getBytes(localCharset));
				channel.write(buffer);
				//channel.close();
			}
		}
	}

	public static void main(String args[]) throws Exception {
		ServerSocketChannel channel = ServerSocketChannel.open();
		channel.bind(new InetSocketAddress(8080));
		channel.configureBlocking(false);// 设置为非阻塞模式
		Selector sel = Selector.open();
		channel.register(sel, SelectionKey.OP_ACCEPT);// 请求操作
		// 创建处理器
		NIOHandler handler = new NIOHandler(1024);
		while (true) {
			// 等待请求，每次等待阻塞3s，超过3s后县城继续向下运行，如果传入0或者不传参数则一直阻塞
			if (0 == sel.select(15000)) {
				Logger.getGlobal().info("超时。。。。。。。");
				continue;
			}
			Logger.getGlobal().info("处理请求。。。。。。。");
			// 获取处理的SelectionKey
			Iterator<SelectionKey> iter = sel.selectedKeys().iterator();
			while (iter.hasNext()) {
				SelectionKey key = iter.next();
				try {
					// 当收到链接请求时
					if (key.isAcceptable()) {
						handler.handlerAccept(key);
					}
					// 读数据
					if (key.isReadable()) {
						handler.handleRead(key);
					}
				} catch (Exception e) {
					e.printStackTrace();
					continue;
				} finally {
					iter.remove();
				}
			}
		}
	}
}

client:

//001传统的阻塞式的socket
public class Client {

	public static void main(String[] args) {
		try (Socket socket = new Socket("127.0.0.1", 8080);
				PrintWriter pw = new PrintWriter(socket.getOutputStream());
				BufferedReader rd = new BufferedReader(new InputStreamReader(socket.getInputStream()));
				Scanner r = new Scanner(System.in);) {
			while (true) {
				String s = r.nextLine();
				pw.println(s);
				pw.flush();
				String line = rd.readLine();
				Logger.getGlobal().info("rsvd from server:" + line);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}