Alex Mikhalev
7e24b03f22
Change SecRun to have a RunState enum with a cancelled state instead of removing run from the queue when cancelled. This means we no longer need `drain_filter` which is unstable
52 lines
1.2 KiB
Rust
52 lines
1.2 KiB
Rust
use color_eyre::eyre::Result;
|
|
use rusqlite::Connection as DbConnection;
|
|
use rusqlite::NO_PARAMS;
|
|
use tracing::info;
|
|
use tracing_subscriber::EnvFilter;
|
|
|
|
mod db;
|
|
mod migrations;
|
|
mod model;
|
|
mod section_interface;
|
|
mod section_runner;
|
|
#[cfg(test)]
|
|
mod trace_listeners;
|
|
|
|
use model::Section;
|
|
|
|
fn setup_db() -> Result<DbConnection> {
|
|
// let conn = DbConnection::open_in_memory()?;
|
|
let mut conn = DbConnection::open("test.db")?;
|
|
|
|
let migs = db::create_migrations();
|
|
migs.apply(&mut conn)?;
|
|
|
|
Ok(conn)
|
|
}
|
|
|
|
fn query_sections(conn: &DbConnection) -> Result<Vec<Section>> {
|
|
let mut statement = conn.prepare_cached("SELECT id, name, interface_id FROM sections;")?;
|
|
let rows = statement.query_map(NO_PARAMS, Section::from_sql)?;
|
|
Ok(rows.collect::<Result<Vec<_>, _>>()?)
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
tracing_subscriber::fmt()
|
|
.with_ansi(true)
|
|
.with_env_filter(
|
|
EnvFilter::try_from_default_env()
|
|
.unwrap_or_else(|_| EnvFilter::new("info")),
|
|
)
|
|
.init();
|
|
color_eyre::install()?;
|
|
|
|
let conn = setup_db()?;
|
|
|
|
let sections = query_sections(&conn)?;
|
|
for sec in sections {
|
|
info!(section = debug(&sec), "read section");
|
|
}
|
|
|
|
Ok(())
|
|
}
|