-
-
Notifications
You must be signed in to change notification settings - Fork 623
/
props.go
54 lines (45 loc) · 1.26 KB
/
props.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
package log
import (
"context"
"github.com/getfider/fider/app"
"github.com/getfider/fider/app/models/dto"
)
const (
// PropertyKeySessionID is the session id of current logger
PropertyKeySessionID = "SessionID"
// PropertyKeyContextID is the context id of current logger
PropertyKeyContextID = "ContextID"
// PropertyKeyUserID is the user id of current logger
PropertyKeyUserID = "UserID"
// PropertyKeyTenantID is the tenant id of current logger
PropertyKeyTenantID = "TenantID"
// PropertyKeyTag is the tag of current logger
PropertyKeyTag = "Tag"
)
func GetProperties(ctx context.Context) dto.Props {
props, ok := ctx.Value(app.LogPropsCtxKey).(*dto.Props)
if ok {
return *props
}
return dto.Props{}
}
func GetProperty(ctx context.Context, key string) interface{} {
props := GetProperties(ctx)
return props[key]
}
func WithProperty(ctx context.Context, key string, value interface{}) context.Context {
return WithProperties(ctx, dto.Props{
key: value,
})
}
func WithProperties(ctx context.Context, kv dto.Props) context.Context {
props, ok := ctx.Value(app.LogPropsCtxKey).(*dto.Props)
if !ok {
props = &dto.Props{}
ctx = context.WithValue(ctx, app.LogPropsCtxKey, props)
}
for key, value := range kv {
(*props)[key] = value
}
return ctx
}