main: If `primary_key` option is given, select by given column name

1 files changed, 14 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index d3c286b..6c87a0b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -79,7 +79,20 @@ fn main() {
         } => {
             let dbconn = rusqlite::Connection::open(database).unwrap();
 
-            yaqlite::select(&dbconn, &table_name, &record_id).unwrap();
+            match primary_key {
+                Some(pk) => yaqlite::select_by_column(
+                    &dbconn,
+                    &table_name,
+                    &pk,
+                    &record_id,
+                ).unwrap(),
+
+                None => yaqlite::select(
+                    &dbconn,
+                    &table_name,
+                    &record_id,
+                ).unwrap(),
+            };
 
             dbconn.close().unwrap();
         },