add endpoint logging middleware to log total execution time of an end…

…point

pkg/middleware/logging.go

@@ -0,0 +1,20 @@
+package middleware
+
+import (
+	"github.com/lukasjarosch/godin/pkg/log"
+	"github.com/go-kit/kit/endpoint"
+	"context"
+	"time"
+)
+
+func Logging(logger log.Logger, endpointName string) endpoint.Middleware {
+	return func(next endpoint.Endpoint) endpoint.Endpoint {
+		return func(ctx context.Context, request interface{}) (response interface{}, err error) {
+			defer func(begin time.Time) {
+				logger.Info("", "endpoint", endpointName, "took", time.Since(begin))
+			}(time.Now())
+
+			return next(ctx, request)
+		}
+	}
+}

templates/cmd_main.tpl

@@ -44,7 +44,7 @@ func main() {
 
 	// initialize endpoint and transport layers
 	var (
-		endpoints   = endpoint.Endpoints(svc)
+		endpoints   = endpoint.Endpoints(svc, logger)
 		grpcHandler = svcGrpc.NewServer(endpoints, logger)
 	)
 

templates/endpoint_set.tpl

@@ -7,6 +7,7 @@ import (
 
     godinMiddleware "github.com/lukasjarosch/godin/pkg/middleware"
     "{{ .Service.Module }}/internal/service"
+	"github.com/lukasjarosch/godin/pkg/log"
 )
 
 type Set struct {
@@ -15,12 +16,13 @@ type Set struct {
      {{- end }}
 }
 
-func Endpoints(service service.{{ title .Service.Name }}) Set {
+func Endpoints(service service.{{ title .Service.Name }}, logger log.Logger) Set {
     {{ range .Service.Methods }}
     var {{ untitle .Name }} endpoint.Endpoint
     {
         {{ untitle .Name }} = {{ .Name }}Endpoint(service)
         {{ untitle .Name }} = godinMiddleware.Prometheus("{{ .Name }}")({{ untitle .Name }})
+        {{ untitle .Name }} = godinMiddleware.Logging(logger, "{{ .Name }}")({{ untitle .Name }})
     }
     {{- end }}