valis-core 0.1.9

A library for workflow management
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use std::fmt::Error;

use rusqlite::{Connection, Row};

pub mod serializers;

pub trait DatabaseOperations<T> {
    fn save(&self, db: &str) -> Result<(), Error>;
    fn get(id: T, db: &str) -> Result<Self, rusqlite::Error> where Self: Sized;
    fn get_all(db: &str) -> Result<Vec<Self>, rusqlite::Error>
        where
            Self: Sized;
    fn map(row: &Row<'_>) -> Result<Self, rusqlite::Error> where Self: Sized;
}

fn get_sql_schema() -> Vec<String> {
    include_str!("tables.sql").split("---").map(|s| s.to_string()).collect::<Vec<String>>()
}

pub fn init_db(db: &str) -> Result<(), rusqlite::Error> {
    let conn = Connection::open(db)?;

    get_sql_schema().into_iter().for_each(|sql| {
        conn.execute(&sql, []).ok().unwrap();
    });

    Ok(())
}

pub fn get_connection(db: &str) -> Connection {
    match Connection::open(db) {
        Ok(conn) => conn,
        Err(e) => {
            println!("Failed to open database: {}", e);
            std::process::exit(1);
        }
    }
}