57 lines
1.7 KiB
Cheetah
57 lines
1.7 KiB
Cheetah
|
package {{.File.Package}}_clientgrpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/go-kit/kit/log"
|
||
|
|
context "golang.org/x/net/context"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
grpctransport "github.com/go-kit/kit/transport/grpc"
|
||
|
|
"github.com/go-kit/kit/endpoint"
|
||
|
|
jwt "github.com/go-kit/kit/auth/jwt"
|
||
|
|
|
||
|
|
pb "{{cat .GoPWD "/" .DestinationDir | nospace | clean}}/pb"
|
||
|
|
endpoints "{{cat .GoPWD "/" .DestinationDir | nospace | clean}}/endpoints"
|
||
|
|
)
|
||
|
|
|
||
|
|
{{$file:=.File}}
|
||
|
|
|
||
|
|
func New(conn *grpc.ClientConn, logger log.Logger) pb.{{.File.Package | title}}ServiceServer {
|
||
|
|
{{range .Service.Method}}
|
||
|
|
{{if and (not .ServerStreaming) (not .ClientStreaming)}}
|
||
|
|
var {{.Name | lower}}Endpoint endpoint.Endpoint
|
||
|
|
{
|
||
|
|
{{.Name | lower}}Endpoint = grpctransport.NewClient(
|
||
|
|
conn,
|
||
|
|
"{{$file.Package}}.{{$file.Package | title}}Service",
|
||
|
|
"{{.Name}}",
|
||
|
|
Encode{{.Name}}Request,
|
||
|
|
Decode{{.Name}}Response,
|
||
|
|
pb.{{.Name}}Response{},
|
||
|
|
append([]grpctransport.ClientOption{}, grpctransport.ClientBefore(jwt.FromGRPCContext()))...,
|
||
|
|
).Endpoint()
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|
||
|
|
|
||
|
|
return &endpoints.Endpoints {
|
||
|
|
{{range .Service.Method}}
|
||
|
|
{{if and (not .ServerStreaming) (not .ClientStreaming)}}
|
||
|
|
{{.Name | title}}Endpoint: {{.Name | lower}}Endpoint,
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
{{range .Service.Method}}
|
||
|
|
{{if and (not .ServerStreaming) (not .ClientStreaming)}}
|
||
|
|
func Encode{{.Name}}Request(_ context.Context, request interface{}) (interface{}, error) {
|
||
|
|
req := request.(*pb.{{.Name}}Request)
|
||
|
|
return req, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func Decode{{.Name}}Response(_ context.Context, grpcResponse interface{}) (interface{}, error) {
|
||
|
|
response := grpcResponse.(*pb.{{.Name}}Response)
|
||
|
|
return response, nil
|
||
|
|
}
|
||
|
|
{{end}}
|
||
|
|
{{end}}
|