From f60f10dd3dd19df8e5508aef51c9f2364583ad8b Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 14:00:23 -0700
Subject: [PATCH 01/12] Add a run-once lambda mode.

---
 .../promunifi/pkg/unifi-poller/config.go      |  3 +
 .../promunifi/pkg/unifi-poller/helpers.go     | 30 ++----
 .../promunifi/pkg/unifi-poller/poller.go      |  7 ++
 .../promunifi/pkg/unifi-poller/unifi.go       | 91 ++++++++++---------
 4 files changed, 68 insertions(+), 63 deletions(-)

diff --git a/integrations/promunifi/pkg/unifi-poller/config.go b/integrations/promunifi/pkg/unifi-poller/config.go
index 31beff55..8f2596d6 100644
--- a/integrations/promunifi/pkg/unifi-poller/config.go
+++ b/integrations/promunifi/pkg/unifi-poller/config.go
@@ -33,6 +33,7 @@ type UnifiPoller struct {
 	ConfigFile string
 	DumpJSON   string
 	ShowVer    bool
+	Lambda     bool
 	Flag       *pflag.FlagSet
 	errorCount int
 	influx.Client
@@ -49,12 +50,14 @@ type Metrics struct {
 }
 
 // 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"`
+	Lambda     bool     `json:"lambda" toml:"lambda" xml:"lambda" yaml:"lambda"`
 	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"`
diff --git a/integrations/promunifi/pkg/unifi-poller/helpers.go b/integrations/promunifi/pkg/unifi-poller/helpers.go
index 9f8384a7..a138a052 100644
--- a/integrations/promunifi/pkg/unifi-poller/helpers.go
+++ b/integrations/promunifi/pkg/unifi-poller/helpers.go
@@ -6,30 +6,18 @@ import (
 	"strings"
 )
 
-// hasErr checks a list of errors for a non-nil.
-func hasErr(errs []error) bool {
-	for _, err := range errs {
-		if err != nil {
-			return true
-		}
-	}
-	return false
-}
-
-// LogErrors writes a slice of errors, with a prefix, to log-out.
-// It also increments the error counter.
-func (u *UnifiPoller) LogErrors(errs []error, prefix string) {
-	for _, err := range errs {
-		if err != nil {
-			u.errorCount++
-			_ = log.Output(2, fmt.Sprintf("[ERROR] (%v/%v) %v: %v", u.errorCount, u.MaxErrors, prefix, err))
-		}
+// LogError logs an error and increments the error counter.
+// Should be used in the poller loop.
+func (u *UnifiPoller) LogError(err error, prefix string) {
+	if err != nil {
+		u.errorCount++
+		_ = log.Output(2, fmt.Sprintf("[ERROR] (%v/%v) %v: %v", u.errorCount, u.MaxErrors, prefix, err))
 	}
 }
 
 // StringInSlice returns true if a string is in a slice.
-func StringInSlice(str string, slc []string) bool {
-	for _, s := range slc {
+func StringInSlice(str string, slice []string) bool {
+	for _, s := range slice {
 		if strings.EqualFold(s, str) {
 			return true
 		}
@@ -51,7 +39,7 @@ func (u *UnifiPoller) LogDebugf(m string, v ...interface{}) {
 	}
 }
 
-// LogErrorf prints an error log entry.
+// LogErrorf prints an error log entry. This is used for external library logging.
 func (u *UnifiPoller) LogErrorf(m string, v ...interface{}) {
 	_ = log.Output(2, fmt.Sprintf("[ERROR] "+m, v...))
 }
diff --git a/integrations/promunifi/pkg/unifi-poller/poller.go b/integrations/promunifi/pkg/unifi-poller/poller.go
index 81b43bf8..2636ced0 100644
--- a/integrations/promunifi/pkg/unifi-poller/poller.go
+++ b/integrations/promunifi/pkg/unifi-poller/poller.go
@@ -78,6 +78,13 @@ func (u *UnifiPoller) Run() (err error) {
 	if err = u.GetInfluxDB(); err != nil {
 		return err
 	}
+	if u.Lambda {
+		metrics, err := u.CollectMetrics()
+		if err != nil {
+			return err
+		}
+		return u.ReportMetrics(metrics)
+	}
 	return u.PollController()
 }
 
diff --git a/integrations/promunifi/pkg/unifi-poller/unifi.go b/integrations/promunifi/pkg/unifi-poller/unifi.go
index 4fcd2bbe..3da6d10a 100644
--- a/integrations/promunifi/pkg/unifi-poller/unifi.go
+++ b/integrations/promunifi/pkg/unifi-poller/unifi.go
@@ -42,46 +42,11 @@ FIRST:
 func (u *UnifiPoller) PollController() error {
 	log.Println("[INFO] Everything checks out! Poller started, interval:", u.Interval.Round(time.Second))
 	ticker := time.NewTicker(u.Interval.Round(time.Second))
-	var err error
 	for range ticker.C {
-		m := &Metrics{}
-		// Get the sites we care about.
-		if m.Sites, err = u.GetFilteredSites(); err != nil {
-			u.LogErrors([]error{err}, "unifi.GetSites()")
+		metrics, err := u.CollectMetrics()
+		if err == nil {
+			u.LogError(u.ReportMetrics(metrics), "reporting metrics")
 		}
-		// Get all the points.
-		if m.Clients, err = u.GetClients(m.Sites); err != nil {
-			u.LogErrors([]error{err}, "unifi.GetClients()")
-		}
-		if m.Devices, err = u.GetDevices(m.Sites); err != nil {
-			u.LogErrors([]error{err}, "unifi.GetDevices()")
-		}
-
-		// Make a new Points Batcher.
-		m.BatchPoints, err = influx.NewBatchPoints(influx.BatchPointsConfig{Database: u.InfluxDB})
-		if err != nil {
-			u.LogErrors([]error{err}, "influx.NewBatchPoints")
-			continue
-		}
-		// Batch (and send) all the points.
-		if errs := m.SendPoints(); errs != nil && hasErr(errs) {
-			u.LogErrors(errs, "asset.Points()")
-		}
-		if err := u.Write(m.BatchPoints); err != nil {
-			u.LogErrors([]error{err}, "infdb.Write(bp)")
-		}
-
-		// Talk about the data.
-		var fieldcount, pointcount int
-		for _, p := range m.Points() {
-			pointcount++
-			i, _ := p.Fields()
-			fieldcount += len(i)
-		}
-		u.Logf("UniFi Measurements Recorded. Sites: %d, Clients: %d, "+
-			"Wireless APs: %d, Gateways: %d, Switches: %d, Points: %d, Fields: %d",
-			len(m.Sites), len(m.Clients), len(m.UAPs), len(m.USGs), len(m.USWs), pointcount, fieldcount)
-
 		if u.MaxErrors >= 0 && u.errorCount > u.MaxErrors {
 			return errors.Errorf("reached maximum error count, stopping poller (%d > %d)", u.errorCount, u.MaxErrors)
 		}
@@ -89,10 +54,52 @@ func (u *UnifiPoller) PollController() error {
 	return nil
 }
 
-// SendPoints combines all device and client data into influxdb data points.
+// CollectMetrics grabs all the measurements from a UniFi controller and returns them.
+// This also creates an InfluxDB writer, and retuns error if that fails.
+func (u *UnifiPoller) CollectMetrics() (*Metrics, error) {
+	m := &Metrics{}
+	var err error
+	// Get the sites we care about.
+	m.Sites, err = u.GetFilteredSites()
+	u.LogError(err, "unifi.GetSites()")
+	// Get all the points.
+	m.Clients, err = u.GetClients(m.Sites)
+	u.LogError(err, "unifi.GetClients()")
+	m.Devices, err = u.GetDevices(m.Sites)
+	u.LogError(err, "unifi.GetDevices()")
+	// Make a new Influx Points Batcher.
+	m.BatchPoints, err = influx.NewBatchPoints(influx.BatchPointsConfig{Database: u.InfluxDB})
+	u.LogError(err, "influx.NewBatchPoints")
+	return m, err
+}
+
+// ReportMetrics batches all the metrics and writes them to InfluxDB.
+// Returns an error if the write to influx fails.
+func (u *UnifiPoller) ReportMetrics(metrics *Metrics) error {
+	// Batch (and send) all the points.
+	for _, err := range metrics.ProcessPoints() {
+		u.LogError(err, "asset.Points()")
+	}
+	err := u.Write(metrics.BatchPoints)
+	if err != nil {
+		return errors.Wrap(err, "infdb.Write(bp)")
+	}
+	var fields, points int
+	for _, p := range metrics.Points() {
+		points++
+		i, _ := p.Fields()
+		fields += len(i)
+	}
+	u.Logf("UniFi Measurements Recorded. Sites: %d, Clients: %d, "+
+		"Wireless APs: %d, Gateways: %d, Switches: %d, Points: %d, Fields: %d",
+		len(metrics.Sites), len(metrics.Clients), len(metrics.UAPs),
+		len(metrics.USGs), len(metrics.USWs), points, fields)
+	return nil
+}
+
+// ProcessPoints batches all device and client data into influxdb data points.
 // Call this after you've collected all the data you care about.
-// This sends all the batched points to InfluxDB.
-func (m *Metrics) SendPoints() (errs []error) {
+func (m *Metrics) ProcessPoints() (errs []error) {
 	for _, asset := range m.Sites {
 		errs = append(errs, m.processPoints(asset))
 	}
@@ -114,7 +121,7 @@ func (m *Metrics) SendPoints() (errs []error) {
 	return
 }
 
-// processPoints is helper function for SendPoints.
+// processPoints is helper function for ProcessPoints.
 func (m *Metrics) processPoints(asset Asset) error {
 	if asset == nil {
 		return nil

From f507c2da0f35be09d7eac6445e9128760e9515b7 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 14:04:35 -0700
Subject: [PATCH 02/12] fix typo

---
 integrations/promunifi/pkg/unifi-poller/unifi.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integrations/promunifi/pkg/unifi-poller/unifi.go b/integrations/promunifi/pkg/unifi-poller/unifi.go
index 3da6d10a..88836428 100644
--- a/integrations/promunifi/pkg/unifi-poller/unifi.go
+++ b/integrations/promunifi/pkg/unifi-poller/unifi.go
@@ -55,7 +55,7 @@ func (u *UnifiPoller) PollController() error {
 }
 
 // CollectMetrics grabs all the measurements from a UniFi controller and returns them.
-// This also creates an InfluxDB writer, and retuns error if that fails.
+// This also creates an InfluxDB writer, and returns error if that fails.
 func (u *UnifiPoller) CollectMetrics() (*Metrics, error) {
 	m := &Metrics{}
 	var err error

From 1bc30246c567537e526ef1bcf097720cc03dbc81 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 16:27:58 -0700
Subject: [PATCH 03/12] update examples for lambda config

---
 integrations/promunifi/examples/up.conf.example |  9 +++++++--
 integrations/promunifi/examples/up.json.example |  1 +
 integrations/promunifi/examples/up.xml.example  | 11 +++++++++--
 integrations/promunifi/examples/up.yaml.example |  9 +++++++--
 4 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/integrations/promunifi/examples/up.conf.example b/integrations/promunifi/examples/up.conf.example
index 3d0bcd1b..91a4d86c 100644
--- a/integrations/promunifi/examples/up.conf.example
+++ b/integrations/promunifi/examples/up.conf.example
@@ -1,4 +1,4 @@
-# unifi-poller primary configuration file. TOML FORMAT #
+# UniFi Poller primary configuration file. TOML FORMAT #
 # commented lines are defaults, uncomment to change.   #
 ########################################################
 
@@ -20,7 +20,12 @@
 # Recommend enabling debug with this setting for better error logging.
 #quiet = false
 
-# If the poller experiences an error from the UniFi Controller or from InfluxDB
+# Lambda mode makes the application exit after collecting and reporting metrics one time.
+# This mode requires an external process like an AWS Lambda or a simple crontab to keep
+# the timings accurate on UniFi Poller run intervals.
+#lambda = false
+
+# If the poller experiences an error from the UniFi controller or from InfluxDB
 # it will exit. If you do not want it to exit, change max_errors to -1. You can
 # adjust the config to tolerate more errors by setting this to a higher value.
 # Recommend setting this between 0 and 5. See man page for more explanation.
diff --git a/integrations/promunifi/examples/up.json.example b/integrations/promunifi/examples/up.json.example
index 3ce54de4..019c9b13 100644
--- a/integrations/promunifi/examples/up.json.example
+++ b/integrations/promunifi/examples/up.json.example
@@ -3,6 +3,7 @@
  "interval": "30s",
  "debug": false,
  "quiet": false,
+ "lambda": false,
  "max_errors": 0,
  "influx_url": "http://127.0.0.1:8086",
  "influx_user": "unifi",
diff --git a/integrations/promunifi/examples/up.xml.example b/integrations/promunifi/examples/up.xml.example
index ccd6cf98..5247154d 100644
--- a/integrations/promunifi/examples/up.xml.example
+++ b/integrations/promunifi/examples/up.xml.example
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!--
 #######################################################
-# unifi-poller primary configuration file. XML FORMAT #
+# UniFi Poller primary configuration file. XML FORMAT #
 # provided values are defaults                        #
 #######################################################
 -->
@@ -36,7 +36,14 @@
   <quiet>false</quiet>
 
   <!--
-  # If the poller experiences an error from the UniFi Controller or from InfluxDB
+  # Lambda mode makes the application exit after collecting and reporting metrics one time.
+  # This mode requires an external process like an AWS Lambda or a simple crontab to keep
+  # the timings accurate on UniFi Poller run intervals.
+  -->
+  <lambda>false</lambda>
+
+  <!--
+  # If the poller experiences an error from the UniFi controller or from InfluxDB
   # it will exit. If you do not want it to exit, change max_errors to -1. You can
   # adjust the config to tolerate more errors by setting this to a higher value.
   # Recommend setting this between 0 and 5. See man page for more explanation.
diff --git a/integrations/promunifi/examples/up.yaml.example b/integrations/promunifi/examples/up.yaml.example
index b4685d20..792b05d1 100644
--- a/integrations/promunifi/examples/up.yaml.example
+++ b/integrations/promunifi/examples/up.yaml.example
@@ -1,4 +1,4 @@
-# unifi-poller primary configuration file. YAML FORMAT #
+# UniFi Poller primary configuration file. YAML FORMAT #
 # provided values are defaults                         #
 ########################################################
 ---
@@ -21,7 +21,12 @@ debug: false
 # Recommend enabling debug with this setting for better error logging.
 quiet: false
 
-# If the poller experiences an error from the UniFi Controller or from InfluxDB
+# Lambda mode makes the application exit after collecting and reporting metrics one time.
+# This mode requires an external process like an AWS Lambda or a simple crontab to keep
+# the timings accurate on UniFi Poller run intervals.
+lambda: false
+
+# If the poller experiences an error from the UniFi controller or from InfluxDB
 # it will exit. If you do not want it to exit, change max_errors to -1. You can
 # adjust the config to tolerate more errors by setting this to a higher value.
 # Recommend setting this between 0 and 5. See man page for more explanation.

From 479a857672957170a4f80e8854fc80276d4497b3 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 16:47:22 -0700
Subject: [PATCH 04/12] Update man page.

---
 .../promunifi/cmd/unifi-poller/README.md      | 10 ++++++++
 .../promunifi/cmd/unifi-poller/start.go       | 25 -------------------
 2 files changed, 10 insertions(+), 25 deletions(-)
 delete mode 100644 integrations/promunifi/cmd/unifi-poller/start.go

diff --git a/integrations/promunifi/cmd/unifi-poller/README.md b/integrations/promunifi/cmd/unifi-poller/README.md
index f56daef0..562cce2e 100644
--- a/integrations/promunifi/cmd/unifi-poller/README.md
+++ b/integrations/promunifi/cmd/unifi-poller/README.md
@@ -83,6 +83,16 @@ is provided so the application can be easily adapted to any environment.
         errors will be logged. Using this with debug=true adds line numbers to
         any error logs.
 
+    lambda         default: false
+        * Until reports of astonishing success with this mode, it is EXPERIMENTAL.
+        Setting this to true will invoke a run-once mode where the application
+        immediately polls the controller and reports the metrics to InfluxDB.
+        Then it exits. This mode is useful in an AWS Lambda or a crontab where
+        the execution timings are controlled. This mode may also be adapted
+        to run in other collector scripts and apps like telegraf or diamond.
+        This mode can also be combined with a "test database" in InfluxDB to
+        give yourself a "test config file" you may run ad-hoc to test changes.
+
     max_errors     default: 0
         If you restart the UniFI controller, the poller will lose access until
         it is restarted. Specifying a number greater than -1 for max_errors will
diff --git a/integrations/promunifi/cmd/unifi-poller/start.go b/integrations/promunifi/cmd/unifi-poller/start.go
deleted file mode 100644
index 5ed7608f..00000000
--- a/integrations/promunifi/cmd/unifi-poller/start.go
+++ /dev/null
@@ -1,25 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"log"
-	"os"
-
-	unifipoller "github.com/davidnewhall/unifi-poller/pkg/unifi-poller"
-)
-
-func main() {
-	log.SetFlags(log.LstdFlags)
-	unifi := &unifipoller.UnifiPoller{}
-	if unifi.ParseFlags(os.Args[1:]); unifi.ShowVer {
-		fmt.Printf("unifi-poller v%s\n", unifipoller.Version)
-		return // don't run anything else w/ version request.
-	}
-	if err := unifi.GetConfig(); err != nil {
-		unifi.Flag.Usage()
-		log.Fatalf("[ERROR] config file '%v': %v", unifi.ConfigFile, err)
-	}
-	if err := unifi.Run(); err != nil {
-		log.Fatalln("[ERROR]", err)
-	}
-}

From 05a59e54da64de3631b57010521e847067087461 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 16:47:34 -0700
Subject: [PATCH 05/12] move file.

---
 .../promunifi/cmd/unifi-poller/main.go        | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 integrations/promunifi/cmd/unifi-poller/main.go

diff --git a/integrations/promunifi/cmd/unifi-poller/main.go b/integrations/promunifi/cmd/unifi-poller/main.go
new file mode 100644
index 00000000..597cea74
--- /dev/null
+++ b/integrations/promunifi/cmd/unifi-poller/main.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	unifipoller "github.com/davidnewhall/unifi-poller/pkg/unifi-poller"
+)
+
+func main() {
+	log.SetFlags(log.LstdFlags)
+	if err := run(); err != nil {
+		log.Fatalln("[ERROR]", err)
+	}
+}
+
+func run() error {
+	unifi := &unifipoller.UnifiPoller{}
+	if unifi.ParseFlags(os.Args[1:]); unifi.ShowVer {
+		fmt.Printf("unifi-poller v%s\n", unifipoller.Version)
+		return nil // don't run anything else w/ version request.
+	}
+	if err := unifi.GetConfig(); err != nil {
+		unifi.Flag.Usage()
+		return err
+	}
+	return unifi.Run()
+}

From f2c267ea024a08a70e42827feb8ff780642d5a48 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 18:38:27 -0700
Subject: [PATCH 06/12] Rename a few things.

---
 integrations/promunifi/.gitignore             |  2 ++
 integrations/promunifi/Makefile               | 17 +++++------
 integrations/promunifi/README.md              |  2 +-
 .../promunifi/cmd/unifi-poller/main.go        | 21 ++------------
 .../README.md => examples/MANUAL.md}          |  0
 .../unifi-poller => unifipoller}/config.go    |  0
 .../jsondebug.go => unifipoller/dumper.go}    |  0
 .../unifi-poller => unifipoller}/helpers.go   |  0
 .../poller.go => unifipoller/unifipoller.go}  | 29 ++++++++++++++-----
 .../unifi.go => unifipoller/worker.go}        | 17 ++++++++---
 10 files changed, 48 insertions(+), 40 deletions(-)
 rename integrations/promunifi/{cmd/unifi-poller/README.md => examples/MANUAL.md} (100%)
 rename integrations/promunifi/{pkg/unifi-poller => unifipoller}/config.go (100%)
 rename integrations/promunifi/{pkg/unifi-poller/jsondebug.go => unifipoller/dumper.go} (100%)
 rename integrations/promunifi/{pkg/unifi-poller => unifipoller}/helpers.go (100%)
 rename integrations/promunifi/{pkg/unifi-poller/poller.go => unifipoller/unifipoller.go} (87%)
 rename integrations/promunifi/{pkg/unifi-poller/unifi.go => unifipoller/worker.go} (92%)

diff --git a/integrations/promunifi/.gitignore b/integrations/promunifi/.gitignore
index ae5e0972..933a6837 100644
--- a/integrations/promunifi/.gitignore
+++ b/integrations/promunifi/.gitignore
@@ -15,6 +15,8 @@
 *~
 /package_build_*
 /release
+MANUAL
+MANUAL.html
 README
 README.html
 /unifi-poller_manual.html
diff --git a/integrations/promunifi/Makefile b/integrations/promunifi/Makefile
index 2c4cbfc5..e2053896 100644
--- a/integrations/promunifi/Makefile
+++ b/integrations/promunifi/Makefile
@@ -7,7 +7,6 @@ MAINT=David Newhall II <david at sleepers dot pro>
 DESC=This daemon polls a UniFi controller at a short interval and stores the collected measurements in an Influx Database.
 GOLANGCI_LINT_ARGS=--enable-all -D gochecknoglobals
 PACKAGE:=./cmd/$(BINARY)
-LIBRARY:=./pkg/$(BINARY)
 DOCKER_REPO=golift
 MD2ROFF_BIN=github.com/github/hub/md2roff-bin
 
@@ -46,9 +45,9 @@ clean:
 man: $(BINARY).1.gz
 $(BINARY).1.gz: md2roff
 	# Building man page. Build dependency first: md2roff
-	go run $(MD2ROFF_BIN) --manual $(BINARY) --version $(VERSION) --date "$$(date)" cmd/$(BINARY)/README.md
-	gzip -9nc cmd/$(BINARY)/README > $(BINARY).1.gz
-	mv cmd/$(BINARY)/README.html $(BINARY)_manual.html
+	go run $(MD2ROFF_BIN) --manual $(BINARY) --version $(VERSION) --date "$$(date)" examples/MANUAL.md
+	gzip -9nc examples/MANUAL > $(BINARY).1.gz
+	mv examples/MANUAL.html $(BINARY)_manual.html
 
 md2roff:
 	go get $(MD2ROFF_BIN)
@@ -63,23 +62,23 @@ README.html: md2roff
 
 build: $(BINARY)
 $(BINARY):
-	go build -o $(BINARY) -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
+	go build -o $(BINARY) -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
 
 linux: $(BINARY).linux
 $(BINARY).linux:
 	# Building linux binary.
-	GOOS=linux go build -o $(BINARY).linux -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=linux go build -o $(BINARY).linux -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
 
 macos: $(BINARY).macos
 $(BINARY).macos:
 	# Building darwin binary.
-	GOOS=darwin go build -o $(BINARY).macos -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=darwin go build -o $(BINARY).macos -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
 
 exe: $(BINARY).exe
 windows: $(BINARY).exe
 $(BINARY).exe:
 	# Building windows binary.
-	GOOS=windows go build -o $(BINARY).exe -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=windows go build -o $(BINARY).exe -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
 
 # Packages
 
@@ -148,7 +147,7 @@ $(BINARY).rb: v$(VERSION).tar.gz.sha256
 # Run code tests and lint.
 test: lint
 	# Testing.
-	go test -race -covermode=atomic $(PACKAGE) $(LIBRARY)
+	go test -race -covermode=atomic ./...
 lint:
 	# Checking lint.
 	golangci-lint run $(GOLANGCI_LINT_ARGS)
diff --git a/integrations/promunifi/README.md b/integrations/promunifi/README.md
index 12bce32c..32058e2b 100644
--- a/integrations/promunifi/README.md
+++ b/integrations/promunifi/README.md
@@ -85,7 +85,7 @@ means, if you wanted to do something like make telegraf collect your data instea
 of UniFi Poller you can achieve that with a little bit of Go code. You could write
 a small app that acts as a telegraf input plugin using the [unifi](https://github.com/golift/unifi)
 library to grab the data from your controller. As a bonus, all of the code in UniFi Poller is
-[also a library](https://godoc.org/github.com/davidnewhall/unifi-poller/pkg/unifi-poller)
+[also a library](https://godoc.org/github.com/davidnewhall/unifi-poller/unifipoller)
 and can be used in other projects.
 
 # What now...
diff --git a/integrations/promunifi/cmd/unifi-poller/main.go b/integrations/promunifi/cmd/unifi-poller/main.go
index 597cea74..768d4af6 100644
--- a/integrations/promunifi/cmd/unifi-poller/main.go
+++ b/integrations/promunifi/cmd/unifi-poller/main.go
@@ -1,29 +1,14 @@
 package main
 
 import (
-	"fmt"
 	"log"
-	"os"
 
-	unifipoller "github.com/davidnewhall/unifi-poller/pkg/unifi-poller"
+	"github.com/davidnewhall/unifi-poller/unifipoller"
 )
 
+// Keep it simple.
 func main() {
-	log.SetFlags(log.LstdFlags)
-	if err := run(); err != nil {
+	if err := unifipoller.Start(); err != nil {
 		log.Fatalln("[ERROR]", err)
 	}
 }
-
-func run() error {
-	unifi := &unifipoller.UnifiPoller{}
-	if unifi.ParseFlags(os.Args[1:]); unifi.ShowVer {
-		fmt.Printf("unifi-poller v%s\n", unifipoller.Version)
-		return nil // don't run anything else w/ version request.
-	}
-	if err := unifi.GetConfig(); err != nil {
-		unifi.Flag.Usage()
-		return err
-	}
-	return unifi.Run()
-}
diff --git a/integrations/promunifi/cmd/unifi-poller/README.md b/integrations/promunifi/examples/MANUAL.md
similarity index 100%
rename from integrations/promunifi/cmd/unifi-poller/README.md
rename to integrations/promunifi/examples/MANUAL.md
diff --git a/integrations/promunifi/pkg/unifi-poller/config.go b/integrations/promunifi/unifipoller/config.go
similarity index 100%
rename from integrations/promunifi/pkg/unifi-poller/config.go
rename to integrations/promunifi/unifipoller/config.go
diff --git a/integrations/promunifi/pkg/unifi-poller/jsondebug.go b/integrations/promunifi/unifipoller/dumper.go
similarity index 100%
rename from integrations/promunifi/pkg/unifi-poller/jsondebug.go
rename to integrations/promunifi/unifipoller/dumper.go
diff --git a/integrations/promunifi/pkg/unifi-poller/helpers.go b/integrations/promunifi/unifipoller/helpers.go
similarity index 100%
rename from integrations/promunifi/pkg/unifi-poller/helpers.go
rename to integrations/promunifi/unifipoller/helpers.go
diff --git a/integrations/promunifi/pkg/unifi-poller/poller.go b/integrations/promunifi/unifipoller/unifipoller.go
similarity index 87%
rename from integrations/promunifi/pkg/unifi-poller/poller.go
rename to integrations/promunifi/unifipoller/unifipoller.go
index 2636ced0..ad8716b9 100644
--- a/integrations/promunifi/pkg/unifi-poller/poller.go
+++ b/integrations/promunifi/unifipoller/unifipoller.go
@@ -17,6 +17,22 @@ import (
 	yaml "gopkg.in/yaml.v2"
 )
 
+// Start begins the application from a CLI.
+// Parses flags, parses config and executes Run().
+func Start() error {
+	log.SetFlags(log.LstdFlags)
+	up := &UnifiPoller{}
+	if up.ParseFlags(os.Args[1:]); up.ShowVer {
+		fmt.Printf("unifi-poller v%s\n", Version)
+		return nil // don't run anything else w/ version request.
+	}
+	if err := up.GetConfig(); err != nil {
+		up.Flag.Usage()
+		return err
+	}
+	return up.Run()
+}
+
 // ParseFlags runs the parser.
 func (u *UnifiPoller) ParseFlags(args []string) {
 	u.Flag = flag.NewFlagSet("unifi-poller", flag.ExitOnError)
@@ -71,21 +87,18 @@ func (u *UnifiPoller) Run() (err error) {
 		u.LogDebugf("Debug Logging Enabled")
 	}
 	log.Printf("[INFO] UniFi Poller v%v Starting Up! PID: %d", Version, os.Getpid())
-
 	if err = u.GetUnifi(); err != nil {
 		return err
 	}
 	if err = u.GetInfluxDB(); err != nil {
 		return err
 	}
-	if u.Lambda {
-		metrics, err := u.CollectMetrics()
-		if err != nil {
-			return err
-		}
-		return u.ReportMetrics(metrics)
+	switch {
+	case u.Lambda:
+		return u.CollectAndReport()
+	default:
+		return u.PollController()
 	}
-	return u.PollController()
 }
 
 // GetInfluxDB returns an InfluxDB interface.
diff --git a/integrations/promunifi/pkg/unifi-poller/unifi.go b/integrations/promunifi/unifipoller/worker.go
similarity index 92%
rename from integrations/promunifi/pkg/unifi-poller/unifi.go
rename to integrations/promunifi/unifipoller/worker.go
index 88836428..7ab7709d 100644
--- a/integrations/promunifi/pkg/unifi-poller/unifi.go
+++ b/integrations/promunifi/unifipoller/worker.go
@@ -43,10 +43,7 @@ func (u *UnifiPoller) PollController() error {
 	log.Println("[INFO] Everything checks out! Poller started, interval:", u.Interval.Round(time.Second))
 	ticker := time.NewTicker(u.Interval.Round(time.Second))
 	for range ticker.C {
-		metrics, err := u.CollectMetrics()
-		if err == nil {
-			u.LogError(u.ReportMetrics(metrics), "reporting metrics")
-		}
+		_ = u.CollectAndReport()
 		if u.MaxErrors >= 0 && u.errorCount > u.MaxErrors {
 			return errors.Errorf("reached maximum error count, stopping poller (%d > %d)", u.errorCount, u.MaxErrors)
 		}
@@ -54,6 +51,18 @@ func (u *UnifiPoller) PollController() error {
 	return nil
 }
 
+// CollectAndReport collects measurements and reports them to influxdb.
+// Can be called once or in a ticker loop.
+func (u *UnifiPoller) CollectAndReport() error {
+	metrics, err := u.CollectMetrics()
+	if err != nil {
+		return err
+	}
+	err = u.ReportMetrics(metrics)
+	u.LogError(err, "reporting metrics")
+	return err
+}
+
 // CollectMetrics grabs all the measurements from a UniFi controller and returns them.
 // This also creates an InfluxDB writer, and returns error if that fails.
 func (u *UnifiPoller) CollectMetrics() (*Metrics, error) {

From b49d85272c70ecc963df31941ca433fe292b5856 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 18:53:46 -0700
Subject: [PATCH 07/12]  a few fixes

---
 integrations/promunifi/unifipoller/config.go      | 1 -
 integrations/promunifi/unifipoller/unifipoller.go | 6 +++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/integrations/promunifi/unifipoller/config.go b/integrations/promunifi/unifipoller/config.go
index 8f2596d6..9c66df55 100644
--- a/integrations/promunifi/unifipoller/config.go
+++ b/integrations/promunifi/unifipoller/config.go
@@ -33,7 +33,6 @@ type UnifiPoller struct {
 	ConfigFile string
 	DumpJSON   string
 	ShowVer    bool
-	Lambda     bool
 	Flag       *pflag.FlagSet
 	errorCount int
 	influx.Client
diff --git a/integrations/promunifi/unifipoller/unifipoller.go b/integrations/promunifi/unifipoller/unifipoller.go
index ad8716b9..ff1fe0f1 100644
--- a/integrations/promunifi/unifipoller/unifipoller.go
+++ b/integrations/promunifi/unifipoller/unifipoller.go
@@ -22,7 +22,10 @@ import (
 func Start() error {
 	log.SetFlags(log.LstdFlags)
 	up := &UnifiPoller{}
-	if up.ParseFlags(os.Args[1:]); up.ShowVer {
+	if os.Args != nil && len(os.Args) > 1 {
+		up.ParseFlags(os.Args[1:])
+	}
+	if up.ShowVer {
 		fmt.Printf("unifi-poller v%s\n", Version)
 		return nil // don't run anything else w/ version request.
 	}
@@ -95,6 +98,7 @@ func (u *UnifiPoller) Run() (err error) {
 	}
 	switch {
 	case u.Lambda:
+		u.LogDebugf("Lambda Mode Enabled")
 		return u.CollectAndReport()
 	default:
 		return u.PollController()

From 88c951583a539853a644de888683ab72d5f869b9 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 19:15:53 -0700
Subject: [PATCH 08/12] more docs.

---
 integrations/promunifi/unifipoller/worker.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/integrations/promunifi/unifipoller/worker.go b/integrations/promunifi/unifipoller/worker.go
index 7ab7709d..572f4e8b 100644
--- a/integrations/promunifi/unifipoller/worker.go
+++ b/integrations/promunifi/unifipoller/worker.go
@@ -52,7 +52,10 @@ func (u *UnifiPoller) PollController() error {
 }
 
 // CollectAndReport collects measurements and reports them to influxdb.
-// Can be called once or in a ticker loop.
+// Can be called once or in a ticker loop. This function and all the ones below
+// handle their own logging. An error is returned so the calling function may
+// determine if there was a read or write erorr and act on it. This is currently
+// called in two places in this library. One returns an error, one does not.
 func (u *UnifiPoller) CollectAndReport() error {
 	metrics, err := u.CollectMetrics()
 	if err != nil {
@@ -64,7 +67,7 @@ func (u *UnifiPoller) CollectAndReport() error {
 }
 
 // CollectMetrics grabs all the measurements from a UniFi controller and returns them.
-// This also creates an InfluxDB writer, and returns error if that fails.
+// This also creates an InfluxDB writer, and returns an error if that fails.
 func (u *UnifiPoller) CollectMetrics() (*Metrics, error) {
 	m := &Metrics{}
 	var err error
@@ -91,7 +94,7 @@ func (u *UnifiPoller) ReportMetrics(metrics *Metrics) error {
 	}
 	err := u.Write(metrics.BatchPoints)
 	if err != nil {
-		return errors.Wrap(err, "infdb.Write(bp)")
+		return errors.Wrap(err, "influxdb.Write(points)")
 	}
 	var fields, points int
 	for _, p := range metrics.Points() {
@@ -108,6 +111,12 @@ func (u *UnifiPoller) ReportMetrics(metrics *Metrics) error {
 
 // ProcessPoints batches all device and client data into influxdb data points.
 // Call this after you've collected all the data you care about.
+// This function is sorta weird and returns a slice of errors. The reasoning is
+// that some points may process while others fail, so we attempt to process them
+// all. This is (usually) run in a loop, so we can't really exit on error,
+// we just log the errors and tally them on a counter. In reality, this never
+// returns any errors because we control the data going in; cool right? But we
+// still check&log it in case the data going is skewed up and causes errors!
 func (m *Metrics) ProcessPoints() (errs []error) {
 	for _, asset := range m.Sites {
 		errs = append(errs, m.processPoints(asset))

From e7b09b65ca5e643fd260253590ca6a5ede2b17a5 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 19:33:57 -0700
Subject: [PATCH 09/12] allow missing sites in config

---
 integrations/promunifi/unifipoller/unifipoller.go | 1 +
 integrations/promunifi/unifipoller/worker.go      | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/integrations/promunifi/unifipoller/unifipoller.go b/integrations/promunifi/unifipoller/unifipoller.go
index ff1fe0f1..1f3960e5 100644
--- a/integrations/promunifi/unifipoller/unifipoller.go
+++ b/integrations/promunifi/unifipoller/unifipoller.go
@@ -128,6 +128,7 @@ func (u *UnifiPoller) GetUnifi() (err error) {
 	}
 	u.Unifi.ErrorLog = u.LogErrorf // Log all errors.
 	u.Unifi.DebugLog = u.LogDebugf // Log debug messages.
+	// this may fail? but we'll try one more time with u.CheckSites below.
 	v, err := u.GetServer()
 	if err != nil {
 		v.ServerVersion = "unknown"
diff --git a/integrations/promunifi/unifipoller/worker.go b/integrations/promunifi/unifipoller/worker.go
index 572f4e8b..8192d4d7 100644
--- a/integrations/promunifi/unifipoller/worker.go
+++ b/integrations/promunifi/unifipoller/worker.go
@@ -32,7 +32,8 @@ FIRST:
 				continue FIRST
 			}
 		}
-		return errors.Errorf("configured site not found on controller: %v", s)
+		// This is fine, it may get added later.
+		u.LogErrorf("configured site not found on controller: %v", s)
 	}
 	return nil
 }
@@ -154,7 +155,7 @@ func (m *Metrics) processPoints(asset Asset) error {
 
 // GetFilteredSites returns a list of sites to fetch data for.
 // Omits requested but unconfigured sites. Grabs the full list from the
-// controller and filters the sites provided in the config file.
+// controller and returns the sites provided in the config file.
 func (u *UnifiPoller) GetFilteredSites() (unifi.Sites, error) {
 	sites, err := u.GetSites()
 	if err != nil {

From 9fb83f6c483c0ded090517cc992e0c2c4e3e9f96 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 19:55:26 -0700
Subject: [PATCH 10/12] Call it mode.

---
 integrations/promunifi/examples/MANUAL.md         | 11 ++++++++---
 integrations/promunifi/examples/up.conf.example   | 14 ++++++++++----
 integrations/promunifi/examples/up.json.example   |  2 +-
 integrations/promunifi/examples/up.xml.example    | 14 ++++++++++----
 integrations/promunifi/examples/up.yaml.example   | 14 ++++++++++----
 integrations/promunifi/unifipoller/config.go      |  2 +-
 integrations/promunifi/unifipoller/unifipoller.go |  9 ++++-----
 integrations/promunifi/unifipoller/worker.go      |  4 ++++
 8 files changed, 48 insertions(+), 22 deletions(-)

diff --git a/integrations/promunifi/examples/MANUAL.md b/integrations/promunifi/examples/MANUAL.md
index 562cce2e..a89fb096 100644
--- a/integrations/promunifi/examples/MANUAL.md
+++ b/integrations/promunifi/examples/MANUAL.md
@@ -83,9 +83,14 @@ is provided so the application can be easily adapted to any environment.
         errors will be logged. Using this with debug=true adds line numbers to
         any error logs.
 
-    lambda         default: false
-        * Until reports of astonishing success with this mode, it is EXPERIMENTAL.
-        Setting this to true will invoke a run-once mode where the application
+    mode           default: "influx"
+        * Value: influx
+        This default mode runs this application as a daemon. It will poll
+        the controller at the configured interval. Providing an invalid value
+        will run in this default mode.
+
+        * Value: influxlambda  -  (the only other available option right now)
+        Setting this value will invoke a run-once mode where the application
         immediately polls the controller and reports the metrics to InfluxDB.
         Then it exits. This mode is useful in an AWS Lambda or a crontab where
         the execution timings are controlled. This mode may also be adapted
diff --git a/integrations/promunifi/examples/up.conf.example b/integrations/promunifi/examples/up.conf.example
index 91a4d86c..7f3f33d4 100644
--- a/integrations/promunifi/examples/up.conf.example
+++ b/integrations/promunifi/examples/up.conf.example
@@ -20,10 +20,16 @@
 # Recommend enabling debug with this setting for better error logging.
 #quiet = false
 
-# Lambda mode makes the application exit after collecting and reporting metrics one time.
-# This mode requires an external process like an AWS Lambda or a simple crontab to keep
-# the timings accurate on UniFi Poller run intervals.
-#lambda = false
+# Which mode to run this application in. The default mode is "influx". Providing
+# an invalid mode will also result in "influx". In this default mode the application
+# runs as a daemon and polls the controller at the configured interval.
+#
+# There is only one other option at this time: "influxlambda"
+#
+# Lambda mode makes the application exit after collecting and reporting metrics
+# to InfluxDB one time. This mode requires an external process like an AWS Lambda
+# or a simple crontab to keep the timings accurate on UniFi Poller run intervals.
+#mode = "influx"
 
 # If the poller experiences an error from the UniFi controller or from InfluxDB
 # it will exit. If you do not want it to exit, change max_errors to -1. You can
diff --git a/integrations/promunifi/examples/up.json.example b/integrations/promunifi/examples/up.json.example
index 019c9b13..4ebbf7e8 100644
--- a/integrations/promunifi/examples/up.json.example
+++ b/integrations/promunifi/examples/up.json.example
@@ -3,7 +3,7 @@
  "interval": "30s",
  "debug": false,
  "quiet": false,
- "lambda": false,
+ "mode": "influx",
  "max_errors": 0,
  "influx_url": "http://127.0.0.1:8086",
  "influx_user": "unifi",
diff --git a/integrations/promunifi/examples/up.xml.example b/integrations/promunifi/examples/up.xml.example
index 5247154d..ba09ee5a 100644
--- a/integrations/promunifi/examples/up.xml.example
+++ b/integrations/promunifi/examples/up.xml.example
@@ -36,11 +36,17 @@
   <quiet>false</quiet>
 
   <!--
-  # Lambda mode makes the application exit after collecting and reporting metrics one time.
-  # This mode requires an external process like an AWS Lambda or a simple crontab to keep
-  # the timings accurate on UniFi Poller run intervals.
+  # Which mode to run this application in. The default mode is "influx". Providing
+  # an invalid mode will also result in "influx". In this default mode the application
+  # runs as a daemon and polls the controller at the configured interval.
+  #
+  # There is only one other option at this time: "influxlambda"
+  #
+  # Lambda mode makes the application exit after collecting and reporting metrics
+  # to InfluxDB one time. This mode requires an external process like an AWS Lambda
+  # or a simple crontab to keep the timings accurate on UniFi Poller run intervals.
   -->
-  <lambda>false</lambda>
+  <mode>influx</mode>
 
   <!--
   # If the poller experiences an error from the UniFi controller or from InfluxDB
diff --git a/integrations/promunifi/examples/up.yaml.example b/integrations/promunifi/examples/up.yaml.example
index 792b05d1..407bce22 100644
--- a/integrations/promunifi/examples/up.yaml.example
+++ b/integrations/promunifi/examples/up.yaml.example
@@ -21,10 +21,16 @@ debug: false
 # Recommend enabling debug with this setting for better error logging.
 quiet: false
 
-# Lambda mode makes the application exit after collecting and reporting metrics one time.
-# This mode requires an external process like an AWS Lambda or a simple crontab to keep
-# the timings accurate on UniFi Poller run intervals.
-lambda: false
+# Which mode to run this application in. The default mode is "influx". Providing
+# an invalid mode will also result in "influx". In this default mode the application
+# runs as a daemon and polls the controller at the configured interval.
+#
+# There is only one other option at this time: "influxlambda"
+#
+# Lambda mode makes the application exit after collecting and reporting metrics
+# to InfluxDB one time. This mode requires an external process like an AWS Lambda
+# or a simple crontab to keep the timings accurate on UniFi Poller run intervals.
+mode: "influx"
 
 # If the poller experiences an error from the UniFi controller or from InfluxDB
 # it will exit. If you do not want it to exit, change max_errors to -1. You can
diff --git a/integrations/promunifi/unifipoller/config.go b/integrations/promunifi/unifipoller/config.go
index 9c66df55..e869847c 100644
--- a/integrations/promunifi/unifipoller/config.go
+++ b/integrations/promunifi/unifipoller/config.go
@@ -56,7 +56,7 @@ type Config struct {
 	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"`
-	Lambda     bool     `json:"lambda" toml:"lambda" xml:"lambda" yaml:"lambda"`
+	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"`
diff --git a/integrations/promunifi/unifipoller/unifipoller.go b/integrations/promunifi/unifipoller/unifipoller.go
index 1f3960e5..ee03b361 100644
--- a/integrations/promunifi/unifipoller/unifipoller.go
+++ b/integrations/promunifi/unifipoller/unifipoller.go
@@ -96,8 +96,8 @@ func (u *UnifiPoller) Run() (err error) {
 	if err = u.GetInfluxDB(); err != nil {
 		return err
 	}
-	switch {
-	case u.Lambda:
+	switch strings.ToLower(u.Mode) {
+	case "influxlambda", "lambdainflux", "lambda_influx", "influx_lambda":
 		u.LogDebugf("Lambda Mode Enabled")
 		return u.CollectAndReport()
 	default:
@@ -128,13 +128,12 @@ func (u *UnifiPoller) GetUnifi() (err error) {
 	}
 	u.Unifi.ErrorLog = u.LogErrorf // Log all errors.
 	u.Unifi.DebugLog = u.LogDebugf // Log debug messages.
-	// this may fail? but we'll try one more time with u.CheckSites below.
 	v, err := u.GetServer()
 	if err != nil {
-		v.ServerVersion = "unknown"
+		return err
 	}
 	u.Logf("Authenticated to UniFi Controller at %s version %s as user %s", u.UnifiBase, v.ServerVersion, u.UnifiUser)
-	if err = u.CheckSites(); err != nil {
+	if err := u.CheckSites(); err != nil {
 		return err
 	}
 	u.Logf("Polling UniFi Controller Sites: %v", u.Sites)
diff --git a/integrations/promunifi/unifipoller/worker.go b/integrations/promunifi/unifipoller/worker.go
index 8192d4d7..424aaa34 100644
--- a/integrations/promunifi/unifipoller/worker.go
+++ b/integrations/promunifi/unifipoller/worker.go
@@ -11,7 +11,11 @@ import (
 )
 
 // CheckSites makes sure the list of provided sites exists on the controller.
+// This does not run in Lambda (run-once) mode.
 func (u *UnifiPoller) CheckSites() error {
+	if strings.Contains(strings.ToLower(u.Mode), "lambda") {
+		return nil // Skip this in lambda mode.
+	}
 	sites, err := u.GetSites()
 	if err != nil {
 		return err

From f4b0673c772611dd5567800ee9d38ff1a300551d Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 20:05:42 -0700
Subject: [PATCH 11/12] Run parseflags, always.

---
 integrations/promunifi/unifipoller/unifipoller.go | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/integrations/promunifi/unifipoller/unifipoller.go b/integrations/promunifi/unifipoller/unifipoller.go
index ee03b361..9d368f64 100644
--- a/integrations/promunifi/unifipoller/unifipoller.go
+++ b/integrations/promunifi/unifipoller/unifipoller.go
@@ -22,10 +22,7 @@ import (
 func Start() error {
 	log.SetFlags(log.LstdFlags)
 	up := &UnifiPoller{}
-	if os.Args != nil && len(os.Args) > 1 {
-		up.ParseFlags(os.Args[1:])
-	}
-	if up.ShowVer {
+	if up.ParseFlags(os.Args[1:]); up.ShowVer {
 		fmt.Printf("unifi-poller v%s\n", Version)
 		return nil // don't run anything else w/ version request.
 	}

From e3bfc5e9c4efd803206ec0b167e2f6082fa0447c Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Sun, 23 Jun 2019 20:13:40 -0700
Subject: [PATCH 12/12] Rename the pkg back to what it was.

---
 integrations/promunifi/Makefile                           | 8 ++++----
 integrations/promunifi/README.md                          | 2 +-
 integrations/promunifi/cmd/unifi-poller/main.go           | 2 +-
 .../promunifi/{unifipoller => pkg/unifi-poller}/config.go | 0
 .../promunifi/{unifipoller => pkg/unifi-poller}/dumper.go | 0
 .../{unifipoller => pkg/unifi-poller}/helpers.go          | 0
 .../{unifipoller/worker.go => pkg/unifi-poller/unifi.go}  | 0
 .../{unifipoller => pkg/unifi-poller}/unifipoller.go      | 0
 8 files changed, 6 insertions(+), 6 deletions(-)
 rename integrations/promunifi/{unifipoller => pkg/unifi-poller}/config.go (100%)
 rename integrations/promunifi/{unifipoller => pkg/unifi-poller}/dumper.go (100%)
 rename integrations/promunifi/{unifipoller => pkg/unifi-poller}/helpers.go (100%)
 rename integrations/promunifi/{unifipoller/worker.go => pkg/unifi-poller/unifi.go} (100%)
 rename integrations/promunifi/{unifipoller => pkg/unifi-poller}/unifipoller.go (100%)

diff --git a/integrations/promunifi/Makefile b/integrations/promunifi/Makefile
index e2053896..8fc00a07 100644
--- a/integrations/promunifi/Makefile
+++ b/integrations/promunifi/Makefile
@@ -62,23 +62,23 @@ README.html: md2roff
 
 build: $(BINARY)
 $(BINARY):
-	go build -o $(BINARY) -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
+	go build -o $(BINARY) -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
 
 linux: $(BINARY).linux
 $(BINARY).linux:
 	# Building linux binary.
-	GOOS=linux go build -o $(BINARY).linux -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=linux go build -o $(BINARY).linux -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
 
 macos: $(BINARY).macos
 $(BINARY).macos:
 	# Building darwin binary.
-	GOOS=darwin go build -o $(BINARY).macos -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=darwin go build -o $(BINARY).macos -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
 
 exe: $(BINARY).exe
 windows: $(BINARY).exe
 $(BINARY).exe:
 	# Building windows binary.
-	GOOS=windows go build -o $(BINARY).exe -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/unifipoller.Version=$(VERSION)" $(PACKAGE)
+	GOOS=windows go build -o $(BINARY).exe -ldflags "-w -s -X github.com/davidnewhall/unifi-poller/pkg/unifi-poller.Version=$(VERSION)" $(PACKAGE)
 
 # Packages
 
diff --git a/integrations/promunifi/README.md b/integrations/promunifi/README.md
index 32058e2b..12bce32c 100644
--- a/integrations/promunifi/README.md
+++ b/integrations/promunifi/README.md
@@ -85,7 +85,7 @@ means, if you wanted to do something like make telegraf collect your data instea
 of UniFi Poller you can achieve that with a little bit of Go code. You could write
 a small app that acts as a telegraf input plugin using the [unifi](https://github.com/golift/unifi)
 library to grab the data from your controller. As a bonus, all of the code in UniFi Poller is
-[also a library](https://godoc.org/github.com/davidnewhall/unifi-poller/unifipoller)
+[also a library](https://godoc.org/github.com/davidnewhall/unifi-poller/pkg/unifi-poller)
 and can be used in other projects.
 
 # What now...
diff --git a/integrations/promunifi/cmd/unifi-poller/main.go b/integrations/promunifi/cmd/unifi-poller/main.go
index 768d4af6..8f17b276 100644
--- a/integrations/promunifi/cmd/unifi-poller/main.go
+++ b/integrations/promunifi/cmd/unifi-poller/main.go
@@ -3,7 +3,7 @@ package main
 import (
 	"log"
 
-	"github.com/davidnewhall/unifi-poller/unifipoller"
+	unifipoller "github.com/davidnewhall/unifi-poller/pkg/unifi-poller"
 )
 
 // Keep it simple.
diff --git a/integrations/promunifi/unifipoller/config.go b/integrations/promunifi/pkg/unifi-poller/config.go
similarity index 100%
rename from integrations/promunifi/unifipoller/config.go
rename to integrations/promunifi/pkg/unifi-poller/config.go
diff --git a/integrations/promunifi/unifipoller/dumper.go b/integrations/promunifi/pkg/unifi-poller/dumper.go
similarity index 100%
rename from integrations/promunifi/unifipoller/dumper.go
rename to integrations/promunifi/pkg/unifi-poller/dumper.go
diff --git a/integrations/promunifi/unifipoller/helpers.go b/integrations/promunifi/pkg/unifi-poller/helpers.go
similarity index 100%
rename from integrations/promunifi/unifipoller/helpers.go
rename to integrations/promunifi/pkg/unifi-poller/helpers.go
diff --git a/integrations/promunifi/unifipoller/worker.go b/integrations/promunifi/pkg/unifi-poller/unifi.go
similarity index 100%
rename from integrations/promunifi/unifipoller/worker.go
rename to integrations/promunifi/pkg/unifi-poller/unifi.go
diff --git a/integrations/promunifi/unifipoller/unifipoller.go b/integrations/promunifi/pkg/unifi-poller/unifipoller.go
similarity index 100%
rename from integrations/promunifi/unifipoller/unifipoller.go
rename to integrations/promunifi/pkg/unifi-poller/unifipoller.go