Filter disk stats by filesystem type, for Linux.

If we don't want to record the stats of NFS, tmpfs, /proc and
similar then specify:

fstypes = ["ext4", "xfs"]

In the config file.
This commit is contained in:
Simon Fraser
2015-07-30 17:02:58 +01:00
parent 6eb4bdcf0e
commit f1af559f11
3 changed files with 29 additions and 12 deletions

View File

@@ -17,7 +17,7 @@ const (
// Get disk partitions.
// should use setmntent(3) but this implement use /etc/mtab file
func DiskPartitions(all bool) ([]DiskPartitionStat, error) {
func DiskPartitions(fstypes []string) ([]DiskPartitionStat, error) {
filename := "/etc/mtab"
lines, err := common.ReadLines(filename)
@@ -25,17 +25,25 @@ func DiskPartitions(all bool) ([]DiskPartitionStat, error) {
return nil, err
}
set := make(map[string]struct{}, len(fstypes))
for _, s := range fstypes {
set[s] = struct{}{}
}
ret := make([]DiskPartitionStat, 0, len(lines))
for _, line := range lines {
fields := strings.Fields(line)
d := DiskPartitionStat{
Device: fields[0],
Mountpoint: fields[1],
Fstype: fields[2],
Opts: fields[3],
_, ok := set[fields[2]]
if ok || len(fstypes) == 0 {
d := DiskPartitionStat{
Device: fields[0],
Mountpoint: fields[1],
Fstype: fields[2],
Opts: fields[3],
}
ret = append(ret, d)
}
ret = append(ret, d)
}
return ret, nil