diff options
| author | Teddy Wing | 2021-06-07 01:22:39 +0200 | 
|---|---|---|
| committer | Teddy Wing | 2021-06-07 01:25:00 +0200 | 
| commit | de2d6a2ddf747cc7d5d370b69e26e8355b00f1d4 (patch) | |
| tree | eece39ad10565e2ddf4220deb85f7fe406b6ef68 /src | |
| parent | 0da2cb98c759e59772a489b73e91164c6a8f5fe9 (diff) | |
| download | reflectub-de2d6a2ddf747cc7d5d370b69e26e8355b00f1d4.tar.bz2 | |
Switch `futures::executor` to Tokio runtime
Use the Tokio runtime we created to run the blocking async tasks.
Trying to set this up so I can get results back from the spawned tasks,
but I'm currently having trouble working out how to extract them from
the async task and return them from `run()`. I suppose I could just
print out the errors directly in that `while let` loop, but ideally I'd
like to return all errors from `run()` rather than printing in `run()`.
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 24 | 
1 files changed, 18 insertions, 6 deletions
| diff --git a/src/main.rs b/src/main.rs index a8c35be..a476ef0 100644 --- a/src/main.rs +++ b/src/main.rs @@ -2,11 +2,12 @@ use anyhow::{self, Context};  use chrono::DateTime;  use exitcode;  use filetime; -use futures::{executor, future}; +use futures::{self, executor, future};  use getopts::Options;  use parse_size::parse_size;  use sqlx;  use tokio; +use tokio_stream::StreamExt;  use reflectub::{database, git, github}; @@ -99,20 +100,21 @@ fn run() -> anyhow::Result<()> {          .build()?;      let _rt_guard = rt.enter(); -    let repos = executor::block_on(github::fetch_repos(username))?; +    let repos = rt.block_on(github::fetch_repos(username))?;      let db = Arc::new(          tokio::sync::Mutex::new( -            executor::block_on(database::Db::connect(&database_file))?, +            rt.block_on(database::Db::connect(&database_file))?,          )      ); -    executor::block_on(async { +    rt.block_on(async {          db.lock().await              .create().await      })?; -    let mut joins = Vec::new(); +    // let mut joins = futures::stream::FuturesUnordered::new(); +    let mut joins = Vec::with_capacity(repos.len());      for repo in repos {          let db = db.clone(); @@ -134,7 +136,17 @@ fn run() -> anyhow::Result<()> {          joins.push(join);      } -    executor::block_on(future::join_all(joins)); +    // executor::block_on(future::join_all(joins)); +    rt.block_on(async { +        let mut joins = tokio_stream::iter(&joins); + +        while let Some(task) = joins.next().await { +            let a = task.await?; +            dbg!(a); +        } + +        Ok::<(), anyhow::Error>(()) +    })?;      Ok(())  } | 
