Generate endpoints + transports

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

@@ -1 +0,0 @@
-package session_client

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

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

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

@@ -1 +1,63 @@
 package session_endpoints
+
+import (
+	"fmt"
+	"github.com/go-kit/kit/endpoint"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/session"
+	context "golang.org/x/net/context"
+)
+
+var _ = fmt.Errorf
+
+type Endpoints struct {
+	LoginEndpoint endpoint.Endpoint
+
+	LogoutEndpoint endpoint.Endpoint
+}
+
+func (e *Endpoints) Login(ctx context.Context, in *pb.LoginRequest) (*pb.LoginResponse, error) {
+	out, err := e.LoginEndpoint(ctx, in)
+	if err != nil {
+		return &pb.LoginResponse{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.LoginReply), err
+}
+
+func (e *Endpoints) Logout(ctx context.Context, in *pb.LogoutRequest) (*pb.LogoutResponse, error) {
+	out, err := e.LogoutEndpoint(ctx, in)
+	if err != nil {
+		return &pb.LogoutResponse{ErrMsg: err.Error()}, err
+	}
+	return out.(*pb.LogoutReply), err
+}
+
+func MakeLoginEndpoint(svc pb.SessionServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.LoginRequest)
+		rep, err := svc.Login(ctx, req)
+		if err != nil {
+			return &pb.LoginReply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+
+func MakeLogoutEndpoint(svc pb.SessionServiceServer) endpoint.Endpoint {
+	return func(ctx context.Context, request interface{}) (interface{}, error) {
+		req := request.(*pb.LogoutRequest)
+		rep, err := svc.Logout(ctx, req)
+		if err != nil {
+			return &pb.LogoutReply{ErrMsg: err.Error()}, err
+		}
+		return rep, nil
+	}
+}
+
+func MakeEndpoints(svc pb.SessionServiceServer) Endpoints {
+	return Endpoints{
+
+		LoginEndpoint: MakeLoginEndpoint(svc),
+
+		LogoutEndpoint: MakeLogoutEndpoint(svc),
+	}
+}

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

@@ -1 +0,0 @@
-package session_init

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

@@ -1 +1,75 @@
 package session_transportgrpc
+
+import (
+	"fmt"
+
+	grpctransport "github.com/go-kit/kit/transport/grpc"
+	pb "github.com/moul/protoc-gen-gotemplate/examples/go-kit/services/session"
+	endpoint "github.com/moul/protoc-gen-gotemplate/examples/go-kit/session/gen/endpoints"
+	context "golang.org/x/net/context"
+)
+
+// avoid import errors
+var _ = fmt.Errorf
+
+func MakeGRPCServer(ctx context.Context, endpoints endpoint.Endpoints) pb.SessionServiceServer {
+	options := []grpctransport.ServerOption{}
+	return &grpcServer{
+
+		login: grpctransport.NewServer(
+			ctx,
+			endpoints.LoginEndpoint,
+			decodeLoginRequest,
+			encodeLoginResponse,
+			options,
+		),
+
+		logout: grpctransport.NewServer(
+			ctx,
+			endpoints.LogoutEndpoint,
+			decodeLogoutRequest,
+			encodeLogoutResponse,
+			options,
+		),
+	}
+}
+
+type grpcServer struct {
+	login grpctransport.Handler
+
+	logout grpctransport.Handler
+}
+
+func (s *grpcServer) Login(ctx context.Context, req *pb.LoginRequest) (*pb.LoginReply, error) {
+	_, rep, err := s.login.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.LoginReply), nil
+}
+
+func decodeLoginRequest(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encodeLoginResponse(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.LoginReply)
+	return resp, nil
+}
+
+func (s *grpcServer) Logout(ctx context.Context, req *pb.LogoutRequest) (*pb.LogoutReply, error) {
+	_, rep, err := s.logout.ServeGRPC(ctx, req)
+	if err != nil {
+		return nil, err
+	}
+	return rep.(*pb.LogoutReply), nil
+}
+
+func decodeLogoutRequest(ctx context.Context, grpcReq interface{}) (interface{}, error) {
+	return grpcReq, nil
+}
+
+func encodeLogoutResponse(ctx context.Context, response interface{}) (interface{}, error) {
+	resp := response.(*pb.LogoutReply)
+	return resp, nil
+}

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

@@ -1 +1,68 @@
 package session_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/session"
+	endpoints "github.com/moul/protoc-gen-gotemplate/examples/go-kit/session/gen/endpoints"
+)
+
+func MakeLoginHandler(ctx context.Context, svc pb.SessionServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decodeLoginRequest,
+		encodeLoginResponse,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decodeLoginRequest(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.LoginRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encodeLoginResponse(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+
+func MakeLogoutHandler(ctx context.Context, svc pb.SessionServiceServer, endpoint gokit_endpoint.Endpoint) *httptransport.Server {
+	return httptransport.NewServer(
+		ctx,
+		endpoint,
+		decodeLogoutRequest,
+		encodeLogoutResponse,
+		append([]httptransport.ServerOption{}, httptransport.ServerBefore(jwt.ToHTTPContext()))...,
+	)
+}
+
+func decodeLogoutRequest(ctx context.Context, r *http.Request) (interface{}, error) {
+	var req pb.LogoutRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		return nil, err
+	}
+	return &req, nil
+}
+
+func encodeLogoutResponse(ctx context.Context, w http.ResponseWriter, response interface{}) error {
+	return json.NewEncoder(w).Encode(response)
+}
+
+func RegisterHandlers(ctx context.Context, svc pb.SessionServiceServer, mux *http.ServeMux, endpoints endpoints.Endpoints) error {
+
+	log.Println("new HTTP endpoint: \"/Login\" (service=Session)")
+	mux.Handle("/Login", MakeLoginHandler(ctx, svc, endpoints.LoginEndpoint))
+
+	log.Println("new HTTP endpoint: \"/Logout\" (service=Session)")
+	mux.Handle("/Logout", MakeLogoutHandler(ctx, svc, endpoints.LogoutEndpoint))
+
+	return nil
+}