72 lines
1.7 KiB
Rust
72 lines
1.7 KiB
Rust
|
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|