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

Rename influx back to influx.

This commit is contained in:
David Newhall II 2019-01-24 00:14:41 -08:00
parent ae5f475b81
commit e9b4d86c8e
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
integrations/promunifi/cmd/unifi-poller/main.go
View File

@ -8,7 +8,7 @@ import (
"time" "time"
"github.com/golift/unifi" "github.com/golift/unifi"
"github.com/influxdata/influxdb/client/v2" influx "github.com/influxdata/influxdb/client/v2"
"github.com/naoina/toml" "github.com/naoina/toml"
flag "github.com/ogier/pflag" flag "github.com/ogier/pflag"
) )
@ -33,7 +33,7 @@ func main() {
} 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)
} }
infdb, err := client.NewHTTPClient(client.HTTPConfig{ infdb, err := influx.NewHTTPClient(influx.HTTPConfig{
Addr: config.InfluxURL, Addr: config.InfluxURL,
Username: config.InfluxUser, Username: config.InfluxUser,
Password: config.InfluxPass, Password: config.InfluxPass,
@ -91,17 +91,17 @@ func GetConfig(configFile string) (Config, error) {
} }
// PollUnifiController runs forever, polling and pushing. // PollUnifiController runs forever, polling and pushing.
func (c *Config) PollUnifiController(infdb client.Client, device *unifi.AuthedReq, quiet bool) { func (c *Config) PollUnifiController(infdb influx.Client, device *unifi.AuthedReq, quiet bool) {
ticker := time.NewTicker(c.Interval.value) ticker := time.NewTicker(c.Interval.value)
for range ticker.C { for range ticker.C {
var clients, devices []unifi.Asset var clients, devices []unifi.Asset
var bp client.BatchPoints var bp influx.BatchPoints
var err error var err error
if clients, err = device.GetUnifiClientAssets(); err != nil { if clients, err = device.GetUnifiClientAssets(); err != nil {
log.Println("ERROR unifi.GetUnifiClientsAssets():", err) log.Println("ERROR unifi.GetUnifiClientsAssets():", err)
} else if devices, err = device.GetUnifiDeviceAssets(); err != nil { } else if devices, err = device.GetUnifiDeviceAssets(); err != nil {
log.Println("ERROR unifi.GetUnifiDeviceAssets():", err) log.Println("ERROR unifi.GetUnifiDeviceAssets():", err)
} else if bp, err = client.NewBatchPoints(client.BatchPointsConfig{Database: c.InfluxDB}); err != nil { } else if bp, err = influx.NewBatchPoints(influx.BatchPointsConfig{Database: c.InfluxDB}); err != nil {
log.Println("ERROR influx.NewBatchPoints:", err) log.Println("ERROR influx.NewBatchPoints:", err)
} }
if err != nil { if err != nil {