feat(ext): add is_partitioned and is_unlogged to list_queues (#128)

* feat(ext): add is_partitioned and is_unlogged to list_queues

* format

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "pgmq"
-version = "0.28.0"
+version = "0.29.0"
 edition = "2021"
 authors = ["Tembo.io"]
 description = "Postgres extension for PGMQ"

sql/pgmq--0.28.0--0.29.0.sql

@@ -0,0 +1,13 @@
+DROP FUNCTION pgmq.list_queues();
+
+-- Copy-pasted from schema
+-- pgmq::api::list_queues
+CREATE  FUNCTION pgmq."list_queues"() RETURNS TABLE (
+	"queue_name" TEXT,  /* alloc::string::String */
+	"created_at" timestamp with time zone,  /* pgrx::datum::time_stamp_with_timezone::TimestampWithTimeZone */
+	"is_partitioned" bool,  /* bool */
+	"is_unlogged" bool  /* bool */
+)
+STRICT
+LANGUAGE c /* Rust */
+AS 'MODULE_PATHNAME', 'pgmq_list_queues_wrapper';

src/api.rs

@@ -49,6 +49,8 @@ fn pgmq_list_queues() -> Result<
         (
             name!(queue_name, String),
             name!(created_at, TimestampWithTimeZone),
+            name!(is_partitioned, bool),
+            name!(is_unlogged, bool),
         ),
     >,
     spi::Error,
@@ -57,8 +59,8 @@ fn pgmq_list_queues() -> Result<
     Ok(TableIterator::new(results))
 }
 
-pub fn listit() -> Result<Vec<(String, TimestampWithTimeZone)>, spi::Error> {
-    let mut results: Vec<(String, TimestampWithTimeZone)> = Vec::new();
+pub fn listit() -> Result<Vec<(String, TimestampWithTimeZone, bool, bool)>, spi::Error> {
+    let mut results: Vec<(String, TimestampWithTimeZone, bool, bool)> = Vec::new();
     let query = format!("SELECT * FROM {PGMQ_SCHEMA}.meta");
     let _: Result<(), spi::Error> = Spi::connect(|client| {
         let tup_table: SpiTupleTable = client.select(&query, None, None)?;
@@ -67,7 +69,11 @@ pub fn listit() -> Result<Vec<(String, TimestampWithTimeZone)>, spi::Error> {
             let created_at = row["created_at"]
                 .value::<TimestampWithTimeZone>()?
                 .expect("no created_at");
-            results.push((queue_name, created_at));
+            let is_partitioned = row["is_partitioned"]
+                .value::<bool>()?
+                .expect("no is_partitioned");
+            let is_unlogged = row["is_unlogged"].value::<bool>()?.expect("no is_unlogged");
+            results.push((queue_name, created_at, is_partitioned, is_unlogged));
         }
         Ok(())
     });