78 lines
3.0 KiB
Go
78 lines
3.0 KiB
Go
package unifipoller
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/golift/unifi"
|
|
influx "github.com/influxdata/influxdb1-client/v2"
|
|
"github.com/spf13/pflag"
|
|
)
|
|
|
|
// Version is injected by the Makefile
|
|
var Version = "development"
|
|
|
|
const (
|
|
// App defaults in case they're missing from the config.
|
|
defaultConfFile = "/etc/unifi-poller/up.conf"
|
|
defaultInterval = 30 * time.Second
|
|
defaultInfxDb = "unifi"
|
|
defaultInfxUser = "unifi"
|
|
defaultInfxPass = "unifi"
|
|
defaultInfxURL = "http://127.0.0.1:8086"
|
|
defaultUnifUser = "influx"
|
|
defaultUnifURL = "https://127.0.0.1:8443"
|
|
)
|
|
|
|
// Asset is used to give all devices and clients a common interface.
|
|
type Asset interface {
|
|
Points() ([]*influx.Point, error)
|
|
}
|
|
|
|
// UnifiPoller contains the application startup data, and auth info for UniFi & Influx.
|
|
type UnifiPoller struct {
|
|
ConfigFile string
|
|
DumpJSON string
|
|
ShowVer bool
|
|
Flag *pflag.FlagSet
|
|
errorCount int
|
|
influx.Client
|
|
*unifi.Unifi
|
|
*Config
|
|
}
|
|
|
|
// Metrics contains all the data from the controller and an influx endpoint to send it to.
|
|
type Metrics struct {
|
|
unifi.Sites
|
|
unifi.Clients
|
|
*unifi.Devices
|
|
influx.BatchPoints
|
|
}
|
|
|
|
// Config represents the data needed to poll a controller and report to influxdb.
|
|
// This is all of the data stored in the config file.
|
|
type Config struct {
|
|
MaxErrors int `json:"max_errors,_omitempty" toml:"max_errors,_omitempty" xml:"max_errors" yaml:"max_errors"`
|
|
Interval Duration `json:"interval,_omitempty" toml:"interval,_omitempty" xml:"interval" yaml:"interval"`
|
|
Debug bool `json:"debug" toml:"debug" xml:"debug" yaml:"debug"`
|
|
Quiet bool `json:"quiet,_omitempty" toml:"quiet,_omitempty" xml:"quiet" yaml:"quiet"`
|
|
VerifySSL bool `json:"verify_ssl" toml:"verify_ssl" xml:"verify_ssl" yaml:"verify_ssl"`
|
|
Mode string `json:"mode" toml:"mode" xml:"mode" yaml:"mode"`
|
|
InfluxURL string `json:"influx_url,_omitempty" toml:"influx_url,_omitempty" xml:"influx_url" yaml:"influx_url"`
|
|
InfluxUser string `json:"influx_user,_omitempty" toml:"influx_user,_omitempty" xml:"influx_user" yaml:"influx_user"`
|
|
InfluxPass string `json:"influx_pass,_omitempty" toml:"influx_pass,_omitempty" xml:"influx_pass" yaml:"influx_pass"`
|
|
InfluxDB string `json:"influx_db,_omitempty" toml:"influx_db,_omitempty" xml:"influx_db" yaml:"influx_db"`
|
|
UnifiUser string `json:"unifi_user,_omitempty" toml:"unifi_user,_omitempty" xml:"unifi_user" yaml:"unifi_user"`
|
|
UnifiPass string `json:"unifi_pass,_omitempty" toml:"unifi_pass,_omitempty" xml:"unifi_pass" yaml:"unifi_pass"`
|
|
UnifiBase string `json:"unifi_url,_omitempty" toml:"unifi_url,_omitempty" xml:"unifi_url" yaml:"unifi_url"`
|
|
Sites []string `json:"sites,_omitempty" toml:"sites,_omitempty" xml:"sites" yaml:"sites"`
|
|
}
|
|
|
|
// Duration is used to UnmarshalTOML into a time.Duration value.
|
|
type Duration struct{ time.Duration }
|
|
|
|
// UnmarshalText parses a duration type from a config file.
|
|
func (d *Duration) UnmarshalText(data []byte) (err error) {
|
|
d.Duration, err = time.ParseDuration(string(data))
|
|
return
|
|
}
|