vendor/github.com/go-kit/kit/examples/addsvc/pkg/addservice/middleware.go

   1 package addservice
   2
   3 import (
   4         "context"
   5
   6         "github.com/go-kit/kit/log"
   7         "github.com/go-kit/kit/metrics"
   8 )
   9
  10 // Middleware describes a service (as opposed to endpoint) middleware.
  11 type Middleware func(Service) Service
  12
  13 // LoggingMiddleware takes a logger as a dependency
  14 // and returns a ServiceMiddleware.
  15 func LoggingMiddleware(logger log.Logger) Middleware {
  16         return func(next Service) Service {
  17                 return loggingMiddleware{logger, next}
  18         }
  19 }
  20
  21 type loggingMiddleware struct {
  22         logger log.Logger
  23         next   Service
  24 }
  25
  26 func (mw loggingMiddleware) Sum(ctx context.Context, a, b int) (v int, err error) {
  27         defer func() {
  28                 mw.logger.Log("method", "Sum", "a", a, "b", b, "v", v, "err", err)
  29         }()
  30         return mw.next.Sum(ctx, a, b)
  31 }
  32
  33 func (mw loggingMiddleware) Concat(ctx context.Context, a, b string) (v string, err error) {
  34         defer func() {
  35                 mw.logger.Log("method", "Concat", "a", a, "b", b, "v", v, "err", err)
  36         }()
  37         return mw.next.Concat(ctx, a, b)
  38 }
  39
  40 // InstrumentingMiddleware returns a service middleware that instruments
  41 // the number of integers summed and characters concatenated over the lifetime of
  42 // the service.
  43 func InstrumentingMiddleware(ints, chars metrics.Counter) Middleware {
  44         return func(next Service) Service {
  45                 return instrumentingMiddleware{
  46                         ints:  ints,
  47                         chars: chars,
  48                         next:  next,
  49                 }
  50         }
  51 }
  52
  53 type instrumentingMiddleware struct {
  54         ints  metrics.Counter
  55         chars metrics.Counter
  56         next  Service
  57 }
  58
  59 func (mw instrumentingMiddleware) Sum(ctx context.Context, a, b int) (int, error) {
  60         v, err := mw.next.Sum(ctx, a, b)
  61         mw.ints.Add(float64(v))
  62         return v, err
  63 }
  64
  65 func (mw instrumentingMiddleware) Concat(ctx context.Context, a, b string) (string, error) {
  66         v, err := mw.next.Concat(ctx, a, b)
  67         mw.chars.Add(float64(len(v)))
  68         return v, err
  69 }