From 203ba111cc887a2be9f9ead4a33cc57f81f8853c Mon Sep 17 00:00:00 2001 From: seth Date: Sat, 27 Jan 2024 01:50:51 -0500 Subject: [PATCH] fix: compilation on windows --- src/main.rs | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/main.rs b/src/main.rs index 4f83a70..cf89bd6 100644 --- a/src/main.rs +++ b/src/main.rs @@ -15,7 +15,10 @@ use serenity::ShardManager; use redis::ConnectionLike; use tokio::signal::ctrl_c; +#[cfg(target_family = "unix")] use tokio::signal::unix::{signal, SignalKind}; +#[cfg(target_family = "windows")] +use tokio::signal::windows::ctrl_close; use tokio::sync::Mutex; mod api; @@ -125,12 +128,15 @@ async fn main() -> Result<()> { .wrap_err_with(|| "Failed to build framework!")?; let shard_manager = framework.shard_manager().clone(); + #[cfg(target_family = "unix")] let mut sigterm = signal(SignalKind::terminate())?; + #[cfg(target_family = "windows")] + let mut sigterm = ctrl_close()?; tokio::select! { result = framework.start() => result.map_err(Report::from), _ = sigterm.recv() => { - handle_shutdown(shard_manager, "Recieved SIGTERM").await; + handle_shutdown(shard_manager, "Received SIGTERM").await; std::process::exit(0); } _ = ctrl_c() => {