public_git
/
unpoller_unpoller
mirror of https://github.com/unpoller/unpoller.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

rename device to controller

This commit is contained in:
David Newhall II 2019-01-26 16:04:16 -08:00
parent 67862abfef
commit 071d494112
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cmd/unifi-poller/main.go
View File

@ -27,15 +27,15 @@ func main() {
log.Fatalf("Config Error '%v': %v", configFile, err) log.Fatalf("Config Error '%v': %v", configFile, err)
} }
// Create an authenticated session to the Unifi Controller. // Create an authenticated session to the Unifi Controller.
device, err := unifi.GetController(config.UnifiUser, config.UnifiPass, config.UnifiBase, config.VerifySSL) controller, err := unifi.GetController(config.UnifiUser, config.UnifiPass, config.UnifiBase, config.VerifySSL)
if err != nil { if err != nil {
log.Fatalln("Unifi Controller Error:", err) log.Fatalln("Unifi Controller Error:", err)
} else if !config.Quiet { } else if !config.Quiet {
log.Println("Authenticated to Unifi Controller @", config.UnifiBase, "as user", config.UnifiUser) log.Println("Authenticated to Unifi Controller @", config.UnifiBase, "as user", config.UnifiUser)
} }
device.ErrorLog = log.Printf controller.ErrorLog = log.Printf
if log.SetFlags(0); config.Debug { if log.SetFlags(0); config.Debug {
device.DebugLog = log.Printf controller.DebugLog = log.Printf
log.SetFlags(log.Lshortfile | log.Lmicroseconds | log.Ldate) log.SetFlags(log.Lshortfile | log.Lmicroseconds | log.Ldate)
log.Println("Debug Logging Enabled") log.Println("Debug Logging Enabled")
} }
@ -49,12 +49,12 @@ func main() {
} }
if config.Quiet { if config.Quiet {
// Doing it this way allows debug error logs (line numbers, etc) // Doing it this way allows debug error logs (line numbers, etc)
device.DebugLog = nil controller.DebugLog = nil
} else { } else {
log.Println("Logging Unifi Metrics to InfluXDB @", config.InfluxURL, "as user", config.InfluxUser) log.Println("Logging Unifi Metrics to InfluXDB @", config.InfluxURL, "as user", config.InfluxUser)
log.Println("Polling Unifi Controller, interval:", config.Interval.value) log.Println("Polling Unifi Controller, interval:", config.Interval.value)
} }
config.PollUnifiController(infdb, device) config.PollUnifiController(controller, infdb)
} }
func parseFlags() string { func parseFlags() string {
@ -98,13 +98,13 @@ func GetConfig(configFile string) (Config, error) {
} }
// PollUnifiController runs forever, polling and pushing. // PollUnifiController runs forever, polling and pushing.
func (c *Config) PollUnifiController(infdb influx.Client, uni *unifi.Unifi) { func (c *Config) PollUnifiController(controller *unifi.Unifi, infdb influx.Client) {
log.Println("[INFO] Everyting checks out! Beginning Poller Routine.") log.Println("[INFO] Everyting checks out! Beginning Poller Routine.")
ticker := time.NewTicker(c.Interval.value) ticker := time.NewTicker(c.Interval.value)
for range ticker.C { for range ticker.C {
if clients, err := uni.GetClients(); err != nil { if clients, err := controller.GetClients(); err != nil {
logErrors([]error{err}, "uni.GetClients()") logErrors([]error{err}, "uni.GetClients()")
} else if devices, err := uni.GetDevices(); err != nil { } else if devices, err := controller.GetDevices(); err != nil {
logErrors([]error{err}, "uni.GetDevices()") logErrors([]error{err}, "uni.GetDevices()")
} else if bp, err := influx.NewBatchPoints(influx.BatchPointsConfig{Database: c.InfluxDB}); err != nil { } else if bp, err := influx.NewBatchPoints(influx.BatchPointsConfig{Database: c.InfluxDB}); err != nil {
logErrors([]error{err}, "influx.NewBatchPoints") logErrors([]error{err}, "influx.NewBatchPoints")