77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
//go:generate ../../../tools/readme_config_includer/generator
|
|
//go:build linux
|
|
|
|
package kernel_vmstat
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
)
|
|
|
|
//go:embed sample.conf
|
|
var sampleConfig string
|
|
|
|
type KernelVmstat struct {
|
|
statFile string
|
|
}
|
|
|
|
func (*KernelVmstat) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (k *KernelVmstat) Gather(acc telegraf.Accumulator) error {
|
|
data, err := k.getProcVmstat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fields := make(map[string]interface{})
|
|
|
|
dataFields := bytes.Fields(data)
|
|
for i, field := range dataFields {
|
|
// dataFields is an array of {"stat1_name", "stat1_value", "stat2_name",
|
|
// "stat2_value", ...}
|
|
// We only want the even number index as that contain the stat name.
|
|
if i%2 == 0 {
|
|
// Convert the stat value into an integer.
|
|
m, err := strconv.ParseInt(string(dataFields[i+1]), 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fields[string(field)] = m
|
|
}
|
|
}
|
|
|
|
acc.AddFields("kernel_vmstat", fields, make(map[string]string))
|
|
return nil
|
|
}
|
|
|
|
func (k *KernelVmstat) getProcVmstat() ([]byte, error) {
|
|
if _, err := os.Stat(k.statFile); os.IsNotExist(err) {
|
|
return nil, fmt.Errorf("kernel_vmstat: %s does not exist", k.statFile)
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data, err := os.ReadFile(k.statFile)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("kernel_vmstat", func() telegraf.Input {
|
|
return &KernelVmstat{
|
|
statFile: "/proc/vmstat",
|
|
}
|
|
})
|
|
}
|