67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
|
package system
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"syscall"
|
||
|
|
)
|
||
|
|
|
||
|
|
type diskInfoCache struct {
|
||
|
|
stat syscall.Stat_t
|
||
|
|
values map[string]string
|
||
|
|
}
|
||
|
|
|
||
|
|
var udevPath = "/run/udev/data"
|
||
|
|
|
||
|
|
func (s *DiskIOStats) diskInfo(devName string) (map[string]string, error) {
|
||
|
|
fi, err := os.Stat("/dev/" + devName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
stat, ok := fi.Sys().(*syscall.Stat_t)
|
||
|
|
if !ok {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if s.infoCache == nil {
|
||
|
|
s.infoCache = map[string]diskInfoCache{}
|
||
|
|
}
|
||
|
|
ic, ok := s.infoCache[devName]
|
||
|
|
if ok {
|
||
|
|
return ic.values, nil
|
||
|
|
} else {
|
||
|
|
ic = diskInfoCache{
|
||
|
|
stat: *stat,
|
||
|
|
values: map[string]string{},
|
||
|
|
}
|
||
|
|
s.infoCache[devName] = ic
|
||
|
|
}
|
||
|
|
di := ic.values
|
||
|
|
|
||
|
|
major := stat.Rdev >> 8 & 0xff
|
||
|
|
minor := stat.Rdev & 0xff
|
||
|
|
|
||
|
|
f, err := os.Open(fmt.Sprintf("%s/b%d:%d", udevPath, major, minor))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
scnr := bufio.NewScanner(f)
|
||
|
|
|
||
|
|
for scnr.Scan() {
|
||
|
|
l := scnr.Text()
|
||
|
|
if len(l) < 4 || l[:2] != "E:" {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
kv := strings.SplitN(l[2:], "=", 2)
|
||
|
|
if len(kv) < 2 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
di[kv[0]] = kv[1]
|
||
|
|
}
|
||
|
|
|
||
|
|
return di, nil
|
||
|
|
}
|