java socket NIO模式

为什么80%的码农都做不了架构师？>>>   

public class TimeServer {

public static void main(String[] args) {

int port = 8080;

if(null != args && args.length > 0) {

try {

port = Integer.parseInt(args[0]);

} catch (NumberFormatException e) {

e.printStackTrace();

}

}

MultiplexerTimeServer timeServer = new MultiplexerTimeServer(port);

new Thread(timeServer, "NIO-MultiplexerTimeServer-001").start();

}

}

---

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.Date;

import java.util.Iterator;

import java.util.Set;

public class MultiplexerTimeServer implements Runnable {

private Selector selector;

private ServerSocketChannel serverChannel;

private boolean stop; 

/**

* 初始化多路复用器,绑定监听端口

* @param  port

*/

public MultiplexerTimeServer(int port) {

try {

//1、打开ServerSocketChannel

serverChannel = ServerSocketChannel.open();

serverChannel.configureBlocking(false);

//2、绑定监听

serverChannel.socket().bind(new InetSocketAddress(port), 1024);

//3、创建Selector

selector = Selector.open();

//5、将ServerSocketChannel注册至Reactor线程的多路复用器selector上

serverChannel.register(selector, SelectionKey.OP_ACCEPT);

System.out.println("the timeserver is start in port "+port);

} catch (IOException e) {

e.printStackTrace();

System.exit(-1);

} finally{

}

}

public void stop() {

this.stop = true;

}

@Override

public void run() {

while(!stop) {

try {

selector.select(1000);

Set<SelectionKey> selectedKeys = selector.selectedKeys();

Iterator<SelectionKey> it = selectedKeys.iterator();

SelectionKey key = null;

while(it.hasNext()) {

key = it.next();

it.remove();

try {

handlerInput(key);

} catch (Exception e) {

//e.printStackTrace();

if(null != key) {

key.cancel();

if(null != key.channel()) {

key.channel().close();

}

}

} 

}

} catch (IOException e) {

e.printStackTrace();

}

}

//多路复用器关闭

if(null != selector) {

try {

selector.close();

} catch (IOException e) {

e.printStackTrace();

}

}

}

private void handlerInput(SelectionKey key) throws IOException {

if(key.isValid()) {

//处理新接入的请求消息

if(key.isAcceptable()) {

ServerSocketChannel ssc = (ServerSocketChannel)key.channel();

SocketChannel sc = ssc.accept();

sc.configureBlocking(false);

sc.register(selector, SelectionKey.OP_READ);

}

if(key.isReadable()) {

SocketChannel sc = (SocketChannel)key.channel();

ByteBuffer readBuffer = ByteBuffer.allocate(1024);

int readBytes = sc.read(readBuffer);

if(readBytes > 0) {

readBuffer.flip();

byte[] bytes = new byte[readBuffer.remaining()];

readBuffer.get(bytes);

String body = new String(bytes, "UTF-8");

System.out.println("The time server receive order :"+body);

String currentTime = "QUERY TIME ORDER".equalsIgnoreCase(body) ? new Date(System.currentTimeMillis()).toString() : "BAD ORDER";

doWrite(sc, currentTime);

}else if(readBytes < 0){

//对端链路关闭

key.cancel();

sc.close();

}else {

; //读取0字节,忽略

}

}

}

}

private void doWrite(SocketChannel channel, String response) throws IOException {

if(null != response && response.trim().length() > 0){

byte[] bytes = response.getBytes();

ByteBuffer writeBuffer = ByteBuffer.allocate(bytes.length);

writeBuffer.put(bytes);

writeBuffer.flip();

channel.write(writeBuffer);

}

}

}

---

public class TimeClient {

public static void main(String[] args) {

int port = 8080;

if(null != args && args.length > 0) {

try {

port = Integer.valueOf(args[0]);

} catch (NumberFormatException e) {

e.printStackTrace();

}

}

new Thread(new TimeClientHandle("127.0.0.1", port), "TimeClient-001").start();

}

}

---

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.util.Iterator;

import java.util.Set;

public class TimeClientHandle implements Runnable {

private String host;

private int port;

private Selector selector;

private SocketChannel socketChannel;

private boolean stop;

public TimeClientHandle(String host, int port) {

this.host = host == null ? "127.0.0.1" : host;

this.port = port;

try {

selector = Selector.open();

socketChannel = SocketChannel.open();

socketChannel.configureBlocking(false);

} catch (IOException e) {

e.printStackTrace();

System.exit(1);

}

}

@Override

public void run() {

try {

doConnect();

} catch (IOException e) {

e.printStackTrace();

System.exit(1);

}

while(!stop) {

try {

selector.select(1000);

Set<SelectionKey> selectedKeys = selector.selectedKeys();

Iterator<SelectionKey> it = selectedKeys.iterator();

SelectionKey key = null;

while(it.hasNext()) {

key = it.next();

it.remove();

try {

handlerInput(key);

} catch (Exception e) {

//e.printStackTrace();

if(null != key) {

key.cancel();

if(null != key.channel()) {

key.channel().close();

}

}

} 

}

} catch (IOException e) {

e.printStackTrace();

System.exit(1);

}

}

//多路复用器关闭

if(null != selector) {

try {

selector.close();

} catch (IOException e) {

e.printStackTrace();

}

}

}

private void handlerInput(SelectionKey key) throws IOException {

if(key.isValid()) {

SocketChannel sc = (SocketChannel)key.channel();

if(key.isConnectable()) {

if(sc.finishConnect()) {

sc.register(selector,SelectionKey.OP_READ);

doWrite(sc);

}else{

System.exit(1);

}

}

if(key.isReadable()) {

ByteBuffer readBuffer = ByteBuffer.allocate(1024);

int readBytes = sc.read(readBuffer);

if(readBytes > 0) {

readBuffer.flip();

byte[] bytes = new byte[readBuffer.remaining()];

readBuffer.get(bytes);

String body = new String(bytes, "UTF-8");

System.out.println("Now is : "+body);

this.stop = true;

}else if(readBytes < 0) {

key.cancel();

sc.close();

}else {

; //读到0字节,忽略

}

}

}

}

/**

* 创建连接

* @throws  IOException

*/

private void doConnect() throws IOException {

if(socketChannel.connect(new InetSocketAddress(host, port))) {

socketChannel.register(selector, SelectionKey.OP_READ);

doWrite(socketChannel);

}else {

socketChannel.register(selector, SelectionKey.OP_CONNECT);

}

}

/**

* 处理结果

* @param  sc

* @throws IOException

*/

private void doWrite(SocketChannel sc) throws IOException {

byte[] req = "QUERY TIME ORDER".getBytes();

ByteBuffer writeBuffer = ByteBuffer.allocate(req.length);

writeBuffer.put(req);

writeBuffer.flip();

sc.write(writeBuffer);

if(!writeBuffer.hasRemaining()) {

System.out.println("Send Order 2 Server succeed.");

}

}

}

感觉NIO编程比BIO编程要复杂，但性能很好。

转载于:https://my.oschina.net/chaun/blog/391642