Skip to main content

tern_core/executor/sqlx_backend/
sqlite.rs

1//! [`Executor`] for [`sqlx::SqlitePool`][sqlite-pool].
2//!
3//! [`Executor`]: crate::migration::Executor
4//! [sqlite-pool]: https://docs.rs/sqlx/0.8.3/sqlx/type.SqlitePool.html
5use sqlx::Sqlite;
6
7use super::pool::SqlxExecutor;
8use crate::migration::{AppliedMigration, Query, QueryRepository};
9
10/// Specialization of `SqlxExecutor` to `sqlx::SqlitePool`.
11pub type SqlxSqliteExecutor = SqlxExecutor<Sqlite, SqlxSqliteQueryRepo>;
12
13/// The schema history table queries for postgres.
14#[derive(Debug, Clone)]
15pub struct SqlxSqliteQueryRepo;
16
17impl QueryRepository for SqlxSqliteQueryRepo {
18    fn create_history_if_not_exists_query(history_table: &str) -> Query {
19        let sql = format!(
20            "
21CREATE TABLE IF NOT EXISTS {history_table}(
22  version bigint PRIMARY KEY,
23  description text NOT NULL,
24  content text NOT NULL,
25  duration_ms bigint NOT NULL,
26  applied_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP
27);
28"
29        );
30
31        Query::new(sql)
32    }
33
34    fn drop_history_query(history_table: &str) -> Query {
35        let sql = format!("DROP TABLE IF EXISTS {history_table};");
36
37        Query::new(sql)
38    }
39
40    fn insert_into_history_query(
41        history_table: &str,
42        _: &AppliedMigration,
43    ) -> Query {
44        // With `sqlx` we're not going to use the `AppliedMigration`, the values
45        // will get in the query by `bind`ing them.
46        let sql = format!(
47            "
48INSERT INTO {history_table}(version, description, content, duration_ms, applied_at)
49  VALUES (?1, ?2, ?3, ?4, ?5);
50"
51        );
52
53        Query::new(sql)
54    }
55
56    fn select_star_from_history_query(history_table: &str) -> Query {
57        let sql = format!(
58            "
59SELECT
60  version,
61  description,
62  content,
63  duration_ms,
64  applied_at
65FROM
66  {history_table}
67ORDER BY
68  version;
69"
70        );
71
72        Query::new(sql)
73    }
74
75    fn upsert_history_query(
76        history_table: &str,
77        _: &AppliedMigration,
78    ) -> Query {
79        let sql = format!(
80            "
81INSERT INTO {history_table}(version, description, content, duration_ms, applied_at)
82  VALUES (?1, ?2, ?3, ?4, ?5)
83  ON CONFLICT REPLACE;
84"
85        );
86
87        Query::new(sql)
88    }
89}