admin/populate: Convert to async

Author: Turbo87

src/admin/populate.rs

@@ -1,7 +1,8 @@
 use crate::{db, schema::version_downloads};
 
-use crate::tasks::spawn_blocking;
 use diesel::prelude::*;
+use diesel_async::scoped_futures::ScopedFutureExt;
+use diesel_async::{AsyncConnection, AsyncPgConnection, RunQueryDsl};
 use rand::rngs::StdRng;
 use rand::{Rng, SeedableRng};
 
@@ -16,15 +17,13 @@ pub struct Opts {
 }
 
 pub async fn run(opts: Opts) -> anyhow::Result<()> {
-    spawn_blocking(move || {
-        let mut conn = db::oneoff_connection()?;
-        conn.transaction(|conn| update(opts, conn))?;
-        Ok(())
-    })
-    .await
+    let mut conn = db::oneoff_async_connection().await?;
+    conn.transaction(|conn| update(opts, conn).scope_boxed())
+        .await?;
+    Ok(())
 }
 
-fn update(opts: Opts, conn: &mut PgConnection) -> QueryResult<()> {
+async fn update(opts: Opts, conn: &mut AsyncPgConnection) -> QueryResult<()> {
     use diesel::dsl::*;
 
     for id in opts.version_ids {
@@ -40,7 +39,8 @@ fn update(opts: Opts, conn: &mut PgConnection) -> QueryResult<()> {
                     version_downloads::downloads.eq(dls),
                     version_downloads::date.eq(date(now - day.days())),
                 ))
-                .execute(conn)?;
+                .execute(conn)
+                .await?;
         }
     }
     Ok(())