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

More cleanup.

This commit is contained in:
David Newhall II 2019-06-12 23:40:11 -07:00
parent 4606f95384
commit 091ba2bc68
2 changed files with 21 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,7 +17,7 @@ func main() {
u := &UnifiPoller{} u := &UnifiPoller{}
if u.ParseFlags(os.Args[1:]); u.ShowVer { if u.ParseFlags(os.Args[1:]); u.ShowVer {
fmt.Printf("unifi-poller v%s\n", Version) fmt.Printf("unifi-poller v%s\n", Version)
os.Exit(0) // don't run anything else. return // don't run anything else.
} }
if err := u.GetConfig(); err != nil { if err := u.GetConfig(); err != nil {
u.Flag.Usage() u.Flag.Usage()
@ -66,7 +66,7 @@ func (u *UnifiPoller) GetConfig() error {
u.Quiet = true u.Quiet = true
} }
if !u.Config.Quiet { if !u.Config.Quiet {
log.Println("Loaded Configuration:", u.ConfigFile) log.Println("[INFO] Loaded Configuration:", u.ConfigFile)
} }
return nil return nil
} }
@ -75,29 +75,28 @@ func (u *UnifiPoller) GetConfig() error {
func (u *UnifiPoller) Run() error { func (u *UnifiPoller) Run() error {
c := u.Config c := u.Config
if u.DumpJSON != "" { if u.DumpJSON != "" {
if err := c.DumpJSON(u.DumpJSON); err != nil { return c.DumpJSON(u.DumpJSON)
log.Fatalln("[ERROR] dumping JSON:", err)
} }
return nil if log.SetFlags(0); c.Debug {
log.SetFlags(log.Lshortfile | log.Lmicroseconds | log.Ldate)
log.Println("[DEBUG] Debug Logging Enabled")
} }
log.Println("Unifi-Poller Starting Up! PID:", os.Getpid()) log.Printf("[INFO] Unifi-Poller v%v Starting Up! PID: %d", Version, os.Getpid())
// Create an authenticated session to the Unifi Controller. // Create an authenticated session to the Unifi Controller.
controller, err := unifi.NewUnifi(c.UnifiUser, c.UnifiPass, c.UnifiBase, c.VerifySSL) controller, err := unifi.NewUnifi(c.UnifiUser, c.UnifiPass, c.UnifiBase, c.VerifySSL)
if err != nil { if err != nil {
return errors.Wrap(err, "unifi controller") return errors.Wrap(err, "unifi controller")
} }
if c.Debug {
controller.DebugLog = log.Printf // Log debug messages.
}
controller.ErrorLog = log.Printf // Log all errors.
if !c.Quiet { if !c.Quiet {
log.Println("Authenticated to Unifi Controller @", c.UnifiBase, "as user", c.UnifiUser) log.Println("[INFO] Authenticated to Unifi Controller @", c.UnifiBase, "as user", c.UnifiUser)
} }
if err := c.CheckSites(controller); err != nil { if err := c.CheckSites(controller); err != nil {
return err return err
} }
controller.ErrorLog = log.Printf // Log all errors.
if log.SetFlags(0); c.Debug {
log.Println("Debug Logging Enabled")
log.SetFlags(log.Lshortfile | log.Lmicroseconds | log.Ldate)
controller.DebugLog = log.Printf // Log debug messages.
}
infdb, err := influx.NewHTTPClient(influx.HTTPConfig{ infdb, err := influx.NewHTTPClient(influx.HTTPConfig{
Addr: c.InfluxURL, Addr: c.InfluxURL,
Username: c.InfluxUser, Username: c.InfluxUser,
@ -110,8 +109,8 @@ func (u *UnifiPoller) Run() error {
// Doing it this way allows debug error logs (line numbers, etc) // Doing it this way allows debug error logs (line numbers, etc)
controller.DebugLog = nil controller.DebugLog = nil
} else { } else {
log.Println("Logging Unifi Metrics to InfluXDB @", c.InfluxURL, "as user", c.InfluxUser) log.Println("[INFO] Polling Unifi Controller Sites:", c.Sites)
log.Printf("Polling Unifi Controller (sites %v), interval: %v", c.Sites, c.Interval.value) log.Println("[INFO] Logging Measurements to InfluxDB at", c.InfluxURL, "as user", c.InfluxUser)
} }
c.PollUnifiController(controller, infdb) c.PollUnifiController(controller, infdb)
return nil return nil

10
cmd/unifi-poller/unifi.go
View File

@ -21,7 +21,7 @@ func (c *Config) CheckSites(controller *unifi.Unifi) error {
for _, site := range sites { for _, site := range sites {
msg = append(msg, site.Name+" ("+site.Desc+")") msg = append(msg, site.Name+" ("+site.Desc+")")
} }
log.Printf("Found %d site(s) on controller: %v", len(msg), strings.Join(msg, ", ")) log.Printf("[INFO] Found %d site(s) on controller: %v", len(msg), strings.Join(msg, ", "))
} }
if StringInSlice("all", c.Sites) { if StringInSlice("all", c.Sites) {
return nil return nil
@ -40,7 +40,7 @@ FIRST:
// PollUnifiController runs forever, polling and pushing. // PollUnifiController runs forever, polling and pushing.
func (c *Config) PollUnifiController(controller *unifi.Unifi, infdb influx.Client) { func (c *Config) PollUnifiController(controller *unifi.Unifi, infdb influx.Client) {
log.Println("[INFO] Everything checks out! Beginning Poller Routine.") log.Println("[INFO] Everything checks out! Poller started, interval:", c.Interval.value)
ticker := time.NewTicker(c.Interval.value) ticker := time.NewTicker(c.Interval.value)
for range ticker.C { for range ticker.C {
@ -73,8 +73,10 @@ func (c *Config) PollUnifiController(controller *unifi.Unifi, infdb influx.Clien
} }
// Talk about the data. // Talk about the data.
if !c.Quiet { if !c.Quiet {
log.Printf("[INFO] Logged Unifi States. Sites: %d Clients: %d, Wireless APs: %d, Gateways: %d, Switches: %d", log.Printf("[INFO] Unifi Measurements Recorded. Sites: %d Clients: %d, "+
len(sites), len(clients.UCLs), len(devices.UAPs), len(devices.USGs), len(devices.USWs)) "Wireless APs: %d, Gateways: %d, Switches: %d, Metrics: %d",
len(sites), len(clients.UCLs),
len(devices.UAPs), len(devices.USGs), len(devices.USWs), len(bp.Points()))
} }
} }
} }