EjobTaskDecoder.java

package com.hqyg.disjob.rpc.codec;

import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.LengthFieldBasedFrameDecoder;

import java.io.IOException;
import java.nio.ByteOrder;

import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.hqyg.disjob.common.util.LoggerUtil;
import com.hqyg.disjob.common.util.SerializeUtil;
import com.hqyg.disjob.rpc.utils.PhpTaskCmd;

 /**
  * 处理ejob的rpc解码类
  * @author Disjob
  *
  */
public class EjobTaskDecoder extends LengthFieldBasedFrameDecoder{
	public   PhpTaskCmd cmd;
	public EjobTaskDecoder(PhpTaskCmd cmd,int maxFrameLength, int lengthFieldOffset, int lengthFieldLength) {
		super(maxFrameLength, lengthFieldOffset, lengthFieldLength);
		this.cmd = cmd;
	}

	
	public EjobTaskDecoder(ByteOrder byteOrder, int maxFrameLength, int lengthFieldOffset, int lengthFieldLength,
			int lengthAdjustment, int initialBytesToStrip, boolean failFast) {
		super(byteOrder, maxFrameLength, lengthFieldOffset, lengthFieldLength, lengthAdjustment, initialBytesToStrip, failFast);
		
	}


	@Override
	protected Object decode(ChannelHandlerContext ctx, ByteBuf in) throws Exception {
		in = (ByteBuf)super.decode(ctx, in);
		if (in == null) {
	        return null;
	    }
		
		if (in.readableBytes() < 8) {
            LoggerUtil.warn("when server response, readableBytes less than header length!");
        }
		byte type = in.readByte();
		if(type == 0){
			return null;
		}
        int length = in.readInt();
        if (in.readableBytes() < length) {
            LoggerUtil.warn("when server response, the length of header is " + length + " but the readableBytes is less!");
        }
        ByteBuf buf = in.readBytes(length);
        byte[] req = new byte[buf.readableBytes()];
        buf.readBytes(req);
        EjobResponse response = bytes2Response(req);
        return response;
	        
	}


	private EjobResponse bytes2Response(byte[] req) throws JsonParseException,
			JsonMappingException, IOException {
		EjobResponse response = null;
        //kill task  后面第二版返回值要改成15
        switch (cmd.getType()) {
		case 15:
			response = (EjobKillTaskResponse) SerializeUtil.deserialize(req, EjobKillTaskResponse.class);
			break;
		case 14:
			response = (EjobRestartTaskResponse) SerializeUtil.deserialize(req, EjobRestartTaskResponse.class);
		default:
			break;
		}
		return response;
	}

	
}