Faster query with a simpler plan by joining with the CTE instead of m…

…aking IN query (#113) * Faster query with a simpler plan by joining with the CTE instead of making IN query * Update cargo version
tembo-io · Sep 13, 2023 · 7ce03aa · 7ce03aa
1 parent 0ffb034
commit 7ce03aa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/core/Cargo.toml b/core/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pgmq-core"
-version = "0.4.0"
+version = "0.4.1"
 edition = "2021"
 authors = ["Tembo.io"]
 description = "Core functionality shared between the PGMQ Rust SDK and Postgres Extension"

diff --git a/core/src/query.rs b/core/src/query.rs
@@ -219,11 +219,12 @@ pub fn read(name: &str, vt: i32, limit: i32) -> Result<String, PgmqError> {
  LIMIT {limit}
  FOR UPDATE SKIP LOCKED
  )
- UPDATE {PGMQ_SCHEMA}.{TABLE_PREFIX}_{name}
+ UPDATE {PGMQ_SCHEMA}.{TABLE_PREFIX}_{name} t
  SET
  vt = clock_timestamp() + interval '{vt} seconds',
  read_ct = read_ct + 1
- WHERE msg_id in (select msg_id from cte)
+ FROM cte
+ WHERE t.msg_id=cte.msg_id
  RETURNING *;
  "
  ))