From 332e80efa7c3bec05ef51b0243c13aa2abe39e43 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Sat, 19 Mar 2022 16:14:13 +0100
Subject: select: Handle multiple records found

If no records were found, continue to return `Yaml::Null`. If a single
record was found, return a `Yaml::Hash`. Otherwise, return a
`Yaml::Array` containing all the records.
---
 src/select.rs | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

(limited to 'src')

diff --git a/src/select.rs b/src/select.rs
index 4f89244..c77f85b 100644
--- a/src/select.rs
+++ b/src/select.rs
@@ -65,16 +65,18 @@ pub fn select_by_column(
         },
     )?;
 
-    let mut row = None;
+    // Only one record is expected.
+    let mut records = yaml_rust::yaml::Array::with_capacity(1);
+
     for row_result in rows {
-        row = Some(yaml_rust::Yaml::Hash(row_result?));
+        records.push(yaml_rust::Yaml::Hash(row_result?));
     }
 
-    if let Some(r) = row {
-        return Ok(r);
+    match records.len() {
+        0 => Ok(yaml_rust::Yaml::Null),
+        1 => Ok(records.swap_remove(0)),
+        _ => Ok(yaml_rust::Yaml::Array(records)),
     }
-
-    Ok(yaml_rust::Yaml::Null)
 }
 
 
-- 
cgit v1.2.3