git.iko.soy/reshape
on main help sign in
reshape/src/helpers.rs
29 lines 850 B history raw
1use anyhow::Context;
3use crate::db::Conn;
5pub fn set_up_helpers(db: &mut dyn Conn, target_migration: &str) -> anyhow::Result<()> {
6 let query = format!(
7 "
8 CREATE OR REPLACE FUNCTION reshape.is_new_schema()
9 RETURNS BOOLEAN AS $$
10 DECLARE
11 setting TEXT := current_setting('reshape.is_new_schema', TRUE);
12 setting_bool BOOLEAN := setting IS NOT NULL AND setting = 'YES';
13 BEGIN
14 RETURN current_setting('search_path') = 'migration_{}' OR setting_bool;
15 END
16 $$ language 'plpgsql';
17 ",
18 target_migration,
19 );
20 db.query(&query)
21 .context("failed creating helper function reshape.is_new_schema()")?;
23 Ok(())
26pub fn tear_down_helpers(db: &mut dyn Conn) -> anyhow::Result<()> {
27 db.query("DROP FUNCTION IF EXISTS reshape.is_new_schema;")?;
28 Ok(())