From 442539ccf844e92ce432c5920fc7c254997a6acb Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Sat, 19 Mar 2022 15:21:50 +0100
Subject: select(): Parameterize primary key column name

---
 src/select.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/select.rs b/src/select.rs
index 71a096e..86898bb 100644
--- a/src/select.rs
+++ b/src/select.rs
@@ -39,7 +39,7 @@ pub fn select_by_column(
                 WHERE "{}" = :pk;
             "#,
             table_name,
-            "id",
+            primary_key_column,
         ),
     )?;
 
@@ -72,7 +72,7 @@ pub fn select_by_column(
             let mut data = yaml_rust::yaml::Hash::new();
 
             for (i, column) in column_names.iter().enumerate() {
-                if column == "id" {
+                if column == primary_key_column {
                     continue
                 }
 
-- 
cgit v1.2.3