From 760b9d8c59e1c7fd163310f95a9f7622cfc35183 Mon Sep 17 00:00:00 2001
From: asonix <asonix@asonix.dog>
Date: Sat, 7 Oct 2023 12:29:01 -0500
Subject: [PATCH] Install tracing & metrics from within runtime

---
 src/main.rs | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 5f8ee4d..6b671af 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,20 +1,28 @@
 fn main() -> color_eyre::Result<()> {
-    let config = pict_rs::PictRsConfiguration::build_default()?
-        .install_tracing()?
-        .install_metrics()?;
-
-    run(config)
+    run()
 }
 
 #[cfg(feature = "io-uring")]
-fn run(config: pict_rs::PictRsConfiguration) -> color_eyre::Result<()> {
-    tokio_uring::start(config.run())
+fn run() -> color_eyre::Result<()> {
+    tokio_uring::start(async move {
+        pict_rs::PictRsConfiguration::build_default()?
+            .install_tracing()?
+            .install_metrics()?
+            .run()
+            .await
+    })
 }
 
 #[cfg(not(feature = "io-uring"))]
-fn run(config: pict_rs::PictRsConfiguration) -> color_eyre::Result<()> {
+fn run() -> color_eyre::Result<()> {
     tokio::runtime::Builder::new_multi_thread()
         .enable_all()
         .build()?
-        .block_on(config.run())
+        .block_on(async move {
+            pict_rs::PictRsConfiguration::build_default()?
+                .install_tracing()?
+                .install_metrics()?
+                .run_on_localset()
+                .await
+        })
 }