forked from qiniu/logkit
/
memory.go
77 lines (65 loc) · 1.59 KB
/
memory.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
package system
import (
"fmt"
"github.com/qiniu/logkit/metric"
)
type MemStats struct {
ps PS
}
func (_ *MemStats) Name() string {
return "mem"
}
func (s *MemStats) Collect() (datas []map[string]interface{}, err error) {
vm, err := s.ps.VMStat()
if err != nil {
return nil, fmt.Errorf("error getting virtual memory info: %s", err)
}
fields := map[string]interface{}{
"total": vm.Total,
"available": vm.Available,
"used": vm.Used,
"free": vm.Free,
"cached": vm.Cached,
"buffered": vm.Buffers,
"active": vm.Active,
"inactive": vm.Inactive,
"used_percent": 100 * float64(vm.Used) / float64(vm.Total),
"available_percent": 100 * float64(vm.Available) / float64(vm.Total),
}
datas = append(datas, fields)
return
}
type SwapStats struct {
ps PS
}
func (_ *SwapStats) Name() string {
return "swap"
}
func (s *SwapStats) Collect() (datas []map[string]interface{}, err error) {
swap, err := s.ps.SwapStat()
if err != nil {
return nil, fmt.Errorf("error getting swap memory info: %s", err)
}
fieldsG := map[string]interface{}{
"total": swap.Total,
"used": swap.Used,
"free": swap.Free,
"used_percent": swap.UsedPercent,
}
datas = append(datas, fieldsG)
fieldsC := map[string]interface{}{
"in": swap.Sin,
"out": swap.Sout,
}
datas = append(datas, fieldsC)
return
}
func init() {
ps := newSystemPS()
metric.Add("mem", func() metric.Collector {
return &MemStats{ps: ps}
})
metric.Add("swap", func() metric.Collector {
return &SwapStats{ps: ps}
})
}