-
Notifications
You must be signed in to change notification settings - Fork 2
/
cgroup_io.go
123 lines (115 loc) · 2.5 KB
/
cgroup_io.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
package cgroup
import (
"os"
"path"
"strconv"
"strings"
"k8s.io/klog/v2"
)
type IOStat struct {
ReadOps uint64
WriteOps uint64
ReadBytes uint64
WrittenBytes uint64
}
func (cg *Cgroup) IOStat() (map[string]IOStat, error) {
if cg.Version == V1 {
return cg.ioStatV1()
}
return cg.ioStatV2()
}
func (cg *Cgroup) ioStatV1() (map[string]IOStat, error) {
ops, err := readBlkioStatFile(path.Join(cg.cgRoot, "blkio", cg.subsystems["blkio"], "blkio.throttle.io_serviced"))
if err != nil {
return nil, err
}
bytes, err := readBlkioStatFile(path.Join(cg.cgRoot, "blkio", cg.subsystems["blkio"], "blkio.throttle.io_service_bytes"))
if err != nil {
return nil, err
}
res := map[string]IOStat{}
for _, v := range ops {
stat := res[v.majorMinor]
switch v.name {
case "Read":
stat.ReadOps = v.value
case "Write":
stat.WriteOps = v.value
}
res[v.majorMinor] = stat
}
for _, v := range bytes {
stat := res[v.majorMinor]
switch v.name {
case "Read":
stat.ReadBytes = v.value
case "Write":
stat.WrittenBytes = v.value
}
res[v.majorMinor] = stat
}
return res, nil
}
func (cg *Cgroup) ioStatV2() (map[string]IOStat, error) {
payload, err := os.ReadFile(path.Join(cg.cgRoot, cg.subsystems[""], "io.stat"))
if err != nil {
return nil, err
}
res := map[string]IOStat{}
for _, line := range strings.Split(string(payload), "\n") {
parts := strings.Fields(line)
if len(parts) < 5 {
continue
}
s := IOStat{}
for _, value := range parts[1:] {
if kv := strings.SplitN(value, "=", 2); len(kv) == 2 {
v, err := strconv.ParseUint(kv[1], 10, 64)
if err != nil {
continue
}
switch kv[0] {
case "rbytes":
s.ReadBytes = v
case "wbytes":
s.WrittenBytes = v
case "rios":
s.ReadOps = v
case "wios":
s.WriteOps = v
}
}
}
res[parts[0]] = s
}
return res, nil
}
type blkioVariable struct {
majorMinor string
name string
value uint64
}
func readBlkioStatFile(filePath string) ([]blkioVariable, error) {
data, err := os.ReadFile(filePath)
if err != nil {
return nil, err
}
var res []blkioVariable
for _, line := range strings.Split(string(data), "\n") {
parts := strings.Fields(line)
if len(parts) != 3 {
continue
}
v, err := strconv.ParseUint(parts[2], 10, 64)
if err != nil {
klog.Warningf(`failed to parse blkio stat line "%s": %s`, line, err)
continue
}
res = append(res, blkioVariable{
majorMinor: parts[0],
name: parts[1],
value: v,
})
}
return res, nil
}