/
stats.go
125 lines (114 loc) · 2.71 KB
/
stats.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
124
125
package spy
import (
"encoding/json"
"github.com/go-well/spider/silk"
"github.com/shirou/gopsutil/cpu"
"github.com/shirou/gopsutil/disk"
"github.com/shirou/gopsutil/host"
"github.com/shirou/gopsutil/mem"
"github.com/shirou/gopsutil/net"
)
func init() {
RegisterHandler(silk.StatsHost, func(c *Client, p *silk.Package) {
p.Type = silk.StatsHostAck
info, err := host.Info()
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsCpu, func(c *Client, p *silk.Package) {
p.Type = silk.StatsCpuAck
info, err := cpu.Info()
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsCpuUsage, func(c *Client, p *silk.Package) {
p.Type = silk.StatsCpuUsageAck
info, err := cpu.Times(true)
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsMem, func(c *Client, p *silk.Package) {
p.Type = silk.StatsMemAck
info, err := mem.VirtualMemory()
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsDisk, func(c *Client, p *silk.Package) {
p.Type = silk.StatsDiskAck
info, err := disk.Partitions(true)
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsDiskUsage, func(c *Client, p *silk.Package) {
p.Type = silk.StatsDiskUsageAck
path := string(p.Data)
info, err := disk.Usage(path)
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsDiskIO, func(c *Client, p *silk.Package) {
p.Type = silk.StatsDiskIOAck
path := string(p.Data)
info, err := disk.IOCounters(path)
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsNet, func(c *Client, p *silk.Package) {
p.Type = silk.StatsNetAck
info, err := net.Interfaces()
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsNetIO, func(c *Client, p *silk.Package) {
p.Type = silk.StatsNetIOAck
info, err := net.IOCounters(true)
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
RegisterHandler(silk.StatsUser, func(c *Client, p *silk.Package) {
p.Type = silk.StatsUserAck
info, err := host.Users()
if err != nil {
p.SetError(err.Error())
} else {
p.Data, _ = json.Marshal(info)
}
_ = c.Send(p)
})
}