[104] code review changes

HIRO-MicroDataCenters-BV · Jan 13, 2025 · ebcb055 · ebcb055
1 parent 77d0274
commit ebcb055
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 16 deletions.
diff --git a/rhio/src/blobs/actor.rs b/rhio/src/blobs/actor.rs
@@ -131,31 +131,29 @@ impl BlobsActor {
                 DownloadBlobEvent::Abort(err) => {
                     error!(parent: &span, %err, "failed downloading blob");
 
-                    metrics::counter!(
-                        BLOBS_DOWNLOAD_TOTAL,
-                        LABEL_MSG_TYPE => LABEL_BLOB_MSG_TYPE_ERROR,
-                        LABEL_LOCAL_BUCKET => blob.local_bucket_name.to_owned(),
-                        LABEL_REMOTE_BUCKET => blob.remote_bucket_name.to_owned()
-                    )
-                    .increment(1);
+                    BlobsActor::increment_blob_downloads(LABEL_BLOB_MSG_TYPE_ERROR, &blob);
                 }
                 DownloadBlobEvent::Done => {
                     debug!(parent: &span, "finished downloading blob");
 
-                    metrics::counter!(
-                        BLOBS_DOWNLOAD_TOTAL,
-                        LABEL_MSG_TYPE => LABEL_BLOB_MSG_TYPE_DONE,
-                        LABEL_LOCAL_BUCKET => blob.local_bucket_name.to_owned(),
-                        LABEL_REMOTE_BUCKET => blob.remote_bucket_name.to_owned()
-                    )
-                    .increment(1);
+                    BlobsActor::increment_blob_downloads(LABEL_BLOB_MSG_TYPE_DONE, &blob);
                 }
             }
         }
 
         Ok(())
     }
 
+    fn increment_blob_downloads(msg_type: &str, blob: &SignedBlobInfo) {
+        metrics::counter!(
+            BLOBS_DOWNLOAD_TOTAL,
+            LABEL_MSG_TYPE => msg_type.to_owned(),
+            LABEL_LOCAL_BUCKET => blob.local_bucket_name.to_owned(),
+            LABEL_REMOTE_BUCKET => blob.remote_bucket_name.to_owned()
+        )
+        .increment(1);
+    }
+
     async fn shutdown(&mut self) -> Result<()> {
         Ok(())
     }

diff --git a/rhio/src/health.rs b/rhio/src/health.rs
@@ -20,16 +20,20 @@ pub async fn run_http_server(bind_port: u16) -> Result<()> {
         IpAddr::V4(Ipv4Addr::UNSPECIFIED),
         bind_port,
     ))
-    .await?;
+    .await
+    .context("TCP Listener binding")?;
     debug!(
         "HTTP health and metrics endpoint listening on {}",
         listener.local_addr()?
     );
+
     let app = Router::new().route(HTTP_HEALTH_ROUTE, get(health)).route(
         HTTP_METRICS_ROUTE,
         get(move || ready(recorder_handle.render())),
     );
-    axum::serve(listener, app).await?;
+    axum::serve(listener, app)
+        .await
+        .context("HTTP metrics and health serving")?;
     Ok(())
 }