瀏覽代碼

fix: handling rpc error on gateway (#2212)

Kevin Wan 2 年之前
父節點
當前提交
49367f1713
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      gateway/server.go

+ 6 - 0
gateway/server.go

@@ -16,6 +16,7 @@ import (
 	"github.com/zeromicro/go-zero/rest"
 	"github.com/zeromicro/go-zero/rest/httpx"
 	"github.com/zeromicro/go-zero/zrpc"
+	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/reflection/grpc_reflection_v1alpha"
 )
 
@@ -135,6 +136,11 @@ func (s *Server) buildHandler(source grpcurl.DescriptorSource, resolver jsonpb.A
 			handler, parser.Next); err != nil {
 			httpx.Error(w, err)
 		}
+
+		st := handler.Status
+		if st.Code() != codes.OK {
+			httpx.Error(w, st.Err())
+		}
 	}
 }