Patch dollar quote for jobs (#102)

* dollar quote jobs * bump ver * fix test * change ci path
tembo-io · Apr 18, 2024 · 3ac2430 · 3ac2430
1 parent b9e8de2
commit 3ac2430
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 10 deletions.
diff --git a/.github/workflows/extension_upgrade.yml b/.github/workflows/extension_upgrade.yml
@@ -11,13 +11,13 @@ on:
       - main
     paths:
       - ".github/workflows/extension_upgrade.yml"
-      - "./extension/**"
+      - "extension/**"
   push:
     branches:
       - main
     paths:
       - ".github/workflows/extension_upgrade.yml"
-      - "./extension/**"
+      - "extension/**"
 
 jobs:
   test:

diff --git a/extension/Cargo.toml b/extension/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "vectorize"
-version = "0.14.1"
+version = "0.14.2"
 edition = "2021"
 publish = false
 

diff --git a/extension/Trunk.toml b/extension/Trunk.toml
@@ -6,7 +6,7 @@ description = "The simplest way to orchestrate vector search on Postgres."
 homepage = "https://github.com/tembo-io/pg_vectorize"
 documentation = "https://github.com/tembo-io/pg_vectorize"
 categories = ["orchestration", "machine_learning"]
-version = "0.14.1"
+version = "0.14.2"
 
 [build]
 postgres_version = "15"

diff --git a/extension/sql/vectorize--0.14.1--0.14.2.sql b/extension/sql/vectorize--0.14.1--0.14.2.sql
diff --git a/extension/src/job.rs b/extension/src/job.rs
@@ -47,7 +47,7 @@ fn _handle_table_update(job_name: &str, record_ids: Vec<String>, inputs: Vec<Str
 
     // send the job message to the queue
     let query = format!(
-        "select pgmq.send('{VECTORIZE_QUEUE}', '{}');",
+        "select pgmq.send('{VECTORIZE_QUEUE}', $${}$$::jsonb);",
         serde_json::to_string(&job_message).unwrap()
     );
     let _ran: Result<_, spi::Error> = Spi::connect(|mut c| {
@@ -169,10 +169,8 @@ pub fn initalize_table_job(
             inputs: b,
         };
         let query = format!(
-            "select pgmq.send('{VECTORIZE_QUEUE}', '{}');",
-            serde_json::to_string(&job_message)
-                .unwrap()
-                .replace('\'', "''")
+            "select pgmq.send('{VECTORIZE_QUEUE}', $${}$$::jsonb);",
+            serde_json::to_string(&job_message).unwrap()
         );
         let _ran: Result<_, spi::Error> = Spi::connect(|mut c| {
             let _r = c.update(&query, None, None)?;

diff --git a/extension/tests/integration_tests.rs b/extension/tests/integration_tests.rs
@@ -167,7 +167,7 @@ async fn test_realtime_job() {
 
     let insert_query = format!(
         "INSERT INTO \"{test_table_name}\"(product_id, product_name, description)
-        VALUES ({random_product_id}, 'car tester', 'a product for testing cars');"
+        VALUES ({random_product_id}, 'car tester', $$a product for testing car's components$$);"
     );
 
     // insert a new row