select(): Parameterize primary key column name

1 files 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
                 }