tern-core 3.1.6

//! [`Executor`] for [`sqlx::PgPool`][pg-pool].
//!
//! [`Executor`]: crate::migration::Executor
//! [pg-pool]: https://docs.rs/sqlx/0.8.3/sqlx/type.PgPool.html
use sqlx::Postgres;

use super::pool::SqlxExecutor;
use crate::migration::{AppliedMigration, Query, QueryRepository};

/// Specialization of `SqlxExecutor` to `sqlx::PgPool`.
pub type SqlxPgExecutor = SqlxExecutor<Postgres, SqlxPgQueryRepo>;

/// The schema history table queries for postgres.
#[derive(Debug, Clone)]
pub struct SqlxPgQueryRepo;

impl QueryRepository for SqlxPgQueryRepo {
    fn create_history_if_not_exists_query(history_table: &str) -> Query {
        let sql = format!(
            "
CREATE TABLE IF NOT EXISTS {history_table}(
  version bigint PRIMARY KEY,
  description text NOT NULL,
  content text NOT NULL,
  duration_ms bigint NOT NULL,
  applied_at timestamptz NOT NULL DEFAULT now()
);
"
        );

        Query::new(sql)
    }

    fn drop_history_query(history_table: &str) -> Query {
        let sql = format!("DROP TABLE IF EXISTS {history_table};");

        Query::new(sql)
    }

    fn insert_into_history_query(
        history_table: &str,
        _: &AppliedMigration,
    ) -> Query {
        // With `sqlx` we're not going to use the `AppliedMigration`, the values
        // will get in the query by `bind`ing them.
        let sql = format!(
            "
INSERT INTO {history_table}(version, description, content, duration_ms, applied_at)
  VALUES ($1, $2, $3, $4, $5);
"
        );

        Query::new(sql)
    }

    fn select_star_from_history_query(history_table: &str) -> Query {
        let sql = format!(
            "
SELECT
  version,
  description,
  content,
  duration_ms,
  applied_at
FROM
  {history_table}
ORDER BY
  version;
"
        );

        Query::new(sql)
    }

    fn upsert_history_query(
        history_table: &str,
        _: &AppliedMigration,
    ) -> Query {
        let sql = format!(
            "
INSERT INTO {history_table}(version, description, content, duration_ms, applied_at)
  VALUES ($1, $2, $3, $4, $5)
  ON CONFLICT (version) DO UPDATE
  SET
    description = excluded.description,
    content = excluded.content,
    duration_ms = excluded.duration_ms,
    applied_at = excluded.applied_at;
"
        );

        Query::new(sql)
    }
}