add worker routes

SiaFoundation · Sep 12, 2024 · e14e7a6 · e14e7a6
1 parent 8a5f9d8
commit e14e7a6
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 3 deletions.
diff --git a/api/prometheus.go b/api/prometheus.go
@@ -757,6 +757,66 @@ func (txn Transaction) PrometheusMetric() (metrics []prometheus.Metric) {
 	return
 }
 
+func (m MemoryResponse) PrometheusMetric() (metrics []prometheus.Metric) {
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_memory_download_available",
+		Value: float64(m.Download.Available),
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_memory_download_total",
+		Value: float64(m.Download.Total),
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_memory_upload_available",
+		Value: float64(m.Upload.Available),
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_memory_upload_total",
+		Value: float64(m.Upload.Total),
+	})
+	return
+}
+
+func (m DownloadStatsResponse) PrometheusMetric() (metrics []prometheus.Metric) {
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_avgdownloadspeedmbps",
+		Value: m.AvgDownloadSpeedMBPS,
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_avgoverdrivepct_download",
+		Value: m.AvgOverdrivePct,
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_healthydownloaders",
+		Value: float64(m.HealthyDownloaders),
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_numdownloaders",
+		Value: float64(m.NumDownloaders),
+	})
+	return
+}
+
+func (m UploadStatsResponse) PrometheusMetric() (metrics []prometheus.Metric) {
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_avgslabuploadspeedmbps",
+		Value: m.AvgSlabUploadSpeedMBPS,
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_avgoverdrivepct_upload",
+		Value: m.AvgOverdrivePct,
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_healthyuploaders",
+		Value: float64(m.HealthyUploaders),
+	})
+	metrics = append(metrics, prometheus.Metric{
+		Name:  "renterd_worker_stats_numuploaders",
+		Value: float64(m.NumUploaders),
+	})
+	return
+}
+
 type AllowListResp []types.PublicKey
 
 func (a AllowListResp) PrometheusMetric() (metrics []prometheus.Metric) {

diff --git a/worker/worker.go b/worker/worker.go
@@ -453,7 +453,7 @@ func (w *Worker) downloadsStatsHandlerGET(jc jape.Context) {
 	})
 
 	// encode response
-	jc.Encode(api.DownloadStatsResponse{
+	api.WriteResponse(jc, http.StatusOK, api.DownloadStatsResponse{
 		AvgDownloadSpeedMBPS: math.Ceil(stats.avgDownloadSpeedMBPS*100) / 100,
 		AvgOverdrivePct:      math.Floor(stats.avgOverdrivePct*100*100) / 100,
 		HealthyDownloaders:   healthy,
@@ -478,7 +478,7 @@ func (w *Worker) uploadsStatsHandlerGET(jc jape.Context) {
 	})
 
 	// encode response
-	jc.Encode(api.UploadStatsResponse{
+	api.WriteResponse(jc, http.StatusOK, api.UploadStatsResponse{
 		AvgSlabUploadSpeedMBPS: math.Ceil(stats.avgSlabUploadSpeedMBPS*100) / 100,
 		AvgOverdrivePct:        math.Floor(stats.avgOverdrivePct*100*100) / 100,
 		HealthyUploaders:       stats.healthyUploaders,
@@ -856,7 +856,7 @@ func (w *Worker) idHandlerGET(jc jape.Context) {
 }
 
 func (w *Worker) memoryGET(jc jape.Context) {
-	jc.Encode(api.MemoryResponse{
+	api.WriteResponse(jc, http.StatusOK, api.MemoryResponse{
 		Download: w.downloadManager.mm.Status(),
 		Upload:   w.uploadManager.mm.Status(),
 	})