From 9aa2afb0fd0c918b3da04dc6c904b0083f4b25ca Mon Sep 17 00:00:00 2001
From: ssrlive <30760636+ssrlive@users.noreply.github.com>
Date: Wed, 9 Oct 2024 23:54:25 +0800
Subject: [PATCH] Fix daemonize issues

---
 src/bin/main.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/bin/main.rs b/src/bin/main.rs
index 91e88ec..e39b7b4 100644
--- a/src/bin/main.rs
+++ b/src/bin/main.rs
@@ -1,7 +1,6 @@
 use tun2proxy::{Args, BoxError};
 
-#[tokio::main]
-async fn main() -> Result<(), BoxError> {
+fn main() -> Result<(), BoxError> {
     dotenvy::dotenv().ok();
     let args = Args::parse_args();
 
@@ -24,6 +23,11 @@ async fn main() -> Result<(), BoxError> {
         return Ok(());
     }
 
+    let rt = tokio::runtime::Builder::new_multi_thread().enable_all().build()?;
+    rt.block_on(main_async(args))
+}
+
+async fn main_async(args: Args) -> Result<(), BoxError> {
     let default = format!("{:?},hickory_proto=warn", args.verbosity);
     env_logger::Builder::from_env(env_logger::Env::default().default_filter_or(default)).init();