schemerz-postgres 0.191.0

//! An adapter enabling use of the schemerz schema migration library with
//! PostgreSQL.
//!
//! # Examples:
//!
//! ```rust
//! extern crate postgres;
//! #[macro_use]
//! extern crate schemerz;
//! extern crate schemerz_postgres;
//! extern crate uuid;
//!
//! use std::collections::HashSet;
//!
//! use postgres::{Client, NoTls, Transaction};
//! use schemerz::{Migration, Migrator};
//! use schemerz_postgres::{PostgresAdapter, PostgresAdapterError, PostgresMigration};
//! use uuid::uuid;
//!
//! struct MyExampleMigration;
//! migration!(
//!     MyExampleMigration,
//!     uuid!("4885e8ab-dafa-4d76-a565-2dee8b04ef60"),
//!     [],
//!     "An example migration without dependencies.");
//!
//! impl PostgresMigration for MyExampleMigration {
//!     fn up(&self, transaction: &mut Transaction) -> Result<(), PostgresAdapterError> {
//!         transaction.execute("CREATE TABLE my_example (id integer PRIMARY KEY);", &[])?;
//!         Ok(())
//!     }
//!
//!     fn down(&self, transaction: &mut Transaction) -> Result<(), PostgresAdapterError> {
//!         transaction.execute("DROP TABLE my_example;", &[])?;
//!         Ok(())
//!     }
//! }
//!
//! fn main() {
//!     let mut conn = Client::connect(
//!         "postgresql://postgres@localhost",
//!         NoTls).unwrap();
//!     conn.execute("SET search_path = pg_temp", &[]).unwrap();
//!     let adapter = PostgresAdapter::new(&mut conn, None);
//!
//!     let mut migrator = Migrator::new(adapter);
//!
//!     let migration = Box::new(MyExampleMigration {});
//!     migrator.register(migration);
//!     migrator.up(None);
//! }
//! ```
#![warn(clippy::all)]
#![forbid(unsafe_code)]

use std::collections::HashSet;

use postgres::{Client, Error as PostgresError, Transaction};
use uuid::Uuid;

use schemerz::{Adapter, Migration};

/// PostgreSQL-specific trait for schema migrations.
pub trait PostgresMigration: Migration<Uuid> {
    /// Apply a migration to the database using a transaction.
    fn up(&self, _transaction: &mut Transaction<'_>) -> Result<(), PostgresError> {
        Ok(())
    }

    /// Revert a migration to the database using a transaction.
    fn down(&self, _transaction: &mut Transaction<'_>) -> Result<(), PostgresError> {
        Ok(())
    }
}

pub type PostgresAdapterError = PostgresError;

/// Adapter between schemerz and PostgreSQL.
pub struct PostgresAdapter<'a> {
    conn: &'a mut Client,
    migration_metadata_table: String,
}

impl<'a> PostgresAdapter<'a> {
    /// Construct a PostgreSQL schemerz adapter.
    ///
    /// `table_name` specifies the name of the table that schemerz will use
    /// for storing metadata about applied migrations. If `None`, a default
    /// will be used.
    ///
    /// ```rust
    /// # extern crate postgres;
    /// # extern crate schemerz_postgres;
    /// #
    /// # fn main() {
    /// let mut conn = postgres::Client::connect(
    ///     "postgresql://postgres@localhost",
    ///     postgres::NoTls).unwrap();
    /// let adapter = schemerz_postgres::PostgresAdapter::new(&mut conn, None);
    /// # }
    /// ```
    pub fn new(conn: &'a mut Client, table_name: Option<String>) -> PostgresAdapter<'a> {
        PostgresAdapter {
            conn,
            migration_metadata_table: table_name.unwrap_or_else(|| "_schemerz".into()),
        }
    }

    /// Initialize the schemerz metadata schema. This must be called before
    /// using `Migrator` with this adapter. This is safe to call multiple times.
    pub fn init(&mut self) -> Result<(), PostgresError> {
        self.conn.execute(
            format!(
                r#"
                    CREATE TABLE IF NOT EXISTS {} (
                        id uuid PRIMARY KEY
                    ) WITH (
                        OIDS=FALSE
                    )
                "#,
                self.migration_metadata_table
            )
            .as_str(),
            &[],
        )?;
        Ok(())
    }
}

impl<'a> Adapter<Uuid> for PostgresAdapter<'a> {
    type MigrationType = Box<dyn PostgresMigration>;

    type Error = PostgresAdapterError;

    fn applied_migrations(&mut self) -> Result<HashSet<Uuid>, Self::Error> {
        let rows = self.conn.query(
            format!("SELECT id FROM {};", self.migration_metadata_table).as_str(),
            &[],
        )?;
        Ok(rows.iter().map(|row| row.get(0)).collect())
    }

    fn apply_migration(&mut self, migration: &Self::MigrationType) -> Result<(), Self::Error> {
        let mut trans = self.conn.transaction()?;
        migration.up(&mut trans)?;
        trans.execute(
            format!(
                "INSERT INTO {} (id) VALUES ($1::uuid);",
                self.migration_metadata_table
            )
            .as_str(),
            &[&migration.id()],
        )?;
        trans.commit()
    }

    fn revert_migration(&mut self, migration: &Self::MigrationType) -> Result<(), Self::Error> {
        let mut trans = self.conn.transaction()?;
        migration.down(&mut trans)?;
        trans.execute(
            format!(
                "DELETE FROM {} WHERE id = $1::uuid;",
                self.migration_metadata_table
            )
            .as_str(),
            &[&migration.id()],
        )?;
        trans.commit()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use postgres::NoTls;
    use schemerz::test_schemerz_adapter;
    use schemerz::testing::*;

    impl PostgresMigration for TestMigration<Uuid> {}

    impl<'a> TestAdapter<Uuid> for PostgresAdapter<'a> {
        fn mock(id: Uuid, dependencies: HashSet<Uuid>) -> Self::MigrationType {
            Box::new(TestMigration::new(id, dependencies))
        }
    }

    fn build_test_connection() -> Client {
        let mut client = Client::connect("postgresql://postgres@localhost", NoTls).unwrap();
        client.execute("SET search_path = pg_temp", &[]).unwrap();
        client
    }

    fn build_test_adapter(conn: &mut Client) -> PostgresAdapter<'_> {
        let mut adapter = PostgresAdapter::new(conn, None);
        adapter.init().unwrap();
        adapter
    }

    fn uuid_iter() -> impl Iterator<Item = Uuid> {
        (0..).map(|v| Uuid::from_fields(v as u32, v, v, &[0; 8]))
    }

    test_schemerz_adapter!(
        let mut conn = build_test_connection(),
        build_test_adapter(&mut conn),
        uuid_iter(),
    );
}