firehol/netdata

// SPDX-License-Identifier: GPL-3.0-or-later

package ping

import (
    "fmt"
    "strings"

    "github.com/netdata/netdata/go/plugins/plugin/go.d/agent/module"
)

const (
    prioHostRTT = module.Priority + iota
    prioHostStdDevRTT
    prioHostPingPacketLoss
    prioHostPingPackets
)

var hostChartsTmpl = module.Charts{
    hostRTTChartTmpl.Copy(),
    hostStdDevRTTChartTmpl.Copy(),
    hostPacketLossChartTmpl.Copy(),
    hostPacketsChartTmpl.Copy(),
}

var (
    hostRTTChartTmpl = module.Chart{
        ID:       "host_%s_rtt",
        Title:    "Ping round-trip time",
        Units:    "milliseconds",
        Fam:      "latency",
        Ctx:      "ping.host_rtt",
        Priority: prioHostRTT,
        Type:     module.Area,
        Dims: module.Dims{
            {ID: "host_%s_min_rtt", Name: "min", Div: 1e3},
            {ID: "host_%s_max_rtt", Name: "max", Div: 1e3},
            {ID: "host_%s_avg_rtt", Name: "avg", Div: 1e3},
        },
    }
    hostStdDevRTTChartTmpl = module.Chart{
        ID:       "host_%s_std_dev_rtt",
        Title:    "Ping round-trip time standard deviation",
        Units:    "milliseconds",
        Fam:      "latency",
        Ctx:      "ping.host_std_dev_rtt",
        Priority: prioHostStdDevRTT,
        Dims: module.Dims{
            {ID: "host_%s_std_dev_rtt", Name: "std_dev", Div: 1e3},
        },
    }
)

var hostPacketLossChartTmpl = module.Chart{
    ID:       "host_%s_packet_loss",
    Title:    "Ping packet loss",
    Units:    "percentage",
    Fam:      "packet loss",
    Ctx:      "ping.host_packet_loss",
    Priority: prioHostPingPacketLoss,
    Dims: module.Dims{
        {ID: "host_%s_packet_loss", Name: "loss", Div: 1000},
    },
}

var hostPacketsChartTmpl = module.Chart{
    ID:       "host_%s_packets",
    Title:    "Ping packets transferred",
    Units:    "packets",
    Fam:      "packets",
    Ctx:      "ping.host_packets",
    Priority: prioHostPingPackets,
    Dims: module.Dims{
        {ID: "host_%s_packets_recv", Name: "received"},
        {ID: "host_%s_packets_sent", Name: "sent"},
    },
}

func newHostCharts(host string) *module.Charts {
    charts := hostChartsTmpl.Copy()

    for _, chart := range *charts {
        chart.ID = fmt.Sprintf(chart.ID, strings.ReplaceAll(host, ".", "_"))
        chart.Labels = []module.Label{
            {Key: "host", Value: host},
        }
        for _, dim := range chart.Dims {
            dim.ID = fmt.Sprintf(dim.ID, host)
        }
    }

    return charts
}

func (p *Ping) addHostCharts(host string) {
    charts := newHostCharts(host)

    if err := p.Charts().Add(*charts...); err != nil {
        p.Warning(err)
    }
}