diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 15 |
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(); }, |