/
logger.go
85 lines (71 loc) · 1.8 KB
/
logger.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
73
74
75
76
77
78
79
80
81
82
83
84
85
package lib
import (
"fmt"
"io"
"net/http"
"time"
"github.com/google/uuid"
)
type LoggingMiddleware struct {
database Database
infoLevel InfoLevel
// Prints logs to stdout.
debug bool
}
type LoggingMiddlewareBuilder struct {
database Database
infoLevel InfoLevel
// Prints logs to stdout.
debug bool
}
func (b LoggingMiddlewareBuilder) Build() *LoggingMiddleware {
return &LoggingMiddleware{b.database, b.infoLevel, b.debug}
}
func (b LoggingMiddlewareBuilder) WithInfoLevel(infoLevel InfoLevel) *LoggingMiddlewareBuilder {
b.infoLevel = infoLevel
return &b
}
func (b LoggingMiddlewareBuilder) WithDebug(debug bool) *LoggingMiddlewareBuilder {
b.debug = debug
return &b
}
func NewLoggingMiddlewareBuilder(database Database) *LoggingMiddlewareBuilder {
return &LoggingMiddlewareBuilder{database, Minimal, false}
}
// LogRoute middleware logs the request to the database.
func (q LoggingMiddleware) LogRoute(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
stamp := time.Now().UTC()
info := "Request to " + r.URL.Path + " from " + r.RemoteAddr
switch q.infoLevel {
case Verbose:
info += "\nBody:\n"
bodyBytes, err := io.ReadAll(r.Body)
if err == nil {
info += string(bodyBytes)
} else {
info += "Error reading body: " + err.Error()
}
case Normal:
info += "\nHeaders:\n"
for k, v := range r.Header {
info += k + ": " + v[0] + "\n"
}
default: // Minimal, don't do anything
}
if q.debug {
fmt.Println(stamp.String() + ": \n" + info)
}
// Truncate info to 255 characters.
if len(info) > 255 {
info = info[:255]
}
q.database.Write(r.Context(), LogSchema{
ID: uuid.NewString(),
TimeWritten: stamp,
Category: "api",
Info: info,
})
next.ServeHTTP(w, r)
})
}