Generate endpoints + transports
This commit is contained in:
@@ -1 +1,49 @@
|
||||
package {{.File.Package}}_endpoints
|
||||
package {{.File.Package}}_endpoints
|
||||
|
||||
{{$file := .File}}
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
context "golang.org/x/net/context"
|
||||
pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/{{.File.Package}}"
|
||||
"github.com/go-kit/kit/endpoint"
|
||||
)
|
||||
|
||||
var _ = fmt.Errorf
|
||||
|
||||
type Endpoints struct {
|
||||
{{range .Service.Method}}
|
||||
{{.Name}}Endpoint endpoint.Endpoint
|
||||
{{end}}
|
||||
}
|
||||
|
||||
{{range .Service.Method}}
|
||||
func (e *Endpoints){{.Name}}(ctx context.Context, in *pb.{{.Name}}Request) (*pb.{{.Name}}Response, error) {
|
||||
out, err := e.{{.Name}}Endpoint(ctx, in)
|
||||
if err != nil {
|
||||
return &pb.{{.Name}}Response{ErrMsg: err.Error()}, err
|
||||
}
|
||||
return out.(*pb.{{.Name}}Reply), err
|
||||
}
|
||||
{{end}}
|
||||
|
||||
{{range .Service.Method}}
|
||||
func Make{{.Name}}Endpoint(svc pb.{{$file.Package | title}}ServiceServer) endpoint.Endpoint {
|
||||
return func(ctx context.Context, request interface{}) (interface{}, error) {
|
||||
req := request.(*pb.{{.Name}}Request)
|
||||
rep, err := svc.{{.Name}}(ctx, req)
|
||||
if err != nil {
|
||||
return &pb.{{.Name}}Reply{ErrMsg: err.Error()}, err
|
||||
}
|
||||
return rep, nil
|
||||
}
|
||||
}
|
||||
{{end}}
|
||||
|
||||
func MakeEndpoints(svc pb.{{.File.Package | title}}ServiceServer) Endpoints {
|
||||
return Endpoints{
|
||||
{{range .Service.Method}}
|
||||
{{.Name}}Endpoint: Make{{.Name}}Endpoint(svc),
|
||||
{{end}}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user