Migrator for writing sqlx migration using Rust instead of SQL
| License | Crates Version | Docs |
| :---: | :---: | :---: |
| |
|
|
Supported Databases: - [X] PostgreSQL - [X] SQLite
To use postgres migrator along with tokio-native-tls runtime you can configure Cargo.toml
toml
sqlx_migrator = { version = "0.2.1", features=["postgres", "runtime-tokio-native-tls"]
To use sqlxmigrator first you need to implement Operation trait to write your sqlx operation ```rust use sqlxmigrator::error::Error; use sqlxmigrator::operation::Operation; //Its better to use sqlx imported from sqlxmigrator use sqlx_migrator::sqlx;
pub(crate) struct FirstOperation;
impl Operation for FirstOperation { type Database = sqlx::Postgres;
// Up function runs apply migration
async fn up(
&self,
connection: &mut <Self::Database as sqlx::Database>::Connection,
) -> Result<(), Error> {
sqlx::query("CREATE TABLE sample (id INTEGER PRIMARY KEY, name TEXT)")
.execute(connection)
.await?;
Ok(())
}
// down migration runs down migration
async fn down(
&self,
connection: &mut <Self::Database as sqlx::Database>::Connection,
) -> Result<(), Error> {
sqlx::query("DROP TABLE sample").execute(connection).await?;
Ok(())
}
} ```
After creation of operation you can implement Migration struct to create single migration
```rust use sqlxmigrator::error::Error; use sqlxmigrator::migration::Migration; use sqlxmigrator::operation::Operation; use sqlxmigrator::sqlx;
pub(crate) struct FirstMigration;
impl Migration for FirstMigration { type Database = Postgres;
// app where migration lies can be any value
fn app(&self) -> &str {
"main"
}
// name of migration
// Combination of migration app and name must be unique to work properly
fn name(&self) -> &str {
"first_migration"
}
// use operations function to add operation part of migration
fn operations(&self) -> Vec<Box<dyn Operation<Database = Self::Database>>> {
vec![Box::new(FirstOperation)]
}
// Migration trait also have multiple other function see docs for usage
} ```
Now at last you need to create migrator for your database to run migrations ```rust use sqlxmigrator::migrator::Migrator as MigratorTrait; use sqlxmigrator::sqlite::migrator::Migrator;
let uri = std::env::var("DATABASEURL").unwrap(); let pool = sqlx::Pool::connect(&uri).await.unwrap(); let mut migrator = Migrator::new(&pool); migrator.addmigration(Box::new(FirstMigration)); ```
Now you can use two ways to run migrator either directly running migration or creating cli from migrator
For directly run
rust
// use apply all to apply all pending migration
migrator.apply_all().await.unwrap();
// or use revert all to revert all applied migrations
migrator.revert_all().await.unwrap();
Or you can create cli
rust
sqlx_migrator::cli::run(Box::new(migrator)).await.unwrap();
To migrate from sqlx sql based migration you have two alternative: 1. Write all sql migration as rust operation 2. Write single rust based operation which apply and revert all sqlx sql based migration
Can be easily applied by following above usage docs where you only need to write your sql based migration as sqlx query
Then you can create cli for migrator
rust
sqlx_migrator::cli::run(Box::new(migrator)).await.unwrap();
and run fake apply cli command
<COMMAND_NAME> apply --fake
which actually doesn't apply migration query but only update migration table
To run all sqlx sql based migration as single command create new operation ```rust use sqlxmigrator::error::Error; use sqlxmigrator::operation::Operation; use sqlx_migrator::sqlx;
pub(crate) struct SqlxOperation;
impl Operation for SqlxOperation { type Database = sqlx::Postgres;
async fn up(
&self,
connection: &mut <Self::Database as sqlx::Database>::Connection,
) -> Result<(), Error> {
sqlx::migrate!("db/migrations")
.run(connection)
.await?;
Ok(())
}
async fn down(
&self,
connection: &mut <Self::Database as sqlx::Database>::Connection,
) -> Result<(), Error> {
sqlx::migrate!("db/migrations")
.undo(connection, 0)
.await?;
Ok(())
}
} ```