blob: 42ab3de6112f770356b944eb2eee058c3528cc49 (
plain)
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
|
package statsclient
import (
"sync/atomic"
"time"
"unsafe"
)
var (
MaxWaitInProgress = time.Millisecond * 100
CheckDelayInProgress = time.Microsecond * 10
)
type sharedHeaderBase struct {
epoch int64
inProgress int64
directoryOffset int64
errorOffset int64
statsOffset int64
}
type sharedHeaderV0 struct {
sharedHeaderBase
}
type sharedHeader struct {
version uint64
sharedHeaderBase
}
func (h *sharedHeader) legacyVersion() bool {
// older VPP (<=19.04) did not have version in stat segment header
// we try to provide fallback support by skipping it in header
if h.version > maxVersion && h.inProgress > 1 && h.epoch == 0 {
return true
}
return false
}
func loadSharedHeader(b []byte) (header sharedHeader) {
h := (*sharedHeader)(unsafe.Pointer(&b[0]))
header.version = atomic.LoadUint64(&h.version)
header.epoch = atomic.LoadInt64(&h.epoch)
header.inProgress = atomic.LoadInt64(&h.inProgress)
header.directoryOffset = atomic.LoadInt64(&h.directoryOffset)
header.errorOffset = atomic.LoadInt64(&h.errorOffset)
header.statsOffset = atomic.LoadInt64(&h.statsOffset)
return
}
func loadSharedHeaderLegacy(b []byte) (header sharedHeader) {
h := (*sharedHeaderV0)(unsafe.Pointer(&b[0]))
header.version = 0
header.epoch = atomic.LoadInt64(&h.epoch)
header.inProgress = atomic.LoadInt64(&h.inProgress)
header.directoryOffset = atomic.LoadInt64(&h.directoryOffset)
header.errorOffset = atomic.LoadInt64(&h.errorOffset)
header.statsOffset = atomic.LoadInt64(&h.statsOffset)
return
}
type statSegAccess struct {
epoch int64
}
func (c *statSegment) accessStart() statSegAccess {
t := time.Now()
epoch, inprog := c.getEpoch()
for inprog {
if time.Since(t) > MaxWaitInProgress {
return statSegAccess{}
} else {
time.Sleep(CheckDelayInProgress)
}
epoch, inprog = c.getEpoch()
}
return statSegAccess{
epoch: epoch,
}
}
func (c *statSegment) accessEnd(acc *statSegAccess) bool {
epoch, inprog := c.getEpoch()
if acc.epoch != epoch || inprog {
return false
}
return true
}
type vecHeader struct {
length uint64
vectorData [0]uint8
}
func vectorLen(v unsafe.Pointer) uint64 {
vec := *(*vecHeader)(unsafe.Pointer(uintptr(v) - unsafe.Sizeof(uint64(0))))
return vec.length
}
//go:nosplit
func statSegPointer(p unsafe.Pointer, offset uintptr) unsafe.Pointer {
return unsafe.Pointer(uintptr(p) + offset)
}
|