78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
//go:generate ../../../tools/readme_config_includer/generator
|
||
|
|
package hddtemp
|
||
|
|
|
||
|
|
import (
|
||
|
|
_ "embed"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf"
|
||
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
||
|
|
gohddtemp "github.com/influxdata/telegraf/plugins/inputs/hddtemp/go-hddtemp"
|
||
|
|
)
|
||
|
|
|
||
|
|
//go:embed sample.conf
|
||
|
|
var sampleConfig string
|
||
|
|
|
||
|
|
const defaultAddress = "127.0.0.1:7634"
|
||
|
|
|
||
|
|
type HDDTemp struct {
|
||
|
|
Address string `toml:"address"`
|
||
|
|
Devices []string `toml:"devices"`
|
||
|
|
|
||
|
|
fetcher fetcher
|
||
|
|
}
|
||
|
|
|
||
|
|
type fetcher interface {
|
||
|
|
Fetch(address string) ([]gohddtemp.Disk, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*HDDTemp) SampleConfig() string {
|
||
|
|
return sampleConfig
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *HDDTemp) Gather(acc telegraf.Accumulator) error {
|
||
|
|
if h.fetcher == nil {
|
||
|
|
h.fetcher = gohddtemp.New()
|
||
|
|
}
|
||
|
|
source, _, err := net.SplitHostPort(h.Address)
|
||
|
|
if err != nil {
|
||
|
|
source = h.Address
|
||
|
|
}
|
||
|
|
|
||
|
|
disks, err := h.fetcher.Fetch(h.Address)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, disk := range disks {
|
||
|
|
for _, chosenDevice := range h.Devices {
|
||
|
|
if chosenDevice == "*" || chosenDevice == disk.DeviceName {
|
||
|
|
tags := map[string]string{
|
||
|
|
"device": disk.DeviceName,
|
||
|
|
"model": disk.Model,
|
||
|
|
"unit": disk.Unit,
|
||
|
|
"status": disk.Status,
|
||
|
|
"source": source,
|
||
|
|
}
|
||
|
|
|
||
|
|
fields := map[string]interface{}{
|
||
|
|
"temperature": disk.Temperature,
|
||
|
|
}
|
||
|
|
|
||
|
|
acc.AddFields("hddtemp", fields, tags)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
inputs.Add("hddtemp", func() telegraf.Input {
|
||
|
|
return &HDDTemp{
|
||
|
|
Address: defaultAddress,
|
||
|
|
Devices: []string{"*"},
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|