mirror of
https://github.com/supleed2/nanobot.git
synced 2024-11-10 04:35:48 +00:00
Add service module
To multiplex discord bot and http server in shuttle
This commit is contained in:
parent
d62fabb048
commit
1b8914cf67
20
src/service.rs
Normal file
20
src/service.rs
Normal file
|
@ -0,0 +1,20 @@
|
||||||
|
use crate::{Data, Error};
|
||||||
|
|
||||||
|
pub(crate) struct NanoBot {
|
||||||
|
pub discord: poise::FrameworkBuilder<Data, Error>,
|
||||||
|
pub router: axum::Router,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[shuttle_runtime::async_trait]
|
||||||
|
impl shuttle_runtime::Service for NanoBot {
|
||||||
|
async fn bind(mut self, addr: std::net::SocketAddr) -> Result<(), shuttle_runtime::Error> {
|
||||||
|
let serve = axum::Server::bind(&addr).serve(self.router.into_make_service());
|
||||||
|
|
||||||
|
tokio::select! {
|
||||||
|
_ = self.discord.run_autosharded() => {},
|
||||||
|
_ = serve => {},
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue