chore: add download size metric

ethersphere · Oct 19, 2023 · ff3aec3 · ff3aec3
1 parent 2489b23
commit ff3aec3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/pkg/check/longavailability/longavailability.go b/pkg/check/longavailability/longavailability.go
@@ -73,14 +73,15 @@ func (c *Check) Run(ctx context.Context, cluster orchestration.Cluster, o interf
 			c.logger.Infof("node %s: download attempt %d for %s", node.Name(), i+1, addr)
 
 			start := time.Now()
-			_, _, err = node.Client().DownloadFile(ctx, addr)
+			size, _, err := node.Client().DownloadFile(ctx, addr)
 			if err != nil {
 				c.metrics.DownloadErrors.Inc()
 				c.logger.Errorf("node %s: download %s error: %v", node.Name(), addr, err)
 				c.logger.Infof("retrying in: %v", opts.RetryWait)
 				time.Sleep(opts.RetryWait)
 				continue
 			}
+			c.metrics.DownloadSize.Set(float64(size))
 			dur := time.Since(start)
 			c.metrics.DownloadDuration.Observe(dur.Seconds())
 			c.logger.Infof("node %s: downloaded %s successfully in %v", node.Name(), addr, dur)

diff --git a/pkg/check/longavailability/metrics.go b/pkg/check/longavailability/metrics.go
@@ -9,6 +9,7 @@ type metrics struct {
 	DownloadErrors   prometheus.Counter
 	DownloadAttempts prometheus.Counter
 	DownloadDuration prometheus.Histogram
+	DownloadSize     prometheus.Gauge
 }
 
 func newMetrics(subsystem string) metrics {
@@ -34,6 +35,14 @@ func newMetrics(subsystem string) metrics {
 				Name:      "data_download_duration",
 				Help:      "Data download duration through the /bytes endpoint.",
 			}),
+		DownloadSize: prometheus.NewGauge(
+			prometheus.GaugeOpts{
+				Namespace: m.Namespace,
+				Subsystem: subsystem,
+				Name:      "download_size",
+				Help:      "Amount of data downloaded per download.",
+			},
+		),
 	}
 }