Conditionally create schema when creating queue on extensionless

tembo-io · Sep 13, 2023 · 92f418e · 92f418e
1 parent aaa4a5d
commit 92f418e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/core/src/query.rs b/core/src/query.rs
@@ -63,6 +63,10 @@ pub fn create_archive(name: CheckedName<'_>) -> Result<String, PgmqError> {
     ))
 }
 
+pub fn create_schema() -> String {
+    format!("CREATE SCHEMA IF NOT EXISTS {PGMQ_SCHEMA}")
+}
+
 pub fn create_meta() -> String {
     format!(
         "

diff --git a/pgmq-rs/src/query.rs b/pgmq-rs/src/query.rs
@@ -5,6 +5,7 @@ use pgmq_core::{errors, query, util};
 pub fn init_queue_client_only(name: &str) -> Result<Vec<String>, errors::PgmqError> {
     let name = util::CheckedName::new(name)?;
     Ok(vec![
+        query::create_schema(),
         query::create_meta(),
         query::create_queue(name)?,
         query::create_index(name)?,
@@ -19,6 +20,7 @@ pub fn init_queue_client_only(name: &str) -> Result<Vec<String>, errors::PgmqErr
 pub fn destroy_queue_client_only(name: &str) -> Result<Vec<String>, errors::PgmqError> {
     let name = util::CheckedName::new(name)?;
     Ok(vec![
+        query::create_schema(),
         query::drop_queue(name)?,
         query::drop_queue_archive(name)?,
         query::delete_queue_metadata(name)?,