232 lines
9.8 KiB
Go
232 lines
9.8 KiB
Go
package influxunifi
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
influx "github.com/influxdata/influxdb1-client/v2"
|
|
"golift.io/unifi"
|
|
)
|
|
|
|
// USGPoints generates Unifi Gateway datapoints for InfluxDB.
|
|
// These points can be passed directly to influx.
|
|
func USGPoints(u *unifi.USG, now time.Time) ([]*influx.Point, error) {
|
|
if u.Stat.Gw == nil {
|
|
u.Stat.Gw = &unifi.Gw{}
|
|
}
|
|
tags := map[string]string{
|
|
"id": u.ID,
|
|
"mac": u.Mac,
|
|
"device_type": u.Stat.O,
|
|
"device_oid": u.Stat.Oid,
|
|
"site_id": u.SiteID,
|
|
"site_name": u.SiteName,
|
|
"adopted": u.Adopted.Txt,
|
|
"name": u.Name,
|
|
"cfgversion": u.Cfgversion,
|
|
"config_network_ip": u.ConfigNetwork.IP,
|
|
"config_network_type": u.ConfigNetwork.Type,
|
|
"connect_request_ip": u.ConnectRequestIP,
|
|
"connect_request_port": u.ConnectRequestPort,
|
|
"device_id": u.DeviceID,
|
|
"guest_token": u.GuestToken,
|
|
"inform_ip": u.InformIP,
|
|
"known_cfgversion": u.KnownCfgversion,
|
|
"led_override": u.LedOverride,
|
|
"locating": u.Locating.Txt,
|
|
"model": u.Model,
|
|
"outdoor_mode_override": u.OutdoorModeOverride,
|
|
"serial": u.Serial,
|
|
"type": u.Type,
|
|
"usg_caps": u.UsgCaps.Txt,
|
|
"speedtest-status-saved": u.SpeedtestStatusSaved.Txt,
|
|
"wan1_up": u.Wan1.Up.Txt,
|
|
"wan2_up": u.Wan2.Up.Txt,
|
|
}
|
|
fields := map[string]interface{}{
|
|
"ip": u.IP,
|
|
"bytes": u.Bytes.Val,
|
|
"last_seen": u.LastSeen.Val,
|
|
"license_state": u.LicenseState,
|
|
"fw_caps": u.FwCaps.Val,
|
|
"guest-num_sta": u.GuestNumSta.Val,
|
|
"rx_bytes": u.RxBytes.Val,
|
|
"tx_bytes": u.TxBytes.Val,
|
|
"uptime": u.Uptime.Val,
|
|
"roll_upgrade": u.Rollupgrade.Val,
|
|
"state": u.State.Val,
|
|
"upgradable": u.Upgradable.Val,
|
|
"user-num_sta": u.UserNumSta.Val,
|
|
"version": u.Version,
|
|
"num_desktop": u.NumDesktop.Val,
|
|
"num_handheld": u.NumHandheld.Val,
|
|
"num_mobile": u.NumMobile.Val,
|
|
"speedtest-status_latency": u.SpeedtestStatus.Latency.Val,
|
|
"speedtest-status_rundate": u.SpeedtestStatus.Rundate.Val,
|
|
"speedtest-status_runtime": u.SpeedtestStatus.Runtime.Val,
|
|
"speedtest-status_download": u.SpeedtestStatus.StatusDownload.Val,
|
|
"speedtest-status_ping": u.SpeedtestStatus.StatusPing.Val,
|
|
"speedtest-status_summary": u.SpeedtestStatus.StatusSummary.Val,
|
|
"speedtest-status_upload": u.SpeedtestStatus.StatusUpload.Val,
|
|
"speedtest-status_xput_download": u.SpeedtestStatus.XputDownload.Val,
|
|
"speedtest-status_xput_upload": u.SpeedtestStatus.XputUpload.Val,
|
|
"config_network_wan_type": u.ConfigNetwork.Type,
|
|
"wan1_bytes-r": u.Wan1.BytesR.Val,
|
|
"wan1_enable": u.Wan1.Enable.Val,
|
|
"wan1_full_duplex": u.Wan1.FullDuplex.Val,
|
|
"wan1_gateway": u.Wan1.Gateway,
|
|
"wan1_ifname": u.Wan1.Ifname,
|
|
"wan1_ip": u.Wan1.IP,
|
|
"wan1_mac": u.Wan1.Mac,
|
|
"wan1_max_speed": u.Wan1.MaxSpeed.Val,
|
|
"wan1_name": u.Wan1.Name,
|
|
"wan1_netmask": u.Wan1.Netmask,
|
|
"wan1_rx_bytes": u.Wan1.RxBytes.Val,
|
|
"wan1_rx_bytes-r": u.Wan1.RxBytesR.Val,
|
|
"wan1_rx_dropped": u.Wan1.RxDropped.Val,
|
|
"wan1_rx_errors": u.Wan1.RxErrors.Val,
|
|
"wan1_rx_multicast": u.Wan1.RxMulticast.Val,
|
|
"wan1_rx_packets": u.Wan1.RxPackets.Val,
|
|
"wan1_type": u.Wan1.Type,
|
|
"wan1_speed": u.Wan1.Speed.Val,
|
|
"wan1_up": u.Wan1.Up.Val,
|
|
"wan1_tx_bytes": u.Wan1.TxBytes.Val,
|
|
"wan1_tx_bytes-r": u.Wan1.TxBytesR.Val,
|
|
"wan1_tx_dropped": u.Wan1.TxDropped.Val,
|
|
"wan1_tx_errors": u.Wan1.TxErrors.Val,
|
|
"wan1_tx_packets": u.Wan1.TxPackets.Val,
|
|
"wan2_bytes-r": u.Wan2.BytesR.Val,
|
|
"wan2_enable": u.Wan2.Enable.Val,
|
|
"wan2_full_duplex": u.Wan2.FullDuplex.Val,
|
|
"wan2_gateway": u.Wan2.Gateway,
|
|
"wan2_ifname": u.Wan2.Ifname,
|
|
"wan2_ip": u.Wan2.IP,
|
|
"wan2_mac": u.Wan2.Mac,
|
|
"wan2_max_speed": u.Wan2.MaxSpeed.Val,
|
|
"wan2_name": u.Wan2.Name,
|
|
"wan2_netmask": u.Wan2.Netmask,
|
|
"wan2_rx_bytes": u.Wan2.RxBytes.Val,
|
|
"wan2_rx_bytes-r": u.Wan2.RxBytesR.Val,
|
|
"wan2_rx_dropped": u.Wan2.RxDropped.Val,
|
|
"wan2_rx_errors": u.Wan2.RxErrors.Val,
|
|
"wan2_rx_multicast": u.Wan2.RxMulticast.Val,
|
|
"wan2_rx_packets": u.Wan2.RxPackets.Val,
|
|
"wan2_type": u.Wan2.Type,
|
|
"wan2_speed": u.Wan2.Speed.Val,
|
|
"wan2_up": u.Wan2.Up.Val,
|
|
"wan2_tx_bytes": u.Wan2.TxBytes.Val,
|
|
"wan2_tx_bytes-r": u.Wan2.TxBytesR.Val,
|
|
"wan2_tx_dropped": u.Wan2.TxDropped.Val,
|
|
"wan2_tx_errors": u.Wan2.TxErrors.Val,
|
|
"wan2_tx_packets": u.Wan2.TxPackets.Val,
|
|
"loadavg_1": u.SysStats.Loadavg1.Val,
|
|
"loadavg_5": u.SysStats.Loadavg5.Val,
|
|
"loadavg_15": u.SysStats.Loadavg15.Val,
|
|
"mem_used": u.SysStats.MemUsed.Val,
|
|
"mem_buffer": u.SysStats.MemBuffer.Val,
|
|
"mem_total": u.SysStats.MemTotal.Val,
|
|
"cpu": u.SystemStats.CPU.Val,
|
|
"mem": u.SystemStats.Mem.Val,
|
|
"system_uptime": u.SystemStats.Uptime.Val,
|
|
"stat_duration": u.Stat.Duration.Val,
|
|
"stat_datetime": u.Stat.Datetime,
|
|
"gw": u.Stat.Gw,
|
|
"lan-rx_bytes": u.Stat.LanRxBytes.Val,
|
|
"lan-rx_packets": u.Stat.LanRxPackets.Val,
|
|
"lan-tx_bytes": u.Stat.LanTxBytes.Val,
|
|
"lan-tx_packets": u.Stat.LanTxPackets.Val,
|
|
"wan-rx_bytes": u.Stat.WanRxBytes.Val,
|
|
"wan-rx_dropped": u.Stat.WanRxDropped.Val,
|
|
"wan-rx_packets": u.Stat.WanRxPackets.Val,
|
|
"wan-tx_bytes": u.Stat.WanTxBytes.Val,
|
|
"wan-tx_packets": u.Stat.WanTxPackets.Val,
|
|
"uplink_name": u.Uplink.Name,
|
|
"uplink_latency": u.Uplink.Latency.Val,
|
|
"uplink_speed": u.Uplink.Speed.Val,
|
|
"uplink_num_ports": u.Uplink.NumPort.Val,
|
|
"uplink_max_speed": u.Uplink.MaxSpeed.Val,
|
|
}
|
|
pt, err := influx.NewPoint("usg", tags, fields, now)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
points := []*influx.Point{pt}
|
|
for _, p := range u.NetworkTable {
|
|
tags := map[string]string{
|
|
"device_name": u.Name,
|
|
"device_id": u.ID,
|
|
"device_mac": u.Mac,
|
|
"site_name": u.SiteName,
|
|
"up": p.Up.Txt,
|
|
"dhcpd_dns_enabled": p.DhcpdDNSEnabled.Txt,
|
|
"dhcpd_enabled": p.DhcpdEnabled.Txt,
|
|
"dhcpd_time_offset_enabled": p.DhcpdTimeOffsetEnabled.Txt,
|
|
"dhcp_relay_enabledy": p.DhcpRelayEnabled.Txt,
|
|
"dhcpd_gateway_enabled": p.DhcpdGatewayEnabled.Txt,
|
|
"enabled": p.Enabled.Txt,
|
|
"vlan_enabled": p.VlanEnabled.Txt,
|
|
"attr_no_delete": p.AttrNoDelete.Txt,
|
|
"is_guest": p.IsGuest.Txt,
|
|
"is_nat": p.IsNat.Txt,
|
|
"networkgroup": p.Networkgroup,
|
|
"site_id": p.SiteID,
|
|
"ip": p.IP,
|
|
"ip_subnet": p.IPSubnet,
|
|
"mac": p.Mac,
|
|
"name": p.Name,
|
|
"domain_name": p.DomainName,
|
|
"dhcpd_start": p.DhcpdStart,
|
|
"dhcpd_stop": p.DhcpdStop,
|
|
"ipv6_interface_type": p.Ipv6InterfaceType,
|
|
"attr_hidden_id": p.AttrHiddenID,
|
|
"purpose": p.Purpose,
|
|
}
|
|
fields := map[string]interface{}{
|
|
"num_sta": p.NumSta.Val,
|
|
"rx_bytes": p.RxBytes.Val,
|
|
"rx_packets": p.RxPackets.Val,
|
|
"tx_bytes": p.TxBytes.Val,
|
|
"tx_packets": p.TxPackets.Val,
|
|
}
|
|
pt, err = influx.NewPoint("usg_networks", tags, fields, now)
|
|
if err != nil {
|
|
return points, err
|
|
}
|
|
points = append(points, pt)
|
|
}
|
|
for _, p := range u.PortTable {
|
|
tags := map[string]string{
|
|
"device_name": u.Name,
|
|
"device_id": u.ID,
|
|
"device_mac": u.Mac,
|
|
"site_name": u.SiteName,
|
|
"name": p.Name,
|
|
"ifname": p.Ifname,
|
|
"ip": p.IP,
|
|
"mac": p.Mac,
|
|
"up": p.Up.Txt,
|
|
"speed": p.Speed.Txt,
|
|
"full_duplex": p.FullDuplex.Txt,
|
|
"enable": p.Enable.Txt,
|
|
}
|
|
fields := map[string]interface{}{
|
|
"rx_bytes": p.RxBytes.Val,
|
|
"rx_dropped": p.RxDropped.Val,
|
|
"rx_errors": p.RxErrors.Val,
|
|
"rx_packets": p.RxBytes.Val,
|
|
"tx_bytes": p.TxBytes.Val,
|
|
"tx_dropped": p.TxDropped.Val,
|
|
"tx_errors": p.TxErrors.Val,
|
|
"tx_packets": p.TxPackets.Val,
|
|
"rx_multicast": p.RxMulticast.Val,
|
|
"dns_servers": strings.Join(p.DNS, ","),
|
|
}
|
|
pt, err = influx.NewPoint("usg_ports", tags, fields, now)
|
|
if err != nil {
|
|
return points, err
|
|
}
|
|
points = append(points, pt)
|
|
}
|
|
return points, nil
|
|
}
|