From 5c3880c78c096c7ee34c790eedd65a40af143fff Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Wed, 9 Mar 2022 00:05:34 +0100
Subject: get_column_names(): Put column names in a `HashMap`

Make it easier to find out if a column name exists in the table.
---
 src/main.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index 2709005..cfdf0ab 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -46,8 +46,13 @@ fn yaml_extract(doc: &yaml::Yaml) {
     }
 }
 
-fn get_column_names(dbconn: &rusqlite::Connection) -> Vec<String> {
-    let mut column_names = Vec::new();
+#[derive(Debug)]
+struct Zero {}
+
+use std::collections::HashMap;
+
+fn get_column_names(dbconn: &rusqlite::Connection) -> HashMap<String, Zero> {
+    let mut column_names = HashMap::new();
 
     let mut stmt = dbconn.prepare(r#"
         SELECT "name"
@@ -60,7 +65,7 @@ fn get_column_names(dbconn: &rusqlite::Connection) -> Vec<String> {
     ).unwrap();
 
     for row in rows {
-        column_names.push(row.unwrap());
+        column_names.insert(row.unwrap(), Zero{});
     }
 
     column_names
-- 
cgit v1.2.3