Nio建立socket使用注意事项

最近做区块链DEMO时，用到NIO 来建立socket。 初步体验了该API，总的来说 优缺点并存。 不过使用时需要注意:

1. selectorKey 中OP_WRITE注册

    为什么注册该事件，是因为需要底层缓存区准备就绪，在上一次写操作完成后，或者缓冲区足够大的前提下。已经是就绪的。 但是最好把写操作放在 selectorKey.isWritable() 条件后。

2.为什么客户端向服务端发出了消息，服务端无法收到。 这个问题困扰了很久。 而且socket编程中竟然没有说明...  写完一条后 需要以\r\n结束。否则对端无法知道已经写完成。

3.bytebuffer这个东东的设计有点奇葩。构造方法问题会出现大坑很浪费时间...

实例代码如下:

Server端:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
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.nio.charset.CharsetDecoder;
import java.util.Iterator;
import java.util.Set;


public class Server {
static Selector selector = null;
private static int BLOCK = 4096;
static int flag;
private static ByteBuffer sendbuffer = ByteBuffer.allocate(BLOCK);
private static ByteBuffer receivebuffer = ByteBuffer.allocate(BLOCK);


public static void main(String[] args) {
initServer();
try {
listen();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}


private static void initServer() {
try {
ServerSocketChannel ss = ServerSocketChannel.open();// (8001);
ss.socket().bind(new InetSocketAddress(8001));
ss.configureBlocking(false);
selector = Selector.open();
ss.register(selector, SelectionKey.OP_ACCEPT);
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}


private static void listen() throws IOException {
while (true) {
selector.select();
Set<SelectionKey> selectionKeys = selector.selectedKeys();
Iterator<SelectionKey> iterator = selectionKeys.iterator();
while (iterator.hasNext()) {
SelectionKey selectionKey = iterator.next();
iterator.remove();
handleKey(selectionKey);
}
}
}


private static void handleKey(SelectionKey selectionKey) throws IOException {
ServerSocketChannel server = null;
SocketChannel client = null;
String receiveText;
String sendText;
int count = 0;
if (selectionKey.isAcceptable()) {
server = (ServerSocketChannel) selectionKey.channel();
client = server.accept();
client.configureBlocking(false);
client.register(selector, SelectionKey.OP_READ);
} else if (selectionKey.isReadable()) {
client = (SocketChannel) selectionKey.channel();
receivebuffer.clear();
count = client.read(receivebuffer);
if (count > 0) {
receiveText = new String(receivebuffer.array(), 0, count);
System.out.println("client send:" + receiveText);
client.register(selector, SelectionKey.OP_WRITE);
// back
if (receiveText != null && !"".equals(receiveText)) {
ByteBuffer buffer = ByteBuffer.allocate(256);
if (receiveText.contains("client hello")) {
byte[] backInfo = new String("server hello\r\n").getBytes("UTF-8");
buffer = ByteBuffer.wrap(backInfo);
}
if (receiveText.contains("client ack")) {
byte[] backInfo = new String("server ack\r\n").getBytes("UTF-8");
buffer = ByteBuffer.wrap(backInfo);
}


client.write(buffer);
buffer.rewind();
System.out.println("server send:" + new String(buffer.array()));
}
}
} else if (selectionKey.isWritable()) {
client = (SocketChannel) selectionKey.channel();
client.register(selector, SelectionKey.OP_READ);
}
}
}

Client端:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;


public class ClientTest {


public static void main(String[] args) {


try {
InetAddress address = InetAddress.getLocalHost();
String ip = address.getHostAddress();
Socket socket = new Socket(ip, 8001);
OutputStream os = socket.getOutputStream();// 字节输出流
InputStream is = socket.getInputStream();
BufferedWriter pw = new BufferedWriter(new OutputStreamWriter(os));
BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
pw.write("client hello");
pw.newLine();
pw.flush();
while (true) {


// System.out.println("client say");
String text = br.readLine();
if (text != null) {
if ("server hello".equals(text)) {
pw.write("client ack");
pw.newLine();
pw.flush();
} else if ("server ack".equals(text)) {
// 完成握手
System.out.println("handshake complete");
}
}
Thread.sleep(1000);
}
} catch (UnknownHostException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}


}
}