-
Notifications
You must be signed in to change notification settings - Fork 784
/
wrappers.go
65 lines (54 loc) · 1.77 KB
/
wrappers.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
package log
import (
"context"
"github.com/go-kit/log"
kitlog "github.com/go-kit/log"
"github.com/weaveworks/common/tracing"
"github.com/cortexproject/cortex/pkg/tenant"
)
// WithUserID returns a Logger that has information about the current user in
// its details.
func WithUserID(userID string, l kitlog.Logger) kitlog.Logger {
// See note in WithContext.
return kitlog.With(l, "org_id", userID)
}
// WithTraceID returns a Logger that has information about the traceID in
// its details.
func WithTraceID(traceID string, l kitlog.Logger) kitlog.Logger {
// See note in WithContext.
return kitlog.With(l, "traceID", traceID)
}
// WithContext returns a Logger that has information about the current user in
// its details.
//
// e.g.
//
// log := util.WithContext(ctx)
// log.Errorf("Could not chunk chunks: %v", err)
func WithContext(ctx context.Context, l kitlog.Logger) kitlog.Logger {
l = headersFromContext(ctx, l)
// Weaveworks uses "orgs" and "orgID" to represent Cortex users,
// even though the code-base generally uses `userID` to refer to the same thing.
userID, err := tenant.TenantID(ctx)
if err == nil {
l = WithUserID(userID, l)
}
traceID, ok := tracing.ExtractSampledTraceID(ctx)
if !ok {
return l
}
return WithTraceID(traceID, l)
}
// WithSourceIPs returns a Logger that has information about the source IPs in
// its details.
func WithSourceIPs(sourceIPs string, l log.Logger) log.Logger {
return log.With(l, "sourceIPs", sourceIPs)
}
// HeadersFromContext enables the logging of specified HTTP Headers that have been added to a context
func headersFromContext(ctx context.Context, l log.Logger) log.Logger {
headerContentsMap := HeaderMapFromContext(ctx)
for header, contents := range headerContentsMap {
l = kitlog.With(l, header, contents)
}
return l
}