firehol/netdata

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

package zfspool

import (
    "fmt"
    "strings"

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

const (
    prioZpoolHealthState = 2820 + iota
    prioVdevHealthState

    prioZpoolSpaceUtilization
    prioZpoolSpaceUsage

    prioZpoolFragmentation
)

var zpoolChartsTmpl = module.Charts{
    zpoolHealthStateChartTmpl.Copy(),

    zpoolSpaceUtilizationChartTmpl.Copy(),
    zpoolSpaceUsageChartTmpl.Copy(),

    zpoolFragmentationChartTmpl.Copy(),
}

var (
    zpoolHealthStateChartTmpl = module.Chart{
        ID:       "zfspool_%s_health_state",
        Title:    "Zpool health state",
        Units:    "state",
        Fam:      "health",
        Ctx:      "zfspool.pool_health_state",
        Type:     module.Line,
        Priority: prioZpoolHealthState,
        Dims: module.Dims{
            {ID: "zpool_%s_health_state_online", Name: "online"},
            {ID: "zpool_%s_health_state_degraded", Name: "degraded"},
            {ID: "zpool_%s_health_state_faulted", Name: "faulted"},
            {ID: "zpool_%s_health_state_offline", Name: "offline"},
            {ID: "zpool_%s_health_state_unavail", Name: "unavail"},
            {ID: "zpool_%s_health_state_removed", Name: "removed"},
            {ID: "zpool_%s_health_state_suspended", Name: "suspended"},
        },
    }

    zpoolSpaceUtilizationChartTmpl = module.Chart{
        ID:       "zfspool_%s_space_utilization",
        Title:    "Zpool space utilization",
        Units:    "percentage",
        Fam:      "space usage",
        Ctx:      "zfspool.pool_space_utilization",
        Type:     module.Area,
        Priority: prioZpoolSpaceUtilization,
        Dims: module.Dims{
            {ID: "zpool_%s_cap", Name: "utilization"},
        },
    }
    zpoolSpaceUsageChartTmpl = module.Chart{
        ID:       "zfspool_%s_space_usage",
        Title:    "Zpool space usage",
        Units:    "bytes",
        Fam:      "space usage",
        Ctx:      "zfspool.pool_space_usage",
        Type:     module.Stacked,
        Priority: prioZpoolSpaceUsage,
        Dims: module.Dims{
            {ID: "zpool_%s_free", Name: "free"},
            {ID: "zpool_%s_alloc", Name: "used"},
        },
    }

    zpoolFragmentationChartTmpl = module.Chart{
        ID:       "zfspool_%s_fragmentation",
        Title:    "Zpool fragmentation",
        Units:    "percentage",
        Fam:      "fragmentation",
        Ctx:      "zfspool.pool_fragmentation",
        Type:     module.Line,
        Priority: prioZpoolFragmentation,
        Dims: module.Dims{
            {ID: "zpool_%s_frag", Name: "fragmentation"},
        },
    }
)

var vdevChartsTmpl = module.Charts{
    vdevHealthStateChartTmpl.Copy(),
}

var (
    vdevHealthStateChartTmpl = module.Chart{
        ID:       "vdev_%s_health_state",
        Title:    "Zpool Vdev health state",
        Units:    "state",
        Fam:      "health",
        Ctx:      "zfspool.vdev_health_state",
        Type:     module.Line,
        Priority: prioVdevHealthState,
        Dims: module.Dims{
            {ID: "vdev_%s_health_state_online", Name: "online"},
            {ID: "vdev_%s_health_state_degraded", Name: "degraded"},
            {ID: "vdev_%s_health_state_faulted", Name: "faulted"},
            {ID: "vdev_%s_health_state_offline", Name: "offline"},
            {ID: "vdev_%s_health_state_unavail", Name: "unavail"},
            {ID: "vdev_%s_health_state_removed", Name: "removed"},
            {ID: "vdev_%s_health_state_suspended", Name: "suspended"},
        },
    }
)

func (z *ZFSPool) addZpoolCharts(name string) {
    charts := zpoolChartsTmpl.Copy()

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

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

func (z *ZFSPool) removeZpoolCharts(name string) {
    px := fmt.Sprintf("zfspool_%s_", name)
    z.removeCharts(px)
}

func (z *ZFSPool) addVdevCharts(pool, vdev string) {
    charts := vdevChartsTmpl.Copy()

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

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

func (z *ZFSPool) removeVdevCharts(vdev string) {
    px := fmt.Sprintf("vdev_%s_", cleanVdev(vdev))
    z.removeCharts(px)
}

func (z *ZFSPool) removeCharts(px string) {
    for _, chart := range *z.Charts() {
        if strings.HasPrefix(chart.ID, px) {
            chart.MarkRemove()
            chart.MarkNotCreated()
        }
    }
}

func cleanVdev(vdev string) string {
    r := strings.NewReplacer(".", "_")
    return r.Replace(vdev)
}