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

inhert config pointer

This commit is contained in:
davidnewhall2 2019-12-29 15:06:45 -08:00
parent 78f5a150ee
commit e3fd22fee1
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
integrations/influxunifi/influxdb.go
View File

@ -34,7 +34,7 @@ type Config struct {
// InfluxDB allows the data to be nested in the config file. // InfluxDB allows the data to be nested in the config file.
type InfluxDB struct { type InfluxDB struct {
Config Config `json:"influxdb" toml:"influxdb" xml:"influxdb" yaml:"influxdb"` *Config `json:"influxdb" toml:"influxdb" xml:"influxdb" yaml:"influxdb"`
} }
// InfluxUnifi is returned by New() after you provide a Config. // InfluxUnifi is returned by New() after you provide a Config.
@ -64,7 +64,7 @@ func init() {
// PollController runs forever, polling UniFi and pushing to InfluxDB // PollController runs forever, polling UniFi and pushing to InfluxDB
// This is started by Run() or RunBoth() after everything checks out. // This is started by Run() or RunBoth() after everything checks out.
func (u *InfluxUnifi) PollController() { func (u *InfluxUnifi) PollController() {
interval := u.Config.Interval.Round(time.Second) interval := u.Interval.Round(time.Second)
ticker := time.NewTicker(interval) ticker := time.NewTicker(interval)
log.Printf("[INFO] Everything checks out! Poller started, InfluxDB interval: %v", interval) log.Printf("[INFO] Everything checks out! Poller started, InfluxDB interval: %v", interval)
@ -93,7 +93,7 @@ func (u *InfluxUnifi) PollController() {
func (u *InfluxUnifi) Run(c poller.Collect) error { func (u *InfluxUnifi) Run(c poller.Collect) error {
var err error var err error
if u.Config.Disable { if u.Config == nil || u.Disable {
return nil return nil
} }
@ -101,10 +101,10 @@ func (u *InfluxUnifi) Run(c poller.Collect) error {
u.setConfigDefaults() u.setConfigDefaults()
u.influx, err = influx.NewHTTPClient(influx.HTTPConfig{ u.influx, err = influx.NewHTTPClient(influx.HTTPConfig{
Addr: u.Config.URL, Addr: u.URL,
Username: u.Config.User, Username: u.User,
Password: u.Config.Pass, Password: u.Pass,
TLSConfig: &tls.Config{InsecureSkipVerify: !u.Config.VerifySSL}, TLSConfig: &tls.Config{InsecureSkipVerify: !u.VerifySSL},
}) })
if err != nil { if err != nil {
return err return err
@ -116,29 +116,29 @@ func (u *InfluxUnifi) Run(c poller.Collect) error {
} }
func (u *InfluxUnifi) setConfigDefaults() { func (u *InfluxUnifi) setConfigDefaults() {
if u.Config.URL == "" { if u.URL == "" {
u.Config.URL = defaultInfluxURL u.URL = defaultInfluxURL
} }
if u.Config.User == "" { if u.User == "" {
u.Config.User = defaultInfluxUser u.User = defaultInfluxUser
} }
if u.Config.Pass == "" { if u.Pass == "" {
u.Config.Pass = defaultInfluxUser u.Pass = defaultInfluxUser
} }
if u.Config.DB == "" { if u.DB == "" {
u.Config.DB = defaultInfluxDB u.DB = defaultInfluxDB
} }
if u.Config.Interval.Duration == 0 { if u.Interval.Duration == 0 {
u.Config.Interval = cnfg.Duration{Duration: defaultInterval} u.Interval = cnfg.Duration{Duration: defaultInterval}
} else if u.Config.Interval.Duration < minimumInterval { } else if u.Interval.Duration < minimumInterval {
u.Config.Interval = cnfg.Duration{Duration: minimumInterval} u.Interval = cnfg.Duration{Duration: minimumInterval}
} }
u.Config.Interval = cnfg.Duration{Duration: u.Config.Interval.Duration.Round(time.Second)} u.Interval = cnfg.Duration{Duration: u.Interval.Duration.Round(time.Second)}
} }
// ReportMetrics batches all device and client data into influxdb data points. // ReportMetrics batches all device and client data into influxdb data points.
@ -151,7 +151,7 @@ func (u *InfluxUnifi) ReportMetrics(m *poller.Metrics) (*Report, error) {
var err error var err error
// Make a new Influx Points Batcher. // Make a new Influx Points Batcher.
r.bp, err = influx.NewBatchPoints(influx.BatchPointsConfig{Database: u.Config.DB}) r.bp, err = influx.NewBatchPoints(influx.BatchPointsConfig{Database: u.DB})
if err != nil { if err != nil {
return nil, fmt.Errorf("influx.NewBatchPoints: %v", err) return nil, fmt.Errorf("influx.NewBatchPoints: %v", err)