readit: pre-allocate enough elements

tembo-io · Jul 25, 2023 · 6517c29 · 6517c29
1 parent 6a30cd9
commit 6517c29
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -132,10 +132,13 @@ fn readit(
         TimestampWithTimeZone,
         pgrx::JsonB,
     )> = Vec::new();
+
     let _: Result<(), PgmqExtError> = Spi::connect(|mut client| {
         let query = read(queue_name, &vt, &limit)?;
-        let mut tup_table: SpiTupleTable = client.update(&query, None, None)?;
-        while let Some(row) = tup_table.next() {
+        let tup_table: SpiTupleTable = client.update(&query, None, None)?;
+        results.reserve_exact(tup_table.len());
+
+        for row in tup_table {
             let msg_id = row["msg_id"].value::<i64>()?.expect("no msg_id");
             let read_ct = row["read_ct"].value::<i32>()?.expect("no read_ct");
             let vt = row["vt"].value::<TimestampWithTimeZone>()?.expect("no vt");