func TestUnknownError(t *testing.T) { withSetup(t, func(ctx Context, args testArgs) { args.s1.On("Simple", ctxArg()).Return(errors.New("unexpected err")) got := args.c1.Simple(ctx) require.Error(t, got) require.Equal(t, tchannel.NewSystemError(tchannel.ErrCodeUnexpected, "unexpected err"), got) }) }
func (s *Server) handle(origCtx context.Context, handler handler, method string, call *tchannel.InboundCall) error { reader, err := call.Arg2Reader() if err != nil { return err } headers, err := ReadHeaders(reader) if err != nil { return err } if err := reader.Close(); err != nil { return err } reader, err = call.Arg3Reader() if err != nil { return err } tracer := tchannel.TracerFromRegistrar(s.ch) origCtx = tchannel.ExtractInboundSpan(origCtx, call, headers, tracer) ctx := s.ctxFn(origCtx, method, headers) wp := getProtocolReader(reader) success, resp, err := handler.server.Handle(ctx, method, wp.protocol) thriftProtocolPool.Put(wp) if err != nil { if _, ok := err.(thrift.TProtocolException); ok { // We failed to parse the Thrift generated code, so convert the error to bad request. err = tchannel.NewSystemError(tchannel.ErrCodeBadRequest, err.Error()) } reader.Close() call.Response().SendSystemError(err) return nil } if err := reader.Close(); err != nil { return err } if !success { call.Response().SetApplicationError() } writer, err := call.Response().Arg2Writer() if err != nil { return err } if err := WriteHeaders(writer, ctx.ResponseHeaders()); err != nil { return err } if err := writer.Close(); err != nil { return err } writer, err = call.Response().Arg3Writer() wp = getProtocolWriter(writer) resp.Write(wp.protocol) thriftProtocolPool.Put(wp) err = writer.Close() if handler.postResponseCB != nil { handler.postResponseCB(ctx, method, resp) } return err }