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 }