rusqlite-struct 0.1.1

Crate to make working with structs easier in rusqlite
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

use rusqlite::{Connection, Result};
use serde_rusqlite::{from_rows, to_params_named};

use crate::RusqliteStruct;

/// The name of the struct must be the same as the one used in the DB table.
pub trait RusqliteStructHelper {
    fn insert_into_table_struct<T: RusqliteStruct + serde::Serialize>(&self, from: &T) -> Result<()>;
    fn get_from_table_struct<T: RusqliteStruct + for<'a> serde::Deserialize<'a>>(&self) -> Result<Vec<T>>;
}

impl RusqliteStructHelper for Connection {
    fn insert_into_table_struct<T: RusqliteStruct + serde::Serialize>(&self, from: &T) -> Result<()> {
        let to_execute = format!(
            "INSERT INTO {} ({}) VALUES ({})",
            T::struct_name(),
            format!("{}", T::field_names().join(",")),
            format!("{}", T::field_names().iter().map(|s| format!(":{}", s)).collect::<Vec<String>>().join(","))
        );

        self.execute(&to_execute, to_params_named(from).unwrap().to_slice().as_slice())?;

        Ok(())
    }

    fn get_from_table_struct<T: RusqliteStruct + for<'a> serde::Deserialize<'a>>(&self) -> Result<Vec<T>> {
        let mut stmt = self.prepare(
            format!("SELECT {} FROM {}", T::field_names().join(","), T::struct_name()).as_str()
        )?;
        let rows = stmt.query([])?;

        from_rows::<T>(rows)
            .map(|r| r.map_err(|e| rusqlite::Error::ToSqlConversionFailure(Box::new(e))))
            .collect::<rusqlite::Result<Vec<_>>>()
    }
}