-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathevent.go
More file actions
125 lines (111 loc) · 2.1 KB
/
event.go
File metadata and controls
125 lines (111 loc) · 2.1 KB
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
125
package hc
import (
"fmt"
"maps"
"sync"
"time"
)
// Event accumulates request-scoped structured fields.
type Event struct {
mu sync.RWMutex
fields map[string]any
startTime time.Time
hasError bool
requestedLevel Level
hasRequestedLevel bool
}
type snapshot struct {
fields map[string]any
startTime time.Time
hasError bool
}
func newEvent() *Event {
return &Event{
startTime: time.Now(),
}
}
func (e *Event) addKV(key string, value any, kv ...any) bool {
if len(kv)%2 != 0 {
return false
}
for i := 0; i < len(kv); i += 2 {
if _, ok := kv[i].(string); !ok {
return false
}
}
e.mu.Lock()
defer e.mu.Unlock()
if e.fields == nil {
pairs := 1 + len(kv)/2
capHint := 8
if pairs > capHint {
capHint = pairs
}
e.fields = make(map[string]any, capHint)
}
e.fields[key] = value
for i := 0; i < len(kv); i += 2 {
e.fields[kv[i].(string)] = kv[i+1]
}
return true
}
func (e *Event) setRoute(route string) {
if route == "" {
return
}
e.mu.Lock()
if e.fields == nil {
e.fields = make(map[string]any, 8)
}
e.fields["http.route"] = route
e.mu.Unlock()
}
func (e *Event) setError(err error) {
if err == nil {
return
}
e.mu.Lock()
defer e.mu.Unlock()
if e.fields == nil {
e.fields = make(map[string]any, 8)
}
e.hasError = true
e.fields["error"] = map[string]any{
"message": err.Error(),
"type": fmt.Sprintf("%T", err),
}
}
func (e *Event) hasErrorValue() bool {
e.mu.RLock()
defer e.mu.RUnlock()
return e.hasError
}
func (e *Event) startedAt() time.Time {
e.mu.RLock()
defer e.mu.RUnlock()
return e.startTime
}
func (e *Event) setLevel(level Level) bool {
if !isValidLevel(level) {
return false
}
e.mu.Lock()
e.requestedLevel = level
e.hasRequestedLevel = true
e.mu.Unlock()
return true
}
func (e *Event) requestedLevelValue() (Level, bool) {
e.mu.RLock()
defer e.mu.RUnlock()
return e.requestedLevel, e.hasRequestedLevel
}
func (e *Event) snapshot() snapshot {
e.mu.RLock()
defer e.mu.RUnlock()
return snapshot{
fields: maps.Clone(e.fields),
startTime: e.startTime,
hasError: e.hasError,
}
}