alemi/postwoman
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: default to single thread runtime

Browse source 
but allow multi-thread-runtime with --multi-threaded flag
This commit is contained in:
əlemi 2024-10-20 01:33:45 +02:00
parent 2695c850bb
commit 6915a71802
Signed by: alemi
GPG key ID: A4895B84D311642C
1 changed files with 25 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/main.rs
View file

@ -20,6 +20,10 @@ struct PostWomanArgs {
/// action to run /// action to run
#[clap(subcommand)] #[clap(subcommand)]
action: Option<PostWomanActions>, action: Option<PostWomanActions>,
/// start a multi-thread runtime, with multiple worker threads
#[arg(long, default_value_t = false)]
multi_threaded: bool,
} }
#[derive(Subcommand, Debug, Default)] #[derive(Subcommand, Debug, Default)]
@ -62,12 +66,29 @@ pub enum PostWomanActions {
const TIMESTAMP_FMT: &str = "%H:%M:%S%.6f"; const TIMESTAMP_FMT: &str = "%H:%M:%S%.6f";
#[tokio::main] fn main() -> Result<(), PostWomanError> {
async fn main() -> Result<(), PostWomanError> {
let args = PostWomanArgs::parse(); let args = PostWomanArgs::parse();
let collection = std::fs::read_to_string(args.collection)?; let collection_raw = std::fs::read_to_string(&args.collection)?;
let config: PostWomanConfig = toml::from_str(&collection)?; let collection: PostWomanCollection = toml::from_str(&collection_raw)?;
if args.multi_threaded {
tokio::runtime::Builder::new_multi_thread()
.enable_all()
.build()
.expect("failed creating tokio multi-thread runtime")
.block_on(async { run_postwoman(args, collection).await })
} else {
tokio::runtime::Builder::new_current_thread()
.enable_all()
.build()
.expect("failed creating tokio current-thread runtime")
.block_on(async { run_postwoman(args, collection).await })
}
}
async fn run_postwoman(args: PostWomanArgs, collection: PostWomanCollection) -> Result<(), PostWomanError> {
let action = args.action.unwrap_or(PostWomanActions::List { verbose: false });
match args.action.unwrap_or_default() { match args.action.unwrap_or_default() {
PostWomanActions::List => { PostWomanActions::List => {