init

internal/metrics/collector.go

@@ -0,0 +1,54 @@
+package metrics
+
+import (
+	"context"
+	"github.com/jakekeeys/hg612-exporter/pkg/hg612"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+	"time"
+)
+
+type Collector interface {
+	Collect()
+	Start()
+	Stop()
+}
+
+type MetricsCollector struct {
+	ctx                    context.Context
+	collectIntervalSeconds int
+	dslMetricsCollector    dslMetricsCollector
+}
+
+func New(client hg612.Client, host string, collectIntervalSeconds int) Collector {
+	return MetricsCollector{
+		collectIntervalSeconds: collectIntervalSeconds,
+		ctx:                    context.Background(),
+		dslMetricsCollector:    newDSLMetricsCollector(client, host),
+	}
+}
+
+func (c MetricsCollector) Collect() {
+	err := c.dslMetricsCollector.collect()
+	if err != nil {
+		logrus.Error(errors.Wrap(err, "error collecting dsl metrics"))
+	}
+
+}
+
+func (c MetricsCollector) Start() {
+	go func() {
+		for {
+			select {
+			case <-time.After(time.Second * time.Duration(c.collectIntervalSeconds)):
+				c.Collect()
+			case <-c.ctx.Done():
+				return
+			}
+		}
+	}()
+}
+
+func (c MetricsCollector) Stop() {
+	c.ctx.Done()
+}