cargo fmt
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
use diesel::*;
|
||||
use serde::{Serialize, de::DeserializeOwned};
|
||||
use anyhow::Result;
|
||||
use diesel::*;
|
||||
use serde::{de::DeserializeOwned, Serialize};
|
||||
|
||||
#[derive(Insertable, Queryable, AsChangeset)]
|
||||
#[diesel(table_name = crate::schema::settings)]
|
||||
@@ -18,16 +18,15 @@ impl<'a> Settings<'a> {
|
||||
Self { connection }
|
||||
}
|
||||
|
||||
pub fn put<T: Serialize>(
|
||||
&mut self,
|
||||
k: &str,
|
||||
v: &T,
|
||||
) -> Result<()> {
|
||||
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 })
|
||||
.values(SettingsRow {
|
||||
key: k,
|
||||
value: &bytes,
|
||||
})
|
||||
.on_conflict(key)
|
||||
.do_update()
|
||||
.set(value.eq(&bytes))
|
||||
@@ -36,10 +35,7 @@ impl<'a> Settings<'a> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get<T: DeserializeOwned>(
|
||||
&mut self,
|
||||
k: &str,
|
||||
) -> Result<Option<T>> {
|
||||
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)
|
||||
@@ -49,7 +45,7 @@ impl<'a> Settings<'a> {
|
||||
|
||||
Ok(match blob {
|
||||
Some(b) => Some(bincode::deserialize(&b)?),
|
||||
None => None,
|
||||
None => None,
|
||||
})
|
||||
}
|
||||
|
||||
@@ -60,12 +56,8 @@ impl<'a> Settings<'a> {
|
||||
|
||||
pub fn list_keys(&mut self) -> Result<Vec<String>> {
|
||||
use crate::schema::settings::dsl::*;
|
||||
let keys: Vec<String> = settings
|
||||
.select(key)
|
||||
.load(self.connection)?;
|
||||
|
||||
let keys: Vec<String> = settings.select(key).load(self.connection)?;
|
||||
|
||||
Ok(keys)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user