influx_site.go

package unifipoller

import (
	"strings"
	"time"

	influx "github.com/influxdata/influxdb1-client/v2"
	"golift.io/unifi"
)

// SitePoints generates Unifi Sites' datapoints for InfluxDB.
// These points can be passed directly to influx.
func SitePoints(u *unifi.Site, now time.Time) ([]*influx.Point, error) {
	points := []*influx.Point{}
	for _, s := range u.Health {
		tags := map[string]string{
			"id":                   u.ID,
			"name":                 u.Name,
			"site_name":            u.SiteName,
			"desc":                 u.Desc,
			"status":               s.Status,
			"subsystem":            s.Subsystem,
			"wan_ip":               s.WanIP,
			"netmask":              s.Netmask,
			"gw_name":              s.GwName,
			"gw_mac":               s.GwMac,
			"gw_version":           s.GwVersion,
			"speedtest_status":     s.SpeedtestStatus,
			"lan_ip":               s.LanIP,
			"remote_user_enabled":  s.RemoteUserEnabled.Txt,
			"site_to_site_enabled": s.SiteToSiteEnabled.Txt,
			"nameservers":          strings.Join(s.Nameservers, ","),
			"gateways":             strings.Join(s.Gateways, ","),
			"num_new_alarms":       u.NumNewAlarms.Txt,
			"attr_hidden_id":       u.AttrHiddenID,
			"attr_no_delete":       u.AttrNoDelete.Txt,
		}
		fields := map[string]interface{}{
			"attr_hidden_id":           u.AttrHiddenID,
			"attr_no_delete":           u.AttrNoDelete.Val,
			"num_user":                 s.NumUser.Val,
			"num_guest":                s.NumGuest.Val,
			"num_iot":                  s.NumIot.Val,
			"tx_bytes-r":               s.TxBytesR.Val,
			"rx_bytes-r":               s.RxBytesR.Val,
			"status":                   s.Status,
			"num_ap":                   s.NumAp.Val,
			"num_adopted":              s.NumAdopted.Val,
			"num_disabled":             s.NumDisabled.Val,
			"num_disconnected":         s.NumDisconnected.Val,
			"num_pending":              s.NumPending.Val,
			"num_gw":                   s.NumGw.Val,
			"wan_ip":                   s.WanIP,
			"num_sta":                  s.NumSta.Val,
			"gw_cpu":                   s.GwSystemStats.CPU.Val,
			"gw_mem":                   s.GwSystemStats.Mem.Val,
			"gw_uptime":                s.GwSystemStats.Uptime.Val,
			"latency":                  s.Latency.Val,
			"uptime":                   s.Uptime.Val,
			"drops":                    s.Drops.Val,
			"xput_up":                  s.XputUp.Val,
			"xput_down":                s.XputDown.Val,
			"speedtest_ping":           s.SpeedtestPing.Val,
			"speedtest_lastrun":        s.SpeedtestLastrun.Val,
			"num_sw":                   s.NumSw.Val,
			"remote_user_num_active":   s.RemoteUserNumActive.Val,
			"remote_user_num_inactive": s.RemoteUserNumInactive.Val,
			"remote_user_rx_bytes":     s.RemoteUserRxBytes.Val,
			"remote_user_tx_bytes":     s.RemoteUserTxBytes.Val,
			"remote_user_rx_packets":   s.RemoteUserRxPackets.Val,
			"remote_user_tx_packets":   s.RemoteUserTxPackets.Val,
			"num_new_alarms":           u.NumNewAlarms.Val,
			"nameservers":              len(s.Nameservers),
			"gateways":                 len(s.Gateways),
		}
		pt, err := influx.NewPoint("subsystems", tags, fields, time.Now())
		if err != nil {
			return points, err
		}
		points = append(points, pt)
	}
	return points, nil
}