-
Notifications
You must be signed in to change notification settings - Fork 0
/
log.go
124 lines (109 loc) · 2.76 KB
/
log.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main
import (
"fmt"
log "github.com/sirupsen/logrus"
"net/http"
"regexp"
"strings"
"time"
)
type responseWriter struct {
http.ResponseWriter
status int
wroteHeader bool
header string
}
// Setup logging level and format
func Setup() {
switch GlobalConfig.LogFormat {
case "json":
log.SetFormatter(&log.JSONFormatter{DisableTimestamp: false})
case "text":
log.SetFormatter(&log.TextFormatter{FullTimestamp: true, DisableColors: true})
case "color":
log.SetFormatter(&log.TextFormatter{FullTimestamp: true, ForceColors: true})
default:
log.SetFormatter(&log.TextFormatter{FullTimestamp: true, DisableColors: true})
}
var logLevel log.Level
switch strings.ToLower(GlobalConfig.LogLevel) {
case "debug":
logLevel = log.DebugLevel
case "trace":
logLevel = log.TraceLevel
default:
logLevel = log.InfoLevel
}
log.SetLevel(logLevel)
}
func wrapResponseWriter(w http.ResponseWriter) *responseWriter {
return &responseWriter{ResponseWriter: w}
}
func (rw *responseWriter) Status() int {
return rw.status
}
func (rw *responseWriter) WriteHeader(code int) {
if rw.wroteHeader {
return
}
rw.status = code
rw.ResponseWriter.WriteHeader(code)
rw.wroteHeader = true
}
// Logs the incoming HTTP request and part of response
func LoggingMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
defer func() {
if err := recover(); err != nil {
w.WriteHeader(http.StatusInternalServerError)
log.Errorf("err %s", err)
}
}()
start := time.Now()
wrapped := wrapResponseWriter(w)
next.ServeHTTP(wrapped, r)
logHTTPReq(wrapped, r, start)
})
}
func logHTTPReq(w *responseWriter, r *http.Request, startTime time.Time) {
if skip, _ := regexp.MatchString("^/health[/]?$", r.URL.Path); skip {
return
}
log.Tracef("%+v", r)
if skipHTTPRequestLogging(r) {
return
}
remoteAddr := r.RemoteAddr
if r.Header.Get("x-real-ip") != "" {
remoteAddr = r.Header.Get("x-real-ip")
}
logentry := fmt.Sprintf("%s %s %s %s %d %v",
remoteAddr,
r.Host,
r.Method,
r.URL.EscapedPath(),
w.status,
time.Since(startTime))
if r.Header.Get("Referer") != "" {
logentry += fmt.Sprintf(" referer:%s", r.Header.Get("Referer"))
}
if r.Header.Get("x-original-uri") != "" {
logentry += fmt.Sprintf(" x-original-uri:%s", r.Header.Get("x-original-uri"))
}
if w.Header().Get("x-accel-redirect") != "" {
logentry += fmt.Sprintf(" x-redirect:%s", w.Header().Get("X-Accel-Redirect"))
}
log.Infoln(logentry)
}
// Checks to skip logging some requests
func skipHTTPRequestLogging(r *http.Request) bool {
switch r.URL.String() {
case "/favicon.png":
return true
case "/favicon.ico":
return true
case "/health":
return true
}
return strings.HasPrefix(r.URL.String(), "/favicon-")
}