mirror of
https://github.com/datarhei/core.git
synced 2025-10-31 03:16:21 +08:00
50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package monitor
|
|
|
|
import (
|
|
"github.com/datarhei/core/v16/monitor/metric"
|
|
"github.com/datarhei/core/v16/psutil"
|
|
)
|
|
|
|
type netCollector struct {
|
|
rxDescr *metric.Description
|
|
txDescr *metric.Description
|
|
}
|
|
|
|
func NewNetCollector() metric.Collector {
|
|
c := &netCollector{}
|
|
|
|
c.rxDescr = metric.NewDesc("net_rx", "Number of received bytes", []string{"interface"})
|
|
c.txDescr = metric.NewDesc("net_tx", "Number of transmitted bytes", []string{"interface"})
|
|
|
|
return c
|
|
}
|
|
|
|
func (c *netCollector) Prefix() string {
|
|
return "net"
|
|
}
|
|
|
|
func (c *netCollector) Describe() []*metric.Description {
|
|
return []*metric.Description{
|
|
c.rxDescr,
|
|
c.txDescr,
|
|
}
|
|
}
|
|
|
|
func (c *netCollector) Collect() metric.Metrics {
|
|
metrics := metric.NewMetrics()
|
|
|
|
devs, err := psutil.NetIOCounters(true)
|
|
if err != nil {
|
|
return metrics
|
|
}
|
|
|
|
for _, dev := range devs {
|
|
metrics.Add(metric.NewValue(c.rxDescr, float64(dev.BytesRecv), dev.Name))
|
|
metrics.Add(metric.NewValue(c.txDescr, float64(dev.BytesSent), dev.Name))
|
|
}
|
|
|
|
return metrics
|
|
}
|
|
|
|
func (c *netCollector) Stop() {}
|