compiles

src/bin/cratesfyi.rs

@@ -802,7 +802,11 @@ impl Context for BinContext {
         Ok(self
             .pool
             .get_or_try_init::<_, Error>(|| {
-                Ok(Pool::new(&*self.config()?, self.instance_metrics()?)?)
+                Ok(Pool::new(
+                    &*self.config()?,
+                    &*self.runtime()?,
+                    self.instance_metrics()?,
+                )?)
             })?
             .clone())
     }

src/db/pool.rs

@@ -5,6 +5,7 @@ use r2d2_postgres::PostgresConnectionManager;
 use sqlx::postgres::PgPoolOptions;
 use sqlx::Executor;
 use std::sync::Arc;
+use tokio::runtime::Runtime;
 use tracing::debug;
 
 pub type PoolClient = r2d2::PooledConnection<PostgresConnectionManager<NoTls>>;
@@ -24,24 +25,30 @@ pub struct Pool {
 }
 
 impl Pool {
-    pub fn new(config: &Config, metrics: Arc<InstanceMetrics>) -> Result<Pool, PoolError> {
+    pub fn new(
+        config: &Config,
+        runtime: &Runtime,
+        metrics: Arc<InstanceMetrics>,
+    ) -> Result<Pool, PoolError> {
         debug!(
             "creating database pool (if this hangs, consider running `docker-compose up -d db s3`)"
         );
-        Self::new_inner(config, metrics, DEFAULT_SCHEMA)
+        Self::new_inner(config, runtime, metrics, DEFAULT_SCHEMA)
     }
 
     #[cfg(test)]
     pub(crate) fn new_with_schema(
         config: &Config,
+        runtime: &Runtime,
         metrics: Arc<InstanceMetrics>,
         schema: &str,
     ) -> Result<Pool, PoolError> {
-        Self::new_inner(config, metrics, schema)
+        Self::new_inner(config, runtime, metrics, schema)
     }
 
     fn new_inner(
         config: &Config,
+        runtime: &Runtime,
         metrics: Arc<InstanceMetrics>,
         schema: &str,
     ) -> Result<Pool, PoolError> {
@@ -57,6 +64,7 @@ impl Pool {
             .build(manager)
             .map_err(PoolError::PoolCreationFailed)?;
 
+        let _guard = runtime.enter();
         let async_pool = PgPoolOptions::new()
             .max_connections(config.max_pool_size)
             .min_connections(config.min_pool_idle)