Generate endpoints + transports

examples/go-kit/services/sprint/gen/client/client.go

@@ -1 +0,0 @@
-package sprint_client

examples/go-kit/services/sprint/gen/cmd/sprint/main.go

@@ -1 +0,0 @@
-package main

examples/go-kit/services/sprint/gen/endpoints/endpoints.go

@@ -1 +1,86 @@
 package sprint_endpoints
+
+import (
+	"fmt"
+	"github.com/go-kit/kit/endpoint"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/sprint"
+	context "golang.org/x/net/context"
+)
+
+var _ = fmt.Errorf
+
+type Endpoints struct {
+	AddSprintEndpoint endpoint.Endpoint
+
+	CloseSprintEndpoint endpoint.Endpoint
+
+	GetSprintEndpoint endpoint.Endpoint
+}
+
+func (e *Endpoints) AddSprint(ctx context.Context, in *pb.AddSprintRequest) (*pb.AddSprintResponse, error) {
+	out, err := e.AddSprintEndpoint(ctx, in)
+	if err != nil {
+		return &pb.AddSprintResponse{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.AddSprintReply), err
+}
+
+func (e *Endpoints) CloseSprint(ctx context.Context, in *pb.CloseSprintRequest) (*pb.CloseSprintResponse, error) {
+	out, err := e.CloseSprintEndpoint(ctx, in)
+	if err != nil {
+		return &pb.CloseSprintResponse{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.CloseSprintReply), err
+}
+
+func (e *Endpoints) GetSprint(ctx context.Context, in *pb.GetSprintRequest) (*pb.GetSprintResponse, error) {
+	out, err := e.GetSprintEndpoint(ctx, in)
+	if err != nil {
+		return &pb.GetSprintResponse{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.GetSprintReply), err
+}
+
+func MakeAddSprintEndpoint(svc pb.SprintServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.AddSprintRequest)
+		rep, err := svc.AddSprint(ctx, req)
+		if err != nil {
+			return &pb.AddSprintReply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+
+func MakeCloseSprintEndpoint(svc pb.SprintServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.CloseSprintRequest)
+		rep, err := svc.CloseSprint(ctx, req)
+		if err != nil {
+			return &pb.CloseSprintReply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+
+func MakeGetSprintEndpoint(svc pb.SprintServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.GetSprintRequest)
+		rep, err := svc.GetSprint(ctx, req)
+		if err != nil {
+			return &pb.GetSprintReply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+
+func MakeEndpoints(svc pb.SprintServiceServer) Endpoints {
+	return Endpoints{
+
+		AddSprintEndpoint: MakeAddSprintEndpoint(svc),
+
+		CloseSprintEndpoint: MakeCloseSprintEndpoint(svc),
+
+		GetSprintEndpoint: MakeGetSprintEndpoint(svc),
+	}
+}

examples/go-kit/services/sprint/gen/init/init.go

@@ -1 +0,0 @@
-package sprint_init

examples/go-kit/services/sprint/gen/transports/grpc/grpc.go

@@ -1 +1,102 @@
 package sprint_transportgrpc
+
+import (
+	"fmt"
+
+	grpctransport "github.com/go-kit/kit/transport/grpc"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/sprint"
+	endpoint "github.com/moul/protoc-gen-gotemplate/examples/go-kit/sprint/gen/endpoints"
+	context "golang.org/x/net/context"
+)
+
+// avoid import errors
+var _ = fmt.Errorf
+
+func MakeGRPCServer(ctx context.Context, endpoints endpoint.Endpoints) pb.SprintServiceServer {
+	options := []grpctransport.ServerOption{}
+	return &grpcServer{
+
+		addsprint: grpctransport.NewServer(
+			ctx,
+			endpoints.AddSprintEndpoint,
+			decodeAddSprintRequest,
+			encodeAddSprintResponse,
+			options,
+		),
+
+		closesprint: grpctransport.NewServer(
+			ctx,
+			endpoints.CloseSprintEndpoint,
+			decodeCloseSprintRequest,
+			encodeCloseSprintResponse,
+			options,
+		),
+
+		getsprint: grpctransport.NewServer(
+			ctx,
+			endpoints.GetSprintEndpoint,
+			decodeGetSprintRequest,
+			encodeGetSprintResponse,
+			options,
+		),
+	}
+}
+
+type grpcServer struct {
+	addsprint grpctransport.Handler
+
+	closesprint grpctransport.Handler
+
+	getsprint grpctransport.Handler
+}
+
+func (s *grpcServer) AddSprint(ctx context.Context, req *pb.AddSprintRequest) (*pb.AddSprintReply, error) {
+	_, rep, err := s.addsprint.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.AddSprintReply), nil
+}
+
+func decodeAddSprintRequest(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encodeAddSprintResponse(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.AddSprintReply)
+	return resp, nil
+}
+
+func (s *grpcServer) CloseSprint(ctx context.Context, req *pb.CloseSprintRequest) (*pb.CloseSprintReply, error) {
+	_, rep, err := s.closesprint.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.CloseSprintReply), nil
+}
+
+func decodeCloseSprintRequest(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encodeCloseSprintResponse(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.CloseSprintReply)
+	return resp, nil
+}
+
+func (s *grpcServer) GetSprint(ctx context.Context, req *pb.GetSprintRequest) (*pb.GetSprintReply, error) {
+	_, rep, err := s.getsprint.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.GetSprintReply), nil
+}
+
+func decodeGetSprintRequest(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encodeGetSprintResponse(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.GetSprintReply)
+	return resp, nil
+}

examples/go-kit/services/sprint/gen/transports/http/http.go

@@ -1 +1,93 @@
 package sprint_transporthttp
+
+import (
+	"encoding/json"
+	context "golang.org/x/net/context"
+	"log"
+	"net/http"
+
+	gokit_endpoint "github.com/go-kit/kit/endpoint"
+	httptransport "github.com/go-kit/kit/transport/http"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/sprint"
+	endpoints "github.com/moul/protoc-gen-gotemplate/examples/go-kit/sprint/gen/endpoints"
+)
+
+func MakeAddSprintHandler(ctx context.Context, svc pb.SprintServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decodeAddSprintRequest,
+		encodeAddSprintResponse,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decodeAddSprintRequest(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.AddSprintRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encodeAddSprintResponse(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+
+func MakeCloseSprintHandler(ctx context.Context, svc pb.SprintServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decodeCloseSprintRequest,
+		encodeCloseSprintResponse,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decodeCloseSprintRequest(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.CloseSprintRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encodeCloseSprintResponse(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+
+func MakeGetSprintHandler(ctx context.Context, svc pb.SprintServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decodeGetSprintRequest,
+		encodeGetSprintResponse,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decodeGetSprintRequest(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.GetSprintRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encodeGetSprintResponse(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+
+func RegisterHandlers(ctx context.Context, svc pb.SprintServiceServer, mux *http.ServeMux, endpoints endpoints.Endpoints) error {
+
+	log.Println("new HTTP endpoint: \"/AddSprint\" (service=Sprint)")
+	mux.Handle("/AddSprint", MakeAddSprintHandler(ctx, svc, endpoints.AddSprintEndpoint))
+
+	log.Println("new HTTP endpoint: \"/CloseSprint\" (service=Sprint)")
+	mux.Handle("/CloseSprint", MakeCloseSprintHandler(ctx, svc, endpoints.CloseSprintEndpoint))
+
+	log.Println("new HTTP endpoint: \"/GetSprint\" (service=Sprint)")
+	mux.Handle("/GetSprint", MakeGetSprintHandler(ctx, svc, endpoints.GetSprintEndpoint))
+
+	return nil
+}