allow pulling specific sites through output plugins

pkg/poller/inputs.go

@@ -15,13 +15,15 @@ var (
 
 // Input plugins must implement this interface.
 type Input interface {
-	Initialize(Logger) error    // Called once on startup to initialize the plugin.
-	Metrics() (*Metrics, error) // Called every time new metrics are requested.
+	Initialize(Logger) error                    // Called once on startup to initialize the plugin.
+	Metrics() (*Metrics, bool, error)           // Called every time new metrics are requested.
+	MetricsFrom(Filter) (*Metrics, bool, error) // Called every time new metrics are requested.
 	RawMetrics(Filter) ([]byte, error)
 }
 
 // InputPlugin describes an input plugin's consumable interface.
 type InputPlugin struct {
+	Name   string
 	Config interface{} // Each config is passed into an unmarshaller later.
 	Input
 }
@@ -61,12 +63,13 @@ func (u *UnifiPoller) InitializeInputs() error {
 }
 
 // Metrics aggregates all the measurements from all configured inputs and returns them.
-func (u *UnifiPoller) Metrics() (*Metrics, error) {
+func (u *UnifiPoller) Metrics() (*Metrics, bool, error) {
 	errs := []string{}
 	metrics := &Metrics{}
+	ok := false
 
 	for _, input := range inputs {
-		m, err := input.Metrics()
+		m, _, err := input.Metrics()
 		if err != nil {
 			errs = append(errs, err.Error())
 		}
@@ -75,6 +78,8 @@ func (u *UnifiPoller) Metrics() (*Metrics, error) {
 			continue
 		}
 
+		ok = true
+
 		metrics.Sites = append(metrics.Sites, m.Sites...)
 		metrics.Clients = append(metrics.Clients, m.Clients...)
 		metrics.IDSList = append(metrics.IDSList, m.IDSList...)
@@ -99,5 +104,54 @@ func (u *UnifiPoller) Metrics() (*Metrics, error) {
 		err = fmt.Errorf(strings.Join(errs, ", "))
 	}
 
-	return metrics, err
+	return metrics, ok, err
+}
+
+// MetricsFrom aggregates all the measurements from all configured inputs and returns them.
+func (u *UnifiPoller) MetricsFrom(filter Filter) (*Metrics, bool, error) {
+	errs := []string{}
+	metrics := &Metrics{}
+	ok := false
+
+	for _, input := range inputs {
+		if input.Name != filter.Type {
+			continue
+		}
+
+		m, _, err := input.MetricsFrom(filter)
+		if err != nil {
+			errs = append(errs, err.Error())
+		}
+
+		if m == nil {
+			continue
+		}
+
+		ok = true
+
+		metrics.Sites = append(metrics.Sites, m.Sites...)
+		metrics.Clients = append(metrics.Clients, m.Clients...)
+		metrics.IDSList = append(metrics.IDSList, m.IDSList...)
+
+		if m.Devices == nil {
+			continue
+		}
+
+		if metrics.Devices == nil {
+			metrics.Devices = &unifi.Devices{}
+		}
+
+		metrics.UAPs = append(metrics.UAPs, m.UAPs...)
+		metrics.USGs = append(metrics.USGs, m.USGs...)
+		metrics.USWs = append(metrics.USWs, m.USWs...)
+		metrics.UDMs = append(metrics.UDMs, m.UDMs...)
+	}
+
+	var err error
+
+	if len(errs) > 0 {
+		err = fmt.Errorf(strings.Join(errs, ", "))
+	}
+
+	return metrics, ok, err
 }