socket通讯粘包问题

最近在学习netty框架，在学习java nio的时候会遇到socket粘包问题，这里贴出代码，共同学习。

这里解决粘包的方法是(长度+4)，就是发送头信息的时候增加4个字节，通过这4个字节来判断是否粘包，然后再进行拆包

客户端代码：

package com.cg.socket.client;

import java.io.IOException;
import java.net.Socket;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;

/**
 * @author Chalmers 2016年2月24日 下午2:35:39
 */
public class Client {

	public static void main(String[] args) throws UnknownHostException,
			IOException {
		Socket socket = new Socket("127.0.0.1", 9090);

		String message = "hello";

		byte[] bytes = message.getBytes();

		// 设置空间大小为一个存储了长度的int型数据（长度）加上转换后的byte数组
		ByteBuffer buffer = ByteBuffer.allocate(4 + bytes.length);
		// 将长度存入
		buffer.putInt(bytes.length);
		// 将数据存入
		buffer.put(bytes);

		// 转换成字节数组
		byte[] array = buffer.array();

		// 向服务端发送1000次
		for (int i = 0; i < 1000; i++) {
			socket.getOutputStream().write(array);
		}

		// 关闭
		socket.close();
	}
}

package com.cg.socket.client;

import org.jboss.netty.buffer.ChannelBuffer;
import org.jboss.netty.channel.Channel;
import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.handler.codec.frame.FrameDecoder;

/**
 * @author Chalmers 2016年2月24日 下午2:23:49
 */
public class MyDecoder extends FrameDecoder {

	@Override
	protected Object decode(ChannelHandlerContext chc, Channel channel,
			ChannelBuffer buffer) throws Exception {

		// 如果buffer中的可读字节大于4个（即除了长度以外还有数据，因为长度可能是为0的）
		if (buffer.readableBytes() > 4) {

			// 标记，指向当前指针位置，读取数据时使用
			buffer.markReaderIndex();
			// 取得长度
			int len = buffer.readInt();

			// 如果剩余可读字节小于长度的话，则表明发生了拆包现象，那么不对它进行处理
			if (buffer.readableBytes() < len) {
				// 重置标记
				buffer.resetReaderIndex();

				// 返回null，表示等待
				return null;
			}

			// 对数据进行处理
			byte[] bytes = new byte[len];
			buffer.readBytes(bytes);
			// 将数据返回到ServerHandler中进行处理
			return new String(bytes);
		}

		return null;
	}

}

ServerHandler类中的代码

</pre><pre name="code" class="java">package com.cg.socket.client;

import org.jboss.netty.channel.ChannelHandlerContext;
import org.jboss.netty.channel.MessageEvent;
import org.jboss.netty.channel.SimpleChannelHandler;

/**
 * @author Chalmers 2016年2月24日 下午2:22:41
 */
public class ServerHandler extends SimpleChannelHandler {

	int count = 1;

	@Override
	public void messageReceived(ChannelHandlerContext ctx, MessageEvent e)
			throws Exception {
		// 对从MyDecoder中传递过来的数据进行处理
		System.out.println((String) e.getMessage() + "  " + count);
		count++;
	}
}

最后服务器端通信的代码：

package com.cg.socket.server;

import java.net.InetSocketAddress;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import org.jboss.netty.bootstrap.ServerBootstrap;
import org.jboss.netty.channel.ChannelPipeline;
import org.jboss.netty.channel.ChannelPipelineFactory;
import org.jboss.netty.channel.Channels;
import org.jboss.netty.channel.socket.nio.NioServerSocketChannelFactory;
import org.jboss.netty.handler.codec.string.StringEncoder;

import com.cg.socket.client.MyDecoder;
import com.cg.socket.client.ServerHandler;

/**
 * @author Chalmers 2016年2月24日 下午2:21:33
 */
public class Server {

	public static void main(String[] args) {
		ServerBootstrap serverBootstrap = new ServerBootstrap();

		ExecutorService boss = Executors.newCachedThreadPool();
		ExecutorService worker = Executors.newCachedThreadPool();

		serverBootstrap.setFactory(new NioServerSocketChannelFactory(boss,worker));
		serverBootstrap.setPipelineFactory(new ChannelPipelineFactory() {

			public ChannelPipeline getPipeline() throws Exception {
				ChannelPipeline pipeline = Channels.pipeline();
				pipeline.addLast("decoder", new MyDecoder());
				pipeline.addLast("encoder", new StringEncoder());
				pipeline.addLast("handler", new ServerHandler());

				return pipeline;
			}
		});

		serverBootstrap.bind(new InetSocketAddress(9090));
		System.out.println("start...");
	}
}