origin/rpc/msgpprocessor.go

package rpc

type IMsgp interface {
	UnmarshalMsg(bts []byte) (o []byte, err error)
	MarshalMsg(b []byte) (o []byte, err error)
}

type MsgpProcessor struct {
}

//go:generate msgp
type MsgpRpcRequestData struct {
	//packhead
	Seq uint64             // sequence number chosen by client
	ServiceMethod string   // format: "Service.Method"
	NoReply bool           //是否需要返回
	//packbody
	InParam []byte
}

//go:generate msgp
type MsgpRpcResponseData struct {
	//head
	Seq           uint64   // sequence number chosen by client
	Err string

	//returns
	Reply []byte
}


func (slf *MsgpProcessor) Marshal(v interface{}) ([]byte, error){
	msgp := v.(IMsgp)
	return msgp.MarshalMsg(nil)
}

func (slf *MsgpProcessor) Unmarshal(data []byte, v interface{}) error{
	msgp := v.(IMsgp)
	_,err := msgp.UnmarshalMsg(data)
	return err
}

func (slf *MsgpProcessor) MakeRpcRequest(seq uint64,serviceMethod string,noReply bool,inParam []byte) IRpcRequestData{
	return &MsgpRpcRequestData{Seq:seq,ServiceMethod:serviceMethod,NoReply:noReply,InParam:inParam}
}

func (slf *MsgpProcessor) MakeRpcResponse(seq uint64,err *RpcError,reply []byte) IRpcResponseData {
	return &MsgpRpcResponseData{
		Seq:   seq,
		Err:   err.Error(),
		Reply: reply,
	}
}

func (slf *MsgpRpcRequestData) IsReply() bool{
	return slf.NoReply
}

func (slf *MsgpRpcRequestData) GetSeq() uint64{
	return slf.Seq
}

func (slf *MsgpRpcRequestData) GetServiceMethod() string{
	return slf.ServiceMethod
}

func (slf *MsgpRpcRequestData) GetInParam() []byte{
	return slf.InParam
}

func (slf *MsgpRpcResponseData)	GetSeq() uint64 {
	return slf.Seq
}

func (slf *MsgpRpcResponseData)		GetErr() *RpcError {
	if slf.Err == ""{
		return nil
	}

	return Errorf(slf.Err)
}


func (slf *MsgpRpcResponseData)		GetReply() []byte{
	return slf.Reply
}