Refactor the go-kit templates to make imports generic

This commit is contained in:
Manfred Touron
2016-12-15 16:28:04 +01:00
parent f9e191c6a4
commit df709b52ac
3 changed files with 9 additions and 8 deletions

View File

@@ -4,8 +4,9 @@ package {{.File.Package}}_endpoints
import (
"fmt"
context "golang.org/x/net/context"
pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/{{.File.Package}}/gen/pb"
pb "{{cat .GoPWD "/" .DestinationDir | nospace | clean}}/pb"
"github.com/go-kit/kit/endpoint"
)
@@ -29,7 +30,7 @@ func (e *Endpoints){{.Name}}(ctx context.Context, in *pb.{{.Name}}Request) (*pb.
{{range .Service.Method}}
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)
rep, err := svc.{{.Name}}(ctx, req)
if err != nil {