com.github.javaclub.mq.common.netty.protocol.PubMsgProtocol Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mq-common Show documentation

Javaclub MQ

There is a newer version: 2.1.10

package com.github.javaclub.mq.common.netty.protocol;

import io.netty.buffer.ByteBuf;

/**
 * 	producer向broker发送消息
 */
public class PubMsgProtocol extends AbstractProtocol {
	private static byte code = CodeProtocol.pub_msg;
	private int length; // header + body
	private int headerLength;
	private byte[] header;
	private byte[] body;

	private String msgid;	// 用于返回发送者的，不参与协议
	
	
	public void encode(ByteBuf out) {
		super.setCode(code);
		super.encode(out);
		
		out.writeInt(length);
		out.writeInt(headerLength);
		out.writeBytes(header);
		out.writeBytes(body);
	}

	public int decode(ByteBuf in){
		int readindex = super.decode(in);
		if(readindex != -1){  
			if(in.readableBytes() >= 8){
				readindex = in.readerIndex();
				length = in.readInt();
				headerLength = in.readInt();
				if(in.readableBytes() >= length){
					byte[] header = new byte[headerLength];
					in.readBytes(header);
					this.header = header;
					byte[] body = new byte[length - headerLength];
					in.readBytes(body);
					this.body = body;
					return 0;
				}else{
					in.readerIndex(readindex); // 恢复readindex
					return -1;
				}
			}else{
				in.readerIndex(readindex); // 恢复readindex
				return -1;
			}
		}
		return -1;
	}
	
	public byte getCode() {
		return code;
	}

	public void setCode(byte code) {
		PubMsgProtocol.code = code;
	}

	public int getLength() {
		return length;
	}

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

	public int getHeaderLength() {
		return headerLength;
	}

	public void setHeaderLength(int headerLength) {
		this.headerLength = headerLength;
	}

	public byte[] getHeader() {
		return header;
	}

	public void setHeader(byte[] header) {
		this.header = header;
	}

	public byte[] getBody() {
		return body;
	}

	public void setBody(byte[] body) {
		this.body = body;
	}

	public String getMsgid() {
		return msgid;
	}

	public void setMsgid(String msgid) {
		this.msgid = msgid;
	}

	
}