Apache Mina自定义编解码案例

Mina中已经自带的编解码类：

TextLineCodecFactory：基于文本的，根据回车换行来断点传输数据

ProtocolCodecFactory：自定义协议的编解码数据传输

ObjectSerializationCodecFactory：对象序列化传输

DemuxingProtocolCodecFactory：复用传输

自定义通信协议：

FlightSearch 1.0 \n
startcity:BJS \n
endcity:PEK \n
flightway:1 \n
date:2011-08-10 \n

Domain对象

package domain;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class Flight {
	public String startCity;
	public String endCity;
	public String flightway;
	public String date;
	public String fromDate;
	public String subclass1;
	public String flight1;
	
	/**
	 * 返回出发城市
	 * @return
	 */
	public String getStartCity() {
		return startCity;
	}
	public void setStartCity(String startCity) {
		this.startCity = startCity;
	}
	/**
	 * 返回到达城市
	 * @return
	 */
	public String getEndCity() {
		return endCity;
	}
	public void setEndCity(String endCity) {
		this.endCity = endCity;
	}
	/**
	 * 返回行程类型
	 * @return
	 */
	public String getFlightway() {
		return flightway;
	}
	public void setFlightway(String flightway) {
		this.flightway = flightway;
	}
	/**
	 * 返回出发日期
	 * @return
	 */
	public String getDate() {
		return date;
	}
	public void setDate(String date) {
		this.date = date;
	}
	@Override
	public String toString() {
		return "Flight [startCity=" + startCity + ", endCity=" + endCity + ", flightway=" + flightway + ", date="
				+ date + "]";
	}
	/**
	 * 返回往返日期
	 * @return
	 */
	public String getFromDate() {
		return fromDate;
	}
	public void setFromDate(String fromDate) {
		this.fromDate = fromDate;
	}
	public String getFlight1() {
		return flight1;
	}
	public void setFlight1(String flight1) {
		this.flight1 = flight1;
	}
	public String getSubclass1() {
		return subclass1;
	}
	public void setSubclass1(String subclass1) {
		this.subclass1 = subclass1;
	}
}

服务器端编码

package server;

import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightEncoder extends ProtocolEncoderAdapter {
	private final Charset charset = Charset.forName("UTF-8");
	/* 
	 * 服务器端编码无需处理，直接将接收到的数据向下传递
	 */
	@Override
	public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
		IoBuffer buf = IoBuffer.allocate(100).setAutoExpand(true);
		
		CharsetEncoder ce = charset.newEncoder();
		
		buf.putString((String)message, ce);
		
		buf.flip();
		
		out.write(buf);
	}

}

重点是服务器端解码

package server;

import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.CumulativeProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolDecoderAdapter;
import org.apache.mina.filter.codec.ProtocolDecoderOutput;

import domain.Flight;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightDecoder extends CumulativeProtocolDecoder {

	@Override
	protected boolean doDecode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception {
		IoBuffer buf = IoBuffer.allocate(100).setAutoExpand(true);
		CharsetDecoder cd = Charset.forName("UTF-8").newDecoder();
		
		int ColumnNumber = 0;
		String status="",startCity="",endCity="",flightway="",date="";
		
		int TextLineNumber = 1;
		
		Flight flight = new Flight();

		/**
		 * FlightSearch 1.0 \n
		 * startcity:BJS \n
		 * endcity:PEK \n
		 * flightway:1 \n
		 * date:2011-08-10 \n
		 */
		while(in.hasRemaining()){
			byte b = in.get();
			buf.put(b);
			if(b == 10 && TextLineNumber <= 5){
				ColumnNumber++;
				if(TextLineNumber == 1){
					buf.flip();
					status = buf.getString(ColumnNumber, cd);
				}
				
				if(TextLineNumber == 2){
					buf.flip();
					startCity = buf.getString(ColumnNumber, cd).split(":")[1];
					startCity = startCity.substring(0, startCity.length()-1);
					flight.setStartCity(startCity);
				}
				
				if(TextLineNumber == 3){
					buf.flip();
					endCity = buf.getString(ColumnNumber, cd).split(":")[1];
					endCity = endCity.substring(0, endCity.length()-1);
					flight.setEndCity(endCity);
				}
				
				if(TextLineNumber == 4){
					buf.flip();
					flightway = buf.getString(ColumnNumber, cd).split(":")[1];
					flightway = flightway.substring(0, flightway.length()-1);
					flight.setFlightway(flightway);
				}
				
				if(TextLineNumber == 5){
					buf.flip();
					date = buf.getString(ColumnNumber, cd).split(":")[1];
					date = date.substring(0, date.length()-1);
					flight.setDate(date);
					break;
				}
				
				ColumnNumber = 0;
				buf.clear();
				TextLineNumber++;
			}else{
				ColumnNumber++;
			}
		}
		out.write(flight);
		return false;
	}

}

服务器端编解码工厂

package server;

import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFactory;
import org.apache.mina.filter.codec.ProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolEncoder;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightCodecFactory implements ProtocolCodecFactory {
	private final ProtocolEncoder encoder = new FlightEncoder();
	private final ProtocolDecoder decoder = new FlightDecoder();

	@Override
	public ProtocolDecoder getDecoder(IoSession session) throws Exception {
		return decoder;
	}

	@Override
	public ProtocolEncoder getEncoder(IoSession session) throws Exception {
		return encoder;
	}

}

下面是客户端的编解码

重点是编码，需要将数据组装成协议格式，发送给服务器

package client;

import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolEncoderAdapter;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;

import domain.Flight;


/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightClientEncoder extends ProtocolEncoderAdapter {
	private final Charset charset;
	
	public FlightClientEncoder(){
		this.charset = Charset.forName("UTF-8");
	}

	@Override
	public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
		Flight flight = (Flight)message;
		
		IoBuffer buf = IoBuffer.allocate(100).setAutoExpand(true);
		
		CharsetEncoder ce = charset.newEncoder();
		
		buf.putString("Flight Search 1.0" + '\n', ce);
		
		buf.putString("startcty:" + flight.getStartCity() + '\n', ce);
		
		buf.putString("endcity:" + flight.getEndCity() + '\n', ce);
		
		buf.putString("flightway:" + flight.getFlightway() + '\n', ce);
		
		buf.putString("date:" + flight.getDate() + '\n', ce);
		
		buf.flip();
		
		out.write(buf);
	}

}

解码无需特殊处理，接收完数据直接向下传递

package client;

import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;

import org.apache.mina.core.buffer.IoBuffer;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.CumulativeProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolDecoderOutput;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightClientDecoder extends CumulativeProtocolDecoder {

	/* (non-Javadoc)
	 * @see org.apache.mina.filter.codec.ProtocolDecoder#decode(org.apache.mina.core.session.IoSession, org.apache.mina.core.buffer.IoBuffer, org.apache.mina.filter.codec.ProtocolDecoderOutput)
	 */
	@Override
	protected boolean doDecode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception {
		CharsetDecoder cd = Charset.forName("UTF-8").newDecoder();
		IoBuffer buf = IoBuffer.allocate(100).setAutoExpand(true);
		
		while(in.hasRemaining()){
			buf.put(in.get());
		}
		buf.flip();
		out.write(buf.getString(cd));
		return false;
	}

}

客户端编解码工厂

package client;

import org.apache.mina.core.session.IoSession;
import org.apache.mina.filter.codec.ProtocolCodecFactory;
import org.apache.mina.filter.codec.ProtocolDecoder;
import org.apache.mina.filter.codec.ProtocolEncoder;

/**
 * @function : 
 * @author   :jy
 * @company  :万里网
 * @date     :2011-8-7
 */
public class FlightClientCodecFactory implements ProtocolCodecFactory {
	private final ProtocolEncoder encoder = new FlightClientEncoder();
	private final ProtocolDecoder decoder = new FlightClientDecoder();
	
	/* (non-Javadoc)
	 * @see org.apache.mina.filter.codec.ProtocolCodecFactory#getDecoder(org.apache.mina.core.session.IoSession)
	 */
	@Override
	public ProtocolDecoder getDecoder(IoSession arg0) throws Exception {
		return decoder;
	}

	/* (non-Javadoc)
	 * @see org.apache.mina.filter.codec.ProtocolCodecFactory#getEncoder(org.apache.mina.core.session.IoSession)
	 */
	@Override
	public ProtocolEncoder getEncoder(IoSession arg0) throws Exception {
		return encoder;
	}

}