Mina 编写自己的编码解码FilterChain(codec)

mina 自己带的FilterChain codec是非常好用，但是在实际网络应用中还是有他的局限性，如编写基于CMPP、SGIP的短信系统。

下面我编写的一个自己FilterChain例子，方便以后查看和其他的人查阅（哎，网上的mina资料少的可怜）

1.mina服务器类

package Minatest;

import java.net.InetSocketAddress;

import org.apache.mina.common.DefaultIoFilterChainBuilder;
import org.apache.mina.common.IdleStatus;
import org.apache.mina.common.IoAcceptor;
import org.apache.mina.common.IoAcceptorConfig;
import org.apache.mina.common.IoHandler;
import org.apache.mina.common.IoSession;
import org.apache.mina.filter.LoggingFilter;
import org.apache.mina.filter.codec.ProtocolCodecFilter;
import org.apache.mina.transport.socket.nio.SocketAcceptor;
import org.apache.mina.transport.socket.nio.SocketAcceptorConfig;

public class MinaServer {

/**
* @param args
* @throws Exception
* zxp
*
*/
public static void main(String[] args) throws Exception {
IoAcceptor acceptor=new SocketAcceptor();
IoAcceptorConfig config=new SocketAcceptorConfig();
//config.setDisconnectOnUnbind(true);
DefaultIoFilterChainBuilder d= config.getFilterChain();
//d.addLast("codec", new ProtocolCodecFilter(new TextLineCodecFactory( Charset.forName("UTF-8"))));
//d.addFirst("a", );
InetSocketAddress address=new InetSocketAddress(9000);
acceptor.bind(address, new MinaServer().new server(), config);

}
public class server implements IoHandler {

public void exceptionCaught(IoSession arg0, Throwable arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("异常:"+arg1);
arg1.fillInStackTrace();
}

public void messageReceived(IoSession arg0, Object arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("收到信息");
SendMessages send=(SendMessages)arg1;
System.out.println(send.msg);
//收到客户端的消息 为0表示刚刚发送的测试消息发送成功 （这个流程是针对CMPP协议，对于普通的来说没意义）
if(send.msg.equals("0")){
this.sessionClosed(arg0);
}


}
//信息发送成功后触发的事件
public void messageSent(IoSession arg0, Object arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("发送信息");
}

public void sessionClosed(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("session 关闭");
arg0.close();
}

public void sessionCreated(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("session 建立");
}

public void sessionIdle(IoSession arg0, IdleStatus arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("session 空闲");
if(arg1==IdleStatus.BOTH_IDLE){
// //自定义发送类
// SendMessages send= new SendMessages();
// send.setMsg("Hi!客户端你空闲咯！");
// this.sendMessage(arg0, send);
}
}

public void sessionOpened(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("session 打开");

//arg0.setIdleTime(IdleStatus.BOTH_IDLE, 10);
arg0.getFilterChain().addFirst("a", new ProtocolCodecFilter(new TestProtocolCodecFactory()));
arg0.getFilterChain().addLast("logger", new LoggingFilter());

//发送条测试消息
SendMessages send= new SendMessages();
send.setMsg("Hi!客户端你好！");
this.sendMessage(arg0, send);

}

private void sendMessage(IoSession arg0, Object arg1){
arg0.write(arg1);
}
}

}

2.mina客户端类

package Minatest;

import java.net.InetSocketAddress;
import java.nio.charset.Charset;

import org.apache.mina.common.DefaultIoFilterChainBuilder;
import org.apache.mina.common.IdleStatus;
import org.apache.mina.common.IoConnector;
import org.apache.mina.common.IoConnectorConfig;
import org.apache.mina.common.IoHandler;
import org.apache.mina.common.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFactory;
import org.apache.mina.filter.codec.ProtocolCodecFilter;
import org.apache.mina.filter.codec.textline.TextLineCodecFactory;
import org.apache.mina.transport.socket.nio.SocketConnector;
import org.apache.mina.transport.socket.nio.SocketConnectorConfig;


public class MinaClient {

/**
* @param args
* zxp
*
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
IoConnector connector=new SocketConnector();
IoConnectorConfig config=new SocketConnectorConfig();
//config.setConnectTimeout(1);
//DefaultIoFilterChainBuilder d= config.getFilterChain();
//d.addLast("codec", new ProtocolCodecFilter(new TextLineCodecFactory(Charset.forName("UTF-8"))));
InetSocketAddress address=new InetSocketAddress("127.0.0.1",9000);
connector.connect(address, new MinaClient().new client(), config);

}
public class client implements IoHandler{

public void exceptionCaught(IoSession arg0, Throwable arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端异常:"+arg1);
arg0.close();
}

public void messageReceived(IoSession arg0, Object arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端收到消息");
SendMessages send=(SendMessages) arg1;
System.out.println(send.getMsg());
if(send.msg!=null){
send.setMsg("0");
this.sendMessages(arg0,send);
}
}

public void messageSent(IoSession arg0, Object arg1) throws Exception {
// TODO Auto-generated method stub
//不能用这个方法发送信息，要不死循环
//arg0.getw
}

public void sessionClosed(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端关闭");
arg0.close();
}

public void sessionCreated(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端建立");
//arg0.getFilterChain()
}

public void sessionIdle(IoSession arg0, IdleStatus arg1) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端空闲");
if(arg1==IdleStatus.BOTH_IDLE){
this.messageSent(arg0, "空闲测试");
SendMessages send=new SendMessages();
send.setMsg("aaaaa");
this.sendMessages(arg0, send);
}
}

public void sessionOpened(IoSession arg0) throws Exception {
// TODO Auto-generated method stub
System.out.println("客户端打开");
//10秒内没有读写就设置为空闲通道
arg0.setIdleTime(IdleStatus.BOTH_IDLE, 10);

//自定义包解析
//ProtocolCodecFactory codec=new TestProtocolCodecFactory();
//arg0.getFilterChain().addFirst("test", new ProtocolCodecFilter(codec));
arg0.getFilterChain().addLast("a", new ProtocolCodecFilter(new TestProtocolCodecFactory()));
}

private void sendMessages(IoSession arg0, Object arg1){
System.out.println("sendMessages 发送");
arg0.write(arg1);
}

}
}

3.工厂类

package Minatest;

import org.apache.mina.filter.codec.demux.DemuxingProtocolCodecFactory;

public class TestProtocolCodecFactory extends DemuxingProtocolCodecFactory {
public TestProtocolCodecFactory(){
super.register(TestDecoder.class);
super.register(TestEncoder.class);
}
}

4.编码类

package Minatest;

import java.util.Collections;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Set;

import org.apache.mina.common.ByteBuffer;
import org.apache.mina.common.IoSession;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;
import org.apache.mina.filter.codec.demux.MessageEncoder;


public class TestEncoder implements MessageEncoder {
/**
* 编码消息
* zxp
*
*/
public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
SendMessages msg=(SendMessages)message;
ByteBuffer b=ByteBuffer.allocate(msg.getmsgleng());
b.setAutoExpand(true);
//编码消息
msg.encoder(b);
//b.putInt(97);
//b.put("a".getBytes());
b.flip();
out.write(b);
}
/**
* 获取编码自定义消息类型(多个)
*
*/
public Set<Class<?>> getMessageTypes() {
Set<Class<?>> set=new HashSet<Class<?>>();
set.add(SendMessages.class);
//返回指定映射的不可修改视图 即只读
return Collections.unmodifiableSet(set);
}

}
5.解码类

package Minatest;

import org.apache.mina.common.ByteBuffer;
import org.apache.mina.common.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 TestDecoder implements MessageDecoder {
/**
* 检查指定的缓冲区decodable解码器
* zxp
*/
public MessageDecoderResult decodable(IoSession arg0, ByteBuffer arg1) {
//包的长度是否大于0
if(arg1.remaining()<=0){
return MessageDecoderResult.NOT_OK;
}
return MessageDecoderResult.OK;
}
/**
* 解码二进制或协议的具体内容到更高级别的消息对象
* (non-Javadoc)
* @see org.apache.mina.filter.codec.demux.MessageDecoder#decode(org.apache.mina.common.IoSession, org.apache.mina.common.ByteBuffer, org.apache.mina.filter.codec.ProtocolDecoderOutput)
*/
public MessageDecoderResult decode(IoSession arg0, ByteBuffer arg1,
ProtocolDecoderOutput arg2) throws Exception {
//解码
byte[] b=new byte[arg1.limit()];
arg1.get(b);
System.out.println("decode:"+arg1.position());
SendMessages send=new SendMessages();
String s=new String(b);
send.setMsg(s);
arg2.write(send);
return MessageDecoderResult.OK;
}

/**
* 解码完成后调用的方法
*
*/
public void finishDecode(IoSession arg0, ProtocolDecoderOutput arg1)
throws Exception {

}

}

6.自定义消息类

package Minatest;

import org.apache.mina.common.ByteBuffer;

public class SendMessages implements java.io.Serializable {


public String msg="0";

public String getMsg() {
return msg;
}

public void setMsg(String msg) {
this.msg = msg;
}

public int getmsgleng(){
return msg.getBytes().length;
}

public void encoder(ByteBuffer b){
b.put(msg.getBytes());
//b.putInt(1);
}

}