Add CWService interface

nyaruka · Dec 13, 2024 · 31d318d · 31d318d
1 parent c430b1e
commit 31d318d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/aws/cwatch/service.go b/aws/cwatch/service.go
@@ -13,6 +13,13 @@ import (
 	"github.com/nyaruka/gocommon/syncx"
 )
 
+type CWService interface {
+	Queue(d types.MetricDatum)
+	StartQueue(wg *sync.WaitGroup)
+	StopQueue()
+	Prepare(data []types.MetricDatum) *cloudwatch.PutMetricDataInput
+}
+
 type Service struct {
 	Client     *cloudwatch.Client
 	namespace  string
@@ -71,3 +78,5 @@ func (s *Service) processBatch(batch []types.MetricDatum) {
 		slog.Error("error sending metrics to cloudwatch", "error", err, "count", len(batch))
 	}
 }
+
+var _ CWService = (*Service)(nil)