java tcp ip组包_Java——基于java自身包实现消息系统间的通信（TCP/IP+NIO）

/**

* Created by LiuHuiChao on 2016/11/15.

* description:based on TCP/IP+NIO to deliver the message

*/

public class TCP_IP_NIO {

@Test

public void clientStart() throws IOException {

SocketChannel channel=SocketChannel.open();

channel.configureBlocking(false);//设置为非阻塞方式

SocketAddress remote=new InetSocketAddress("127.0.0.1",8888);

channel.connect(remote);

Selector selector= Selector.open();

channel.register(selector, SelectionKey.OP_CONNECT);

/**阻塞至有感兴趣的IO事件发生，或到达超时时间，如果希望一直等至有感兴趣的IO事件发生，可调用无参数select方法，

* 如果希望不阻塞直接返回目前是否有感兴趣的事件发生，可以调用selectNow方法

* */

int nkeys=selector.select();//如果nkeys大于0，说明有感兴趣的IO事件发生

SelectionKey selectionKey=null;

if(nkeys>0){

Set keys=selector.selectedKeys();

for(SelectionKey key : keys){

//对于发生连接的事件

if(key.isConnectable()){

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

sc.configureBlocking(false);

/**注册感兴趣的IO读事件，通常不直接注册写事件，在发送缓冲区未满的情况下，一直是可写的，

* 因此，如注册了写事件，而又不用写数据，很容易造成CUP消耗100%的情况；

* */

selectionKey=sc.register(selector,SelectionKey.OP_READ);

sc.finishConnect();

}else if(key.isReadable()){/**有流可读*/

ByteBuffer byteBuffer=ByteBuffer.allocate(1024);

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

int readBytes=0;

try{

int ret=0;

try{

/**读取目前可读的流，sc.read返回的为成功复制到bytebuffer中的字节数；

* 此步骤为阻塞操作，值可能为0；当已经是流的结尾时，返回-1

* */

while((ret=sc.read(byteBuffer))>0){

readBytes+=ret;

}

}finally{

byteBuffer.flip();

}

}finally{

if(byteBuffer!=null){

byteBuffer.clear();

}

}

}else if(key.isWritable()){/**可写入流*/

//取消对OP_WRITE事件的注册

key.interestOps(key.interestOps()&(~selectionKey.OP_WRITE));

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

/**此步骤为阻塞操作，直到写入操作系统发送缓冲区或网路IO出现异常，返回的为成功写入的字节数，当操作系统的发送缓冲区已满，此处返回0*/

ByteBuffer byteBuffer=ByteBuffer.allocate(1024);

sc.read(byteBuffer);

int writtenedSize=sc.write(byteBuffer);

//如未写入，则继续注册感兴趣的OP_WRITE事件

if(writtenedSize==0){

key.interestOps(key.interestOps() | selectionKey.OP_WRITE);

}

}

}

selector.selectedKeys().clear();

}

}

@Test

public void serverStart() throws IOException {

ServerSocketChannel ssc=ServerSocketChannel.open();

ServerSocket serverSocket=ssc.socket();

//绑定要监听的端口

serverSocket.bind(new InetSocketAddress(8888));

ssc.configureBlocking(false);

Selector selector= Selector.open();

//注册感兴趣的事件连接

ssc.register(selector,SelectionKey.OP_ACCEPT);

/**

* 之后采取和客户端相同的方式对selector.select进行轮询。。。但是要增加一个key.isAcceptable的处理。。。

* */

}

}