Skip to content

Commit

Permalink
remove erroneous semicolon
Browse files Browse the repository at this point in the history
  • Loading branch information
ChuckHend committed Aug 21, 2023
1 parent c18e721 commit 4fa8181
Showing 1 changed file with 12 additions and 2 deletions.
14 changes: 12 additions & 2 deletions core/src/query.rs
Original file line number Diff line number Diff line change
Expand Up @@ -336,7 +336,7 @@ pub fn assign_queue(name: CheckedName<'_>) -> Result<String, PgmqError> {
}

pub fn assign_archive(name: CheckedName<'_>) -> Result<String, PgmqError> {
Ok(assign(&format!("{name}_archive; ")))
Ok(assign(&format!("{name}_archive")))
}

pub fn unassign_queue(name: CheckedName<'_>) -> Result<String, PgmqError> {
Expand All @@ -362,7 +362,11 @@ pub fn assign(table_name: &str) -> String {
SELECT 1
FROM pg_depend
WHERE refobjid = (SELECT oid FROM pg_extension WHERE extname = 'pgmq')
AND objid = (SELECT oid FROM pg_class WHERE relname = '{TABLE_PREFIX}_{table_name}')
AND objid = (
SELECT oid
FROM pg_class
WHERE relname = '{TABLE_PREFIX}_{table_name}'
)
) THEN
EXECUTE 'ALTER EXTENSION pgmq ADD TABLE {PGMQ_SCHEMA}.{TABLE_PREFIX}_{table_name}';
END IF;
Expand Down Expand Up @@ -402,6 +406,12 @@ pub fn check_input(input: &str) -> Result<(), PgmqError> {
mod tests {
use super::*;

#[test]
fn test_assign() {
let query = assign("my_queue_archive");
assert!(query.contains("WHERE relname = 'pgmq_my_queue_archive'"));
}

#[test]
fn test_create() {
let queue_name = CheckedName::new("yolo").unwrap();
Expand Down

0 comments on commit 4fa8181

Please sign in to comment.