diff options
| -rw-r--r-- | src/main.rs | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs index f0c11a2..ce18f49 100644 --- a/src/main.rs +++ b/src/main.rs @@ -6,6 +6,11 @@ fn main() { // Get column names from SQLite + let mut dbconn = rusqlite::Connection::open("./test.db").unwrap(); + + let table_columns = get_column_names(&mut dbconn); + dbg!(table_columns); + let text_data = std::fs::read_to_string("test.yml").unwrap(); let yaml_data = yaml::YamlLoader::load_from_str(&text_data).unwrap(); @@ -15,6 +20,8 @@ fn main() { } dbg!(yaml_data); + + dbconn.close().unwrap(); } fn yaml_extract(doc: &yaml::Yaml) { @@ -38,3 +45,32 @@ fn yaml_extract(doc: &yaml::Yaml) { _ => {} } } + +fn get_column_names(dbconn: &mut rusqlite::Connection) -> Vec<String> { + let mut column_names = Vec::new(); + + // let tx = dbconn.transaction().unwrap(); + + // let mut stmt = tx.prepare(r#" + // SELECT "name" + // FROM pragma_table_info("test"); + // "#).unwrap(); + + let mut stmt = dbconn.prepare(r#" + SELECT "name" + FROM pragma_table_info("test"); + "#).unwrap(); + + let rows = stmt.query_map( + [], + |row| row.get(0), + ).unwrap(); + + for row in rows { + column_names.push(row.unwrap()); + } + + // tx.commit().unwrap(); + + column_names +} |