add: api to get migrations

geofmureithi · Oct 5, 2023 · 8a7ba8f · 8a7ba8f
1 parent 1e0fb25
commit 8a7ba8f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 4 deletions.
diff --git a/packages/apalis-sql/src/mysql.rs b/packages/apalis-sql/src/mysql.rs
@@ -48,11 +48,17 @@ impl<T> MysqlStorage<T> {
         Ok(Self::new(pool))
     }
 
-    /// Setup
+    /// Get mysql migrations without running them
+    #[cfg(feature = "migrate")]
+    pub fn migrations() -> sqlx::migrate::Migrator {
+        sqlx::migrate!("migrations/mysql")
+    }
+
+    /// Do migrations for mysql
     #[cfg(feature = "migrate")]
     pub async fn setup(&self) -> Result<(), sqlx::Error> {
         let pool = self.pool.clone();
-        sqlx::migrate!("migrations/mysql").run(&pool).await?;
+        Self::migrations().run(&pool).await?;
         Ok(())
     }
 

diff --git a/packages/apalis-sql/src/postgres.rs b/packages/apalis-sql/src/postgres.rs
@@ -62,11 +62,18 @@ impl<T> PostgresStorage<T> {
         Ok(Self::new(pool))
     }
 
+    /// Get postgres migrations without running them
+    #[cfg(feature = "migrate")]
+    pub fn migrations() -> sqlx::migrate::Migrator {
+        sqlx::migrate!("migrations/postgres")
+    }
+
+
     /// Do migrations for Postgres
     #[cfg(feature = "migrate")]
     pub async fn setup(&self) -> Result<(), sqlx::Error> {
         let pool = self.pool.clone();
-        sqlx::migrate!("migrations/postgres").run(&pool).await?;
+        Self::migrations().run(&pool).await?;
         Ok(())
     }
 

diff --git a/packages/apalis-sql/src/sqlite.rs b/packages/apalis-sql/src/sqlite.rs
@@ -63,10 +63,17 @@ impl<T: Job> SqliteStorage<T> {
         sqlx::query("PRAGMA cache_size = 64000;")
             .execute(&pool)
             .await?;
-        sqlx::migrate!("migrations/sqlite").run(&pool).await?;
+        Self::migrations().run(&pool).await?;
         Ok(())
     }
 
+    /// Get sqlite migrations without running them
+    #[cfg(feature = "migrate")]
+    pub fn migrations() -> sqlx::migrate::Migrator {
+        sqlx::migrate!("migrations/sqlite")
+    }
+
+
     /// Keeps a storage notified that the worker is still alive manually
     pub async fn keep_alive_at<Service>(
         &mut self,