Add ethtool input plugin (#5865)

plugins/inputs/ethtool/ethtool_linux.go

@@ -0,0 +1,136 @@
+// +build linux
+
+package ethtool
+
+import (
+	"net"
+	"sync"
+
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/filter"
+	"github.com/influxdata/telegraf/plugins/inputs"
+	"github.com/pkg/errors"
+	"github.com/safchain/ethtool"
+)
+
+type CommandEthtool struct {
+	ethtool *ethtool.Ethtool
+}
+
+func (e *Ethtool) Gather(acc telegraf.Accumulator) error {
+
+	// Get the list of interfaces
+	interfaces, err := e.command.Interfaces()
+	if err != nil {
+		acc.AddError(err)
+		return nil
+	}
+
+	interfaceFilter, err := filter.NewIncludeExcludeFilter(e.InterfaceInclude, e.InterfaceExclude)
+	if err != nil {
+		return err
+	}
+
+	// parallelize the ethtool call in event of many interfaces
+	var wg sync.WaitGroup
+
+	for _, iface := range interfaces {
+
+		// Check this isn't a loop back and that its matched by the filter
+		if (iface.Flags&net.FlagLoopback == 0) && interfaceFilter.Match(iface.Name) {
+			wg.Add(1)
+
+			go func(i net.Interface) {
+				e.gatherEthtoolStats(i, acc)
+				wg.Done()
+			}(iface)
+		}
+	}
+
+	// Waiting for all the interfaces
+	wg.Wait()
+	return nil
+}
+
+// Initialise the Command Tool
+func (e *Ethtool) Init() error {
+	return e.command.Init()
+}
+
+// Gather the stats for the interface.
+func (e *Ethtool) gatherEthtoolStats(iface net.Interface, acc telegraf.Accumulator) {
+
+	tags := make(map[string]string)
+	tags[tagInterface] = iface.Name
+
+	driverName, err := e.command.DriverName(iface.Name)
+	if err != nil {
+		driverErr := errors.Wrapf(err, "%s driver", iface.Name)
+		acc.AddError(driverErr)
+		return
+	}
+
+	tags[tagDriverName] = driverName
+
+	fields := make(map[string]interface{})
+	stats, err := e.command.Stats(iface.Name)
+	if err != nil {
+		statsErr := errors.Wrapf(err, "%s stats", iface.Name)
+		acc.AddError(statsErr)
+		return
+	}
+
+	for k, v := range stats {
+		fields[k] = v
+	}
+
+	acc.AddFields(pluginName, fields, tags)
+}
+
+func NewCommandEthtool() *CommandEthtool {
+	return &CommandEthtool{}
+}
+
+func (c *CommandEthtool) Init() error {
+
+	if c.ethtool != nil {
+		return nil
+	}
+
+	e, err := ethtool.NewEthtool()
+	if err == nil {
+		c.ethtool = e
+	}
+
+	return err
+}
+
+func (c *CommandEthtool) DriverName(intf string) (string, error) {
+	return c.ethtool.DriverName(intf)
+}
+
+func (c *CommandEthtool) Stats(intf string) (map[string]uint64, error) {
+	return c.ethtool.Stats(intf)
+}
+
+func (c *CommandEthtool) Interfaces() ([]net.Interface, error) {
+
+	// Get the list of interfaces
+	interfaces, err := net.Interfaces()
+	if err != nil {
+		return nil, err
+	}
+
+	return interfaces, nil
+}
+
+func init() {
+
+	inputs.Add(pluginName, func() telegraf.Input {
+		return &Ethtool{
+			InterfaceInclude: []string{},
+			InterfaceExclude: []string{},
+			command:          NewCommandEthtool(),
+		}
+	})
+}