go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-02-07 13:31:50 +02:00
Code Issues Releases Activity

clean transport (#920)

Browse Source
This commit is contained in:
Tony Chen 2021-05-17 20:18:53 +08:00 committed by GitHub
parent 8f8b861f7d
commit 0dbcaa2443
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
transport/grpc/server.go
View File

@ -15,7 +15,7 @@ import (
"github.com/go-kratos/kratos/v2/transport"
"google.golang.org/grpc"
"google.golang.org/grpc/health"
healthpb "google.golang.org/grpc/health/grpc_health_v1"
"google.golang.org/grpc/health/grpc_health_v1"
"google.golang.org/grpc/reflection"
)
@ -105,11 +105,9 @@ func NewServer(opts ...ServerOption) *Server {
}
srv.Server = grpc.NewServer(grpcOpts...)
srv.metaServer = metadata.NewServer(srv.Server)
// grpc health register
healthpb.RegisterHealthServer(srv.Server, srv.health)
// api metadata register
// internal register
grpc_health_v1.RegisterHealthServer(srv.Server, srv.health)
api.RegisterMetadataServer(srv.Server, srv.metaServer)
// reflection register
reflection.Register(srv.Server)
return srv
}

6
transport/http/client.go
View File

@ -74,7 +74,7 @@ func NewTransport(ctx context.Context, opts ...ClientOption) (http.RoundTripper,
ctx: ctx,
timeout: 500 * time.Millisecond,
transport: http.DefaultTransport,
errorDecoder: CheckResponse,
errorDecoder: checkResponse,
}
for _, o := range opts {
o(options)
@ -144,9 +144,9 @@ func Do(client *http.Client, req *http.Request, target interface{}) error {
return codec.Unmarshal(data, target)
}
// CheckResponse returns an error (of type *Error) if the response
// checkResponse returns an error (of type *Error) if the response
// status code is not 2xx.
func CheckResponse(ctx context.Context, res *http.Response) error {
func checkResponse(ctx context.Context, res *http.Response) error {
if res.StatusCode >= 200 && res.StatusCode <= 299 {
return nil
}