-
Notifications
You must be signed in to change notification settings - Fork 16
/
disk.go
85 lines (74 loc) · 2.09 KB
/
disk.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
/*
* Copyright 2014-2019 Li Kexian
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Go module for collecting host stat
* https://www.likexian.com/
*/
package hoststat
import (
"github.com/likexian/gokit/xhuman"
"io/ioutil"
"strings"
"syscall"
)
// DiskStat storing disk stat
type DiskStat struct {
Label string `json:"label"`
Mount string `json:"mount"`
Total uint64 `json:"total"`
Used uint64 `json:"used"`
Free uint64 `json:"free"`
UsedRate float64 `json:"used_rate"`
}
// GetDiskStat returns all disk stat
func GetDiskStat() (stat []DiskStat, err error) {
text, err := ioutil.ReadFile("/etc/mtab")
if err != nil {
return
}
stat = []DiskStat{}
lines := strings.Split(string(text), "\n")
for i := 0; i < len(lines); i++ {
lines[i] = strings.TrimSpace(lines[i])
if lines[i] == "" {
continue
}
fields := strings.Fields(lines[i])
if len(fields) < 3 {
continue
}
if fields[2] != "proc" && fields[2] != "cgroup" && fields[2] != "sysfs" && fields[2] != "devpts" && fields[2] != "mqueue" {
diskStat, _ := getStat(fields[1])
diskStat.Label = fields[0]
stat = append(stat, diskStat)
}
}
return
}
// getStat returns the path stat
func getStat(path string) (stat DiskStat, err error) {
fs := syscall.Statfs_t{}
err = syscall.Statfs(path, &fs)
if err != nil {
return
}
stat = DiskStat{}
stat.Mount = path
stat.Total = fs.Blocks * uint64(fs.Bsize) / (1024 * 1024)
stat.Free = fs.Bfree * uint64(fs.Bsize) / (1024 * 1024)
stat.Used = stat.Total - stat.Free
stat.UsedRate = xhuman.Round(float64(stat.Used)*100/float64(stat.Total), 2)
return
}