mina 集成spring 传输文件和文本

package com.xx.mina; 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.ProtocolDecoderOutput; import org.apache.mina.filter.codec.demux.MessageDecoder; import org.apache.mina.filter.codec.demux.MessageDecoderResult; public class BaseMessageDecoder implements MessageDecoder { private AttributeKey CONTEXT = new AttributeKey(getClass(), "context"); private class Context { public int dataType; public boolean isReadName; public int strLength; public byte[] byteStr; public String fileName; public int fileSize; public byte[] byteFile; public int count; public boolean init = false; public void reset() { dataType = 0; byteFile = null; count = 0; strLength = 0; isReadName = false; fileSize = 0; byteStr = null; fileName = null; } } /** * 验证是否可以解析 * */ public MessageDecoderResult decodable(IoSession session, IoBuffer in) { Context context = (Context) session.getAttribute(CONTEXT); if (context == null) {// 建立连接后第�?��传输 context = new Context(); context.dataType = in.getInt(); if (context.dataType == BeanUtil.SEND_TYPE_FILE) { context.strLength = in.getInt();// 名称长度 context.byteStr = new byte[context.strLength];// 名称内容 context.fileSize = in.getInt();// 文件长度 context.byteFile = new byte[context.fileSize];// 文件内容 session.setAttribute(CONTEXT, context); return MessageDecoderResult.OK; } else if (context.dataType == BeanUtil.SEND_TYPE_TXT) { context.strLength = in.getInt();// 名称长度 context.byteStr = new byte[context.strLength];// 名称内容 context.fileSize = 0;// 文件长度 context.byteFile = new byte[context.fileSize];// 文件内容 session.setAttribute(CONTEXT, context); return MessageDecoderResult.OK; } else { return MessageDecoderResult.NOT_OK; } } else {// 建立后第�?��传输之后 if (context.dataType == BeanUtil.SEND_TYPE_FILE || context.dataType == BeanUtil.SEND_TYPE_TXT) { return MessageDecoderResult.OK; } else { return MessageDecoderResult.NOT_OK; } } } public MessageDecoderResult decode(IoSession session, IoBuffer in, ProtocolDecoderOutput outPut) throws Exception { // TODO Auto-generated method stub System.out.println("开始解析"); Context context = (Context) session.getAttribute(CONTEXT); int sendType = context.dataType; if (!context.init) { context.init = true; sendType = in.getInt();// 文件类型 in.getInt();// 文件名长�? System.out.println("xx" + sendType); if (sendType == BeanUtil.SEND_TYPE_FILE) { in.getInt();// 文件大小 } } if (sendType == BeanUtil.SEND_TYPE_FILE) { byte[] byteFile = context.byteFile; int count = context.count; while (in.hasRemaining()) { byte b = in.get();// 整体读取 if (!context.isReadName) {//文件�? context.byteStr[count] = b; if (count == context.strLength - 1) { context.fileName = new String(context.byteStr, BeanUtil.charset); System.out.println(context.fileName); count = -1; context.isReadName = true; } } if (context.isReadName && count != -1) {//文件�? byteFile[count] = b; } count++; } context.count = count; System.out.println("count:" + count); System.out.println("context.fileSize:" + context.fileSize); session.setAttribute(CONTEXT, context); if (context.count == context.fileSize) { BaseMessage message = new BaseMessage(); message.setDataType(context.dataType); FileBean bean = new FileBean(); bean.setFileName(context.fileName); bean.setFileSize(context.fileSize); bean.setFileContent(context.byteFile); message.setData(bean); outPut.write(message); context.reset(); } return MessageDecoderResult.OK; } else if (sendType == BeanUtil.SEND_TYPE_TXT) { int count = context.count; String resultMsg = null; while (in.hasRemaining()) { byte b = in.get();// 整体读取 context.byteStr[count] = b; if (count == context.strLength - 1) { resultMsg = new String(context.byteStr, BeanUtil.charset); System.out.println(resultMsg); } count++; } context.count = count; System.out.println("count:" + count); System.out.println("context.strLength:" + context.strLength); session.setAttribute(CONTEXT, context); if (context.count == context.strLength) { BaseMessage message = new BaseMessage(); message.setDataType(context.dataType); message.setData(resultMsg); outPut.write(message); context.reset(); } return MessageDecoderResult.OK; } else { System.out .println("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"); return MessageDecoderResult.NOT_OK; } } /** * * */ public void finishDecode(IoSession session, ProtocolDecoderOutput outPut) throws Exception { // TODO Auto-generated method stub System.out.println("end:::::::::::::::::"); } }