use log::info; use rand; use rusqlite; use std::sync::mpsc; use std::time; #[derive(Debug)] pub struct Post { id: u32, title: String, author: String, body: String, } #[derive(Debug)] pub struct Conn { db: rusqlite::Connection, rx: mpsc::Receiver, } #[derive(Debug)] pub enum Cmd { Create, Update, Disconnect, NOOP, } impl Conn { fn init() -> rusqlite::Connection { let start = time::Instant::now(); info!("Connecting to database"); let conn = rusqlite::Connection::open_with_flags( "/tmp/db.sql", rusqlite::OpenFlags::SQLITE_OPEN_FULL_MUTEX | rusqlite::OpenFlags::SQLITE_OPEN_CREATE | rusqlite::OpenFlags::SQLITE_OPEN_READ_WRITE, ) .expect("Could not connect to DB"); conn.execute( "CREATE TABLE IF NOT EXISTS posts ( id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, title TEXT NOT NULL, author TEXT NOT NULL, body TEXT NOT NULL )", rusqlite::NO_PARAMS, ) .expect("Could not initialize DB"); info!( "Database connection established in {}ms", start.elapsed().as_millis() ); conn } pub fn new(rx: mpsc::Receiver) -> Self { Conn { db: Conn::init(), rx, } } } impl Cmd { pub fn new(txt: &str) -> Self { match txt { "create" => Cmd::Create, "update" => Cmd::Update, "disconnect" => Cmd::Disconnect, _ => Cmd::NOOP, } } } impl Post { pub fn new(title: &str, author: &str, body: &str) -> Self { let id = rand::random::(); let title = title.to_string(); let author = author.to_string(); let body = body.to_string(); Post { id, title, author, body, } } pub fn id(&self) -> String { format!("{}", self.id) } pub fn title(&self) -> String { self.title.clone() } pub fn author(&self) -> String { self.author.clone() } pub fn body(&self) -> String { self.body.clone() } }