134 lines
3.6 KiB
Go
134 lines
3.6 KiB
Go
package influxunifi
|
|
|
|
import (
|
|
"golift.io/unifi"
|
|
)
|
|
|
|
// Combine concatenates N maps. This will delete things if not used with caution.
|
|
func Combine(in ...map[string]interface{}) map[string]interface{} {
|
|
out := make(map[string]interface{})
|
|
|
|
for i := range in {
|
|
for k := range in[i] {
|
|
out[k] = in[i][k]
|
|
}
|
|
}
|
|
|
|
return out
|
|
}
|
|
|
|
// batchSysStats is used by all device types.
|
|
func (u *InfluxUnifi) batchSysStats(s unifi.SysStats, ss unifi.SystemStats) map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"loadavg_1": s.Loadavg1.Val,
|
|
"loadavg_5": s.Loadavg5.Val,
|
|
"loadavg_15": s.Loadavg15.Val,
|
|
"mem_used": s.MemUsed.Val,
|
|
"mem_buffer": s.MemBuffer.Val,
|
|
"mem_total": s.MemTotal.Val,
|
|
"cpu": ss.CPU.Val,
|
|
"mem": ss.Mem.Val,
|
|
"system_uptime": ss.Uptime.Val,
|
|
}
|
|
}
|
|
|
|
// batchUDM generates Unifi Gateway datapoints for InfluxDB.
|
|
// These points can be passed directly to influx.
|
|
func (u *InfluxUnifi) batchUDM(r report, s *unifi.UDM) {
|
|
if !s.Adopted.Val || s.Locating.Val {
|
|
return
|
|
}
|
|
|
|
tags := map[string]string{
|
|
"source": s.SourceName,
|
|
"mac": s.Mac,
|
|
"site_name": s.SiteName,
|
|
"name": s.Name,
|
|
"version": s.Version,
|
|
"model": s.Model,
|
|
"serial": s.Serial,
|
|
"type": s.Type,
|
|
}
|
|
fields := Combine(
|
|
u.batchUSGstat(s.SpeedtestStatus, s.Stat.Gw, s.Uplink),
|
|
u.batchSysStats(s.SysStats, s.SystemStats),
|
|
map[string]interface{}{
|
|
"source": s.SourceName,
|
|
"ip": s.IP,
|
|
"bytes": s.Bytes.Val,
|
|
"last_seen": s.LastSeen.Val,
|
|
"license_state": s.LicenseState,
|
|
"guest-num_sta": s.GuestNumSta.Val,
|
|
"rx_bytes": s.RxBytes.Val,
|
|
"tx_bytes": s.TxBytes.Val,
|
|
"uptime": s.Uptime.Val,
|
|
"state": s.State.Val,
|
|
"user-num_sta": s.UserNumSta.Val,
|
|
"version": s.Version,
|
|
"num_desktop": s.NumDesktop.Val,
|
|
"num_handheld": s.NumHandheld.Val,
|
|
"num_mobile": s.NumMobile.Val,
|
|
},
|
|
)
|
|
|
|
r.send(&metric{Table: "usg", Tags: tags, Fields: fields})
|
|
u.batchNetTable(r, tags, s.NetworkTable)
|
|
u.batchUSGwans(r, tags, s.Wan1, s.Wan2)
|
|
|
|
tags = map[string]string{
|
|
"mac": s.Mac,
|
|
"site_name": s.SiteName,
|
|
"source": s.SourceName,
|
|
"name": s.Name,
|
|
"version": s.Version,
|
|
"model": s.Model,
|
|
"serial": s.Serial,
|
|
"type": s.Type,
|
|
}
|
|
fields = Combine(
|
|
u.batchUSWstat(s.Stat.Sw),
|
|
map[string]interface{}{
|
|
"guest-num_sta": s.GuestNumSta.Val,
|
|
"ip": s.IP,
|
|
"bytes": s.Bytes.Val,
|
|
"last_seen": s.LastSeen.Val,
|
|
"rx_bytes": s.RxBytes.Val,
|
|
"tx_bytes": s.TxBytes.Val,
|
|
"uptime": s.Uptime.Val,
|
|
"state": s.State.Val,
|
|
})
|
|
|
|
r.send(&metric{Table: "usw", Tags: tags, Fields: fields})
|
|
u.batchPortTable(r, tags, s.PortTable)
|
|
|
|
if s.Stat.Ap == nil {
|
|
return // we're done now. the following code process UDM (non-pro) UAP data.
|
|
}
|
|
|
|
tags = map[string]string{
|
|
"mac": s.Mac,
|
|
"site_name": s.SiteName,
|
|
"source": s.SourceName,
|
|
"name": s.Name,
|
|
"version": s.Version,
|
|
"model": s.Model,
|
|
"serial": s.Serial,
|
|
"type": s.Type,
|
|
}
|
|
fields = u.processUAPstats(s.Stat.Ap)
|
|
fields["ip"] = s.IP
|
|
fields["bytes"] = s.Bytes.Val
|
|
fields["last_seen"] = s.LastSeen.Val
|
|
fields["rx_bytes"] = s.RxBytes.Val
|
|
fields["tx_bytes"] = s.TxBytes.Val
|
|
fields["uptime"] = s.Uptime.Val
|
|
fields["state"] = s.State
|
|
fields["user-num_sta"] = int(s.UserNumSta.Val)
|
|
fields["guest-num_sta"] = int(s.GuestNumSta.Val)
|
|
fields["num_sta"] = s.NumSta.Val
|
|
|
|
r.send(&metric{Table: "uap", Tags: tags, Fields: fields})
|
|
u.processRadTable(r, tags, *s.RadioTable, *s.RadioTableStats)
|
|
u.processVAPTable(r, tags, *s.VapTable)
|
|
}
|