This repository has been archived by the owner on May 18, 2022. It is now read-only.
/
middlewares.go
72 lines (63 loc) · 1.53 KB
/
middlewares.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package api
import (
"context"
csrmodel "github.com/lamassuiot/device-manufacturing-system/pkg/enroller/models/csr"
"time"
"github.com/go-kit/kit/log"
)
type Middleware func(Service) Service
func LoggingMiddleware(logger log.Logger) Middleware {
return func(next Service) Service {
return &loggingMiddleware{
next: next,
logger: logger,
}
}
}
type loggingMiddleware struct {
next Service
logger log.Logger
}
func (mw loggingMiddleware) Health(ctx context.Context) (healthy bool) {
defer func(begin time.Time) {
mw.logger.Log(
"method", "Health",
"took", time.Since(begin),
"healthy", healthy,
)
}(time.Now())
return mw.next.Health(ctx)
}
func (mw loggingMiddleware) GetCSRs(ctx context.Context) (csrs csrmodel.CSRs) {
defer func(begin time.Time) {
mw.logger.Log(
"method", "GetCSRs",
"took", time.Since(begin),
)
}(time.Now())
return mw.next.GetCSRs(ctx)
}
func (mw loggingMiddleware) GetCSRStatus(ctx context.Context, id int) (csr csrmodel.CSR, err error) {
defer func(begin time.Time) {
mw.logger.Log(
"method", "GetCSRStatus",
"request_csr_id", id,
"response_csr_id", csr.Id,
"csr_status", csr.Status,
"took", time.Since(begin),
"err", err,
)
}(time.Now())
return mw.next.GetCSRStatus(ctx, id)
}
func (mw loggingMiddleware) GetCRT(ctx context.Context, id int) (data []byte, err error) {
defer func(begin time.Time) {
mw.logger.Log(
"method", "GetCRT",
"request_csr_id", id,
"took", time.Since(begin),
"err", err,
)
}(time.Now())
return mw.next.GetCRT(ctx, id)
}