110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
package disk
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
|
|
"flashcat.cloud/categraf/config"
|
|
"flashcat.cloud/categraf/inputs"
|
|
"flashcat.cloud/categraf/inputs/system"
|
|
"flashcat.cloud/categraf/pkg/choice"
|
|
"flashcat.cloud/categraf/types"
|
|
)
|
|
|
|
type DiskStats struct {
|
|
ps system.PS
|
|
|
|
config.PluginConfig
|
|
MountPoints []string `toml:"mount_points"`
|
|
IgnoreFS []string `toml:"ignore_fs"`
|
|
IgnoreMountPoints []string `toml:"ignore_mount_points"`
|
|
}
|
|
|
|
func init() {
|
|
ps := system.NewSystemPS()
|
|
inputs.Add("disk", func() inputs.Input {
|
|
return &DiskStats{
|
|
ps: ps,
|
|
}
|
|
})
|
|
}
|
|
|
|
// just placeholder
|
|
func (s *DiskStats) GetInstances() []inputs.Instance {
|
|
return nil
|
|
}
|
|
|
|
func (s *DiskStats) Init() error {
|
|
return nil
|
|
}
|
|
|
|
func (s *DiskStats) Drop() {
|
|
}
|
|
|
|
func (s *DiskStats) Gather(slist *types.SampleList) {
|
|
disks, partitions, err := s.ps.DiskUsage(s.MountPoints, s.IgnoreFS)
|
|
if err != nil {
|
|
log.Println("E! failed to get disk usage:", err)
|
|
return
|
|
}
|
|
|
|
for i, du := range disks {
|
|
if du.Total == 0 {
|
|
// Skip dummy filesystem (procfs, cgroupfs, ...)
|
|
continue
|
|
}
|
|
|
|
if len(s.IgnoreMountPoints) > 0 {
|
|
if choice.Contains(du.Path, s.IgnoreMountPoints) {
|
|
continue
|
|
}
|
|
}
|
|
|
|
mountOpts := MountOptions(partitions[i].Opts)
|
|
tags := map[string]string{
|
|
"path": du.Path,
|
|
"device": strings.Replace(partitions[i].Device, "/dev/", "", -1),
|
|
"fstype": du.Fstype,
|
|
"mode": mountOpts.Mode(),
|
|
}
|
|
var usedPercent float64
|
|
if du.Used+du.Free > 0 {
|
|
usedPercent = float64(du.Used) /
|
|
(float64(du.Used) + float64(du.Free)) * 100
|
|
}
|
|
|
|
fields := map[string]interface{}{
|
|
"total": du.Total,
|
|
"free": du.Free,
|
|
"used": du.Used,
|
|
"used_percent": usedPercent,
|
|
"inodes_total": du.InodesTotal,
|
|
"inodes_free": du.InodesFree,
|
|
"inodes_used": du.InodesUsed,
|
|
}
|
|
|
|
slist.PushSamples("disk", fields, tags)
|
|
}
|
|
}
|
|
|
|
type MountOptions []string
|
|
|
|
func (opts MountOptions) Mode() string {
|
|
if opts.exists("rw") {
|
|
return "rw"
|
|
} else if opts.exists("ro") {
|
|
return "ro"
|
|
} else {
|
|
return "unknown"
|
|
}
|
|
}
|
|
|
|
func (opts MountOptions) exists(opt string) bool {
|
|
for _, o := range opts {
|
|
if o == opt {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|