Mystara/refraction
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: compilation on windows

Browse source
This commit is contained in:
seth 2024-01-27 01:50:51 -05:00
parent 48c2dba554
commit 203ba111cc
No known key found for this signature in database
GPG key ID: D31BD0D494BBEE86
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/main.rs
View file

@ -15,7 +15,10 @@ use serenity::ShardManager;
use redis::ConnectionLike; use redis::ConnectionLike;
use tokio::signal::ctrl_c; use tokio::signal::ctrl_c;
#[cfg(target_family = "unix")]
use tokio::signal::unix::{signal, SignalKind}; use tokio::signal::unix::{signal, SignalKind};
#[cfg(target_family = "windows")]
use tokio::signal::windows::ctrl_close;
use tokio::sync::Mutex; use tokio::sync::Mutex;
mod api; mod api;
@ -125,12 +128,15 @@ async fn main() -> Result<()> {
.wrap_err_with(|| "Failed to build framework!")?; .wrap_err_with(|| "Failed to build framework!")?;
let shard_manager = framework.shard_manager().clone(); let shard_manager = framework.shard_manager().clone();
#[cfg(target_family = "unix")]
let mut sigterm = signal(SignalKind::terminate())?; let mut sigterm = signal(SignalKind::terminate())?;
#[cfg(target_family = "windows")]
let mut sigterm = ctrl_close()?;
tokio::select! { tokio::select! {
result = framework.start() => result.map_err(Report::from), result = framework.start() => result.map_err(Report::from),
_ = sigterm.recv() => { _ = sigterm.recv() => {
handle_shutdown(shard_manager, "Recieved SIGTERM").await; handle_shutdown(shard_manager, "Received SIGTERM").await;
std::process::exit(0); std::process::exit(0);
} }
_ = ctrl_c() => { _ = ctrl_c() => {