func (p *RpcServiceBaseClient) recvPing() (err error) {
iprot := p.InputProtocol
if iprot == nil {
iprot = p.ProtocolFactory.GetProtocol(p.Transport)
p.InputProtocol = iprot
}
method, mTypeId, seqId, err := iprot.ReadMessageBegin()
if err != nil {
return
}
if method != "ping" {
err = thrift.NewTApplicationException(thrift.WRONG_METHOD_NAME, "ping failed: wrong method name")
return
}
if p.SeqId != seqId {
err = thrift.NewTApplicationException(thrift.BAD_SEQUENCE_ID, "ping failed: out of sequence response")
return
}
if mTypeId == thrift.EXCEPTION {
error0 := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, "Unknown Exception")
var error1 error
error1, err = error0.Read(iprot)
if err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
err = error1
return
}
if mTypeId != thrift.REPLY {
err = thrift.NewTApplicationException(thrift.INVALID_MESSAGE_TYPE_EXCEPTION, "ping failed: invalid message type")
return
}
result := RpcServiceBasePingResult{}
if err = result.Read(iprot); err != nil {
return
}
if err = iprot.ReadMessageEnd(); err != nil {
return
}
return
}
func (p *rpcServiceBaseProcessorPing) Process(seqId int32, iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) {
args := RpcServiceBasePingArgs{}
if err = args.Read(iprot); err != nil {
iprot.ReadMessageEnd()
x := thrift.NewTApplicationException(thrift.PROTOCOL_ERROR, err.Error())
oprot.WriteMessageBegin("ping", thrift.EXCEPTION, seqId)
x.Write(oprot)
oprot.WriteMessageEnd()
oprot.Flush()
return false, err
}
iprot.ReadMessageEnd()
result := RpcServiceBasePingResult{}
var err2 error
if err2 = p.handler.Ping(); err2 != nil {
x := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, "Internal error processing ping: "+err2.Error())
oprot.WriteMessageBegin("ping", thrift.EXCEPTION, seqId)
x.Write(oprot)
oprot.WriteMessageEnd()
oprot.Flush()
return true, err2
}
if err2 = oprot.WriteMessageBegin("ping", thrift.REPLY, seqId); err2 != nil {
err = err2
}
if err2 = result.Write(oprot); err == nil && err2 != nil {
err = err2
}
if err2 = oprot.WriteMessageEnd(); err == nil && err2 != nil {
err = err2
}
if err2 = oprot.Flush(); err == nil && err2 != nil {
err = err2
}
if err != nil {
return
}
return true, err
}
func GetWorkerNotFoundData(req *Request, module string) []byte {
req.Response.TypeId = thrift.EXCEPTION
// 构建thrift的Transport
transport := thrift.NewTMemoryBufferLen(100)
protocol := thrift.NewTBinaryProtocolTransport(transport)
// 构建一个Message, 写入Exception
msg := fmt.Sprintf("Worker FOR %s#%s.%s Not Found", module, req.Service, req.Request.Name)
exc := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, msg)
protocol.WriteMessageBegin(req.Request.Name, thrift.EXCEPTION, req.Request.SeqId)
exc.Write(protocol)
protocol.WriteMessageEnd()
bytes := transport.Bytes()
return bytes
}
func (p *RpcServiceBaseProcessor) Process(iprot, oprot thrift.TProtocol) (success bool, err thrift.TException) {
name, _, seqId, err := iprot.ReadMessageBegin()
if err != nil {
return false, err
}
if processor, ok := p.GetProcessorFunction(name); ok {
return processor.Process(seqId, iprot, oprot)
}
iprot.Skip(thrift.STRUCT)
iprot.ReadMessageEnd()
x3 := thrift.NewTApplicationException(thrift.UNKNOWN_METHOD, "Unknown function "+name)
oprot.WriteMessageBegin(name, thrift.EXCEPTION, seqId)
x3.Write(oprot)
oprot.WriteMessageEnd()
oprot.Flush()
return false, x3
}
//
// 生成Thrift格式的Exception Message
//
func GetServiceNotFoundData(req *Request) []byte {
req.Response.TypeId = thrift.EXCEPTION
// 构建thrift的Transport
transport := thrift.NewTMemoryBufferLen(100)
protocol := thrift.NewTBinaryProtocolTransport(transport)
// 构建一个Message, 写入Exception
msg := fmt.Sprintf("Service: %s Not Found", req.Service)
exc := thrift.NewTApplicationException(thrift.UNKNOWN_APPLICATION_EXCEPTION, msg)
protocol.WriteMessageBegin(req.Request.Name, thrift.EXCEPTION, req.Request.SeqId)
exc.Write(protocol)
protocol.WriteMessageEnd()
protocol.Flush()
bytes := transport.Bytes()
return bytes
}
func GetThriftException(req *Request, module string) []byte {
req.Response.TypeId = thrift.EXCEPTION
// 构建thrift的Transport
transport := thrift.NewTMemoryBufferLen(256)
protocol := thrift.NewTBinaryProtocolTransport(transport)
msg := fmt.Sprintf("Module: %s, Service: %s, Method: %s, Error: %v", module, req.Service, req.Request.Name, req.Response.Err)
// 构建一个Message, 写入Exception
exc := thrift.NewTApplicationException(thrift.INTERNAL_ERROR, msg)
// 注意消息的格式
protocol.WriteMessageBegin(req.Request.Name, thrift.EXCEPTION, req.Request.SeqId)
exc.Write(protocol)
protocol.WriteMessageEnd()
bytes := transport.Bytes()
return bytes
}