unistack-org/protoc-gen-gotemplate
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Better reliability on generated message names

Browse Source
This commit is contained in:
Manfred Touron 2016-12-21 10:32:27 +01:00
parent d0ac6afb18
commit 88a2f47522
No known key found for this signature in database
GPG Key ID: 9CCF47DF1FD978A1
3 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/go-kit/templates/{{.File.Package}}/gen/endpoints/endpoints.go.tmpl
View File

@ -19,22 +19,24 @@ type Endpoints struct {
} }
{{range .Service.Method}} {{range .Service.Method}}
func (e *Endpoints){{.Name}}(ctx context.Context, in *pb.{{.Name}}Request) (*pb.{{.Name}}Response, error) { /*{{. | prettyjson}}*/
func (e *Endpoints){{.Name}}(ctx context.Context, in *pb.{{.InputType | splitArray "." | last}}) (*pb.{{.OutputType | splitArray "." | last}}, error) {
out, err := e.{{.Name}}Endpoint(ctx, in) out, err := e.{{.Name}}Endpoint(ctx, in)
if err != nil { if err != nil {
return &pb.{{.Name}}Response{ErrMsg: err.Error()}, err return &pb.{{.OutputType | splitArray "." | last}}{ErrMsg: err.Error()}, err
} }
return out.(*pb.{{.Name}}Response), err return out.(*pb.{{.OutputType | splitArray "." | last}}), err
} }
{{end}} {{end}}
{{range .Service.Method}} {{range .Service.Method}}
func Make{{.Name}}Endpoint(svc pb.{{$file.Package | title}}ServiceServer) endpoint.Endpoint { func Make{{.Name}}Endpoint(svc pb.{{$file.Package | title}}ServiceServer) endpoint.Endpoint {
return func(ctx context.Context, request interface{}) (interface{}, error) { return func(ctx context.Context, request interface{}) (interface{}, error) {
req := request.(*pb.{{.Name}}Request) req := request.(*pb.{{.InputType | splitArray "." | last}})
rep, err := svc.{{.Name}}(ctx, req) rep, err := svc.{{.Name}}(ctx, req)
if err != nil { if err != nil {
return &pb.{{.Name}}Response{ErrMsg: err.Error()}, err return &pb.{{.OutputType | splitArray "." | last}}{ErrMsg: err.Error()}, err
} }
return rep, nil return rep, nil
} }

6
examples/go-kit/templates/{{.File.Package}}/gen/transports/grpc/grpc.go.tmpl
View File

@ -40,12 +40,12 @@ type grpcServer struct {
} }
{{range .Service.Method}} {{range .Service.Method}}
func (s *grpcServer) {{.Name}}(ctx context.Context, req *pb.{{.Name}}Request) (*pb.{{.Name}}Response, error) { func (s *grpcServer) {{.Name}}(ctx context.Context, req *pb.{{.InputType | splitArray "." | last}}) (*pb.{{.OutputType | splitArray "." | last}}, error) {
_, rep, err := s.{{.Name | lower}}.ServeGRPC(ctx, req) _, rep, err := s.{{.Name | lower}}.ServeGRPC(ctx, req)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return rep.(*pb.{{.Name}}Response), nil return rep.(*pb.{{.OutputType | splitArray "." | last}}), nil
} }
func decode{{.Name}}Request(ctx context.Context, grpcReq interface{}) (interface{}, error) { func decode{{.Name}}Request(ctx context.Context, grpcReq interface{}) (interface{}, error) {
@ -53,7 +53,7 @@ func decode{{.Name}}Request(ctx context.Context, grpcReq interface{}) (interface
} }
func encode{{.Name}}Response(ctx context.Context, response interface{}) (interface{}, error) { func encode{{.Name}}Response(ctx context.Context, response interface{}) (interface{}, error) {
resp := response.(*pb.{{.Name}}Response) resp := response.(*pb.{{.OutputType | splitArray "." | last}})
return resp, nil return resp, nil
} }
{{end}} {{end}}

2
examples/go-kit/templates/{{.File.Package}}/gen/transports/http/http.go.tmpl
View File

@ -26,7 +26,7 @@ func Make{{.Name}}Handler(ctx context.Context, svc pb.{{$file.Package | title}}S
} }
func decode{{.Name}}Request(ctx context.Context, r *http.Request) (interface{}, error) { func decode{{.Name}}Request(ctx context.Context, r *http.Request) (interface{}, error) {
var req pb.{{.Name}}Request var req pb.{{.InputType | splitArray "." | last}}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil { if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
return nil, err return nil, err
} }