使用 mina 传输大字节数组

转载自：http://seara520.blog.163.com/blog/static/16812769820103214817781/

使用mina传输超过2k以上的数据时(采用tcp方式，如果是UDP方式，好像一次传输的数据不能超过256字节，如果超过mina不会分批次发送，而tcp方式会分批次发送)，mina会自动将这些数据分成多次发送。由于是分批次发送数据，所有客服端在接受数据时，需要等所有的数据接受完之后才能解码，否则无法解码，或者只能读取到部分文件。

以下是一个发送、接受大字节数组的主要代码

服务端向客服端发送字节数组

服务端代码：

编码器：

public class ImageDataEncoder extends ProtocolEncoderAdapter {
	@Override
	public void encode(IoSession session, Object message,
			ProtocolEncoderOutput out) throws Exception {
		CharsetEncoder charset = Charset.forName("UTF-8").newEncoder();
		ImageData image = (ImageData) message;
		IoBuffer buffer = IoBuffer.allocate(2048).setAutoExpand(true);
		buffer.putString(image.getYh(), charset);// 发送数据类型
		buffer.putInt(image.getLength());// 发送字节数组的总长度，共解码时使用
		buffer.put(image.getBimage());// 发送字节数据
		buffer.flip();
		out.write(buffer);
		buffer.free();
	}
}

ImageData.java

package org.bruce.mina.cpp.client;

public class ImageData {
	private static final long serialVersionUID = 1L;
	private String yh = YHConstants.YH_IMG;// 数据类型
	public int length = 0;// 字节数组长度
	private byte[] bimage;// 待发送的字节数组
	private BufferedImage image;// 将字节数组转换成图片文件

	public ImageData() {
	}

	public ImageData(byte[] bimage) {
		this.bimage = bimage;
	}

	public byte[] getBimage() {
		return bimage;
	}

	public BufferedImage getImage() {
		try {
			if (bimage.length > 0) {
				ByteArrayInputStream in = new ByteArrayInputStream(bimage);
				this.image = ImageIO.read(in);
				in.close();
			}
		} catch (RemoteException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return this.image;
	}

	public int getLength() {
		return bimage.length;
	}

	public String getYh() {
		return yh;
	}

	public void setBimage(byte[] bimage) {
		this.bimage = bimage;
	}

	public void setYh(String yh) {
		this.yh = yh;
	}
}

YHConstants.java

package org.bruce.mina.cpp.client;

public class YHConstants {
	public static final int LENGTH = 7;// 命令数据类型
	public static final String YH_CMD = "YH CMD ";// 命令数据类型
	public static final String YH_IMG = "YH IMG ";// 图片数据类型
}

客服端：

解码器：分段发送的解码器一定要继承CumulativeProtocolDecoder ，这个是专门用来实现这种解码的

package org.bruce.mina.cpp.client;

import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.AttributeKey;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.CumulativeProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolDecoderOutput;
import com.seara.socket.message.ImageData;
import com.seara.socket.message.YHConstants;

/**
 * 接收图片数据，由于图片数据比较大，tcp是采用分段式发送，所有需要等所有数据接收完之后才能解码
 * 解码原理：首先读取服务器端发送数据的总长度length，然后与当前的buff中的数据长度matchLength比较，如果matchLength>=
 * length则认为数据发送完毕, 否則将当前的buff保存起来，在下次发送buff之时合并为一个buff，然后在按照以上条件判断
 * @author seara
 */
public class ImageDataDecoder extends CumulativeProtocolDecoder {
	private final AttributeKey CONTEXT = new AttributeKey(this.getClass(), "context");

	@Override
	protected boolean doDecode(IoSession session, IoBuffer buff,
			ProtocolDecoderOutput out) throws Exception {
		CharsetDecoder charset = Charset.forName("UTF-8").newDecoder();
		System.out.println("继续解码......." + buff.remaining());
		// 取出context
		Context ctx = this.getContext(session);// 将contex从session中取出
		int length = ctx.getLength();// 数据总长度
		IoBuffer buffer = ctx.getBuffer();// 保存数据的buffer
		int matchLength = ctx.getMatchLength();// 目前已经发送的数据的总长度
		if (0 == length) {// 第一次取值
			String yh = buff.getString(YHConstants.LENGTH, charset);
			length = buff.getInt();
			matchLength = buff.remaining();
			ctx.setYh(yh);
			ctx.setLength(length);
		} else {
			matchLength += buff.remaining();
		}
		ctx.setMatchLength(matchLength);
		if (buff.hasRemaining()) {// 如果buff中还有数据
			buffer.put(buff);// 添加到保存数据的buffer中
			if (matchLength >= length) {// 如果已经发送的数据的长度>=目标数据的长度,则进行解码
				byte[] b = new byte[length];
				// 一定要添加以下这一段，否则不会有任何数据,因为，在执行buffer.put(buff)时buffer的起始位置已经移动到最后，所有需要将buffer的起始位置移动到最开始
				buffer.flip();
				buffer.get(b);
				ImageData image = new ImageData(b);
				out.write(image);
				System.out.println("解码完成.......");
				return true;
			} else {
				ctx.setBuffer(buffer);
			}
		}
		return false;// 返回false时，解码器就不会执行解码，返回true是在解码完成
	}

	/**
	 * 定义一个内部类，用来封转当前解码器中的一些公共数据，主要是用于大数据解析
	 *
	 * @author seara
	 *
	 */
	public class Context {
		public IoBuffer buffer;
		public int length = 0;
		public int matchLength = 0;
		public String yh = "";

		public Context() {
			this.buffer = IoBuffer.allocate(1024).setAutoExpand(true);
		}

		public int getMatchLength() {
			return matchLength;
		}

		public void setMatchLength(int matchLength) {
			this.matchLength = matchLength;
		}

		public IoBuffer getBuffer() {
			return buffer;
		}

		public void setBuffer(IoBuffer buffer) {
			this.buffer = buffer;
		}

		public int getLength() {
			return length;
		}

		public void setLength(int length) {
			this.length = length;
		}

		public String getYh() {
			return yh;
		}

		public void setYh(String yh) {
			this.yh = yh;
		}
	}

	public Context getContext(IoSession session) {
		Context ctx = (Context) session.getAttribute(CONTEXT);
		if (ctx == null) {
			ctx = new Context();
			session.setAttribute(CONTEXT, ctx);
		}
		return ctx;
	}
}