reorg: split repo / database so settings can use db connection as well
This commit is contained in:
71
kordophone-db/src/settings.rs
Normal file
71
kordophone-db/src/settings.rs
Normal file
@@ -0,0 +1,71 @@
|
||||
use diesel::*;
|
||||
use serde::{Serialize, de::DeserializeOwned};
|
||||
use anyhow::Result;
|
||||
use crate::database::Database;
|
||||
#[derive(Insertable, Queryable, AsChangeset)]
|
||||
#[diesel(table_name = crate::schema::settings)]
|
||||
struct SettingsRow<'a> {
|
||||
key: &'a str,
|
||||
value: &'a [u8],
|
||||
}
|
||||
|
||||
pub struct Settings<'a> {
|
||||
db: &'a mut Database,
|
||||
}
|
||||
|
||||
impl<'a> Settings<'a> {
|
||||
pub fn new(db: &'a mut Database) -> Self {
|
||||
Self { db }
|
||||
}
|
||||
|
||||
pub fn put<T: Serialize>(
|
||||
&mut self,
|
||||
k: &str,
|
||||
v: &T,
|
||||
) -> Result<()> {
|
||||
use crate::schema::settings::dsl::*;
|
||||
let bytes = bincode::serialize(v)?;
|
||||
|
||||
diesel::insert_into(settings)
|
||||
.values(SettingsRow { key: k, value: &bytes })
|
||||
.on_conflict(key)
|
||||
.do_update()
|
||||
.set((value.eq(&bytes)))
|
||||
.execute(&mut self.db.connection)?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get<T: DeserializeOwned>(
|
||||
&mut self,
|
||||
k: &str,
|
||||
) -> Result<Option<T>> {
|
||||
use crate::schema::settings::dsl::*;
|
||||
let blob: Option<Vec<u8>> = settings
|
||||
.select(value)
|
||||
.filter(key.eq(k))
|
||||
.first(&mut self.db.connection)
|
||||
.optional()?;
|
||||
|
||||
Ok(match blob {
|
||||
Some(b) => Some(bincode::deserialize(&b)?),
|
||||
None => None,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn del(&mut self, k: &str) -> Result<usize> {
|
||||
use crate::schema::settings::dsl::*;
|
||||
Ok(diesel::delete(settings.filter(key.eq(k))).execute(&mut self.db.connection)?)
|
||||
}
|
||||
|
||||
pub fn list_keys(&mut self) -> Result<Vec<String>> {
|
||||
use crate::schema::settings::dsl::*;
|
||||
let keys: Vec<String> = settings
|
||||
.select(key)
|
||||
.load(&mut self.db.connection)?;
|
||||
|
||||
Ok(keys)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user