48 lines
1.5 KiB
Rust
48 lines
1.5 KiB
Rust
use sea_orm::entity::prelude::*;
|
|
use sea_orm::{ActiveValue, ColumnTrait, EntityTrait, IntoActiveModel, QueryFilter, TryIntoModel};
|
|
|
|
pub use crate::models::_entities::shop_settings::{ActiveModel, Column, Entity, Model};
|
|
pub type ShopSettings = Entity;
|
|
|
|
#[async_trait::async_trait]
|
|
impl ActiveModelBehavior for ActiveModel {
|
|
async fn before_save<C>(self, _db: &C, insert: bool) -> std::result::Result<Self, DbErr>
|
|
where
|
|
C: ConnectionTrait,
|
|
{
|
|
if !insert && self.updated_at.is_unchanged() {
|
|
let mut this = self;
|
|
this.updated_at = ActiveValue::set(chrono::Utc::now().into());
|
|
Ok(this)
|
|
} else {
|
|
Ok(self)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Entity {
|
|
pub async fn get<C: ConnectionTrait>(db: &C, key: &str) -> Result<Option<String>, DbErr> {
|
|
Ok(Entity::find()
|
|
.filter(Column::Key.eq(key))
|
|
.one(db)
|
|
.await?
|
|
.and_then(|setting| setting.value))
|
|
}
|
|
|
|
pub async fn set<C: ConnectionTrait>(db: &C, key: &str, value: Option<String>) -> Result<Model, DbErr> {
|
|
let mut active = match Entity::find()
|
|
.filter(Column::Key.eq(key))
|
|
.one(db)
|
|
.await?
|
|
{
|
|
Some(existing) => existing.into_active_model(),
|
|
None => ActiveModel {
|
|
key: ActiveValue::set(key.to_string()),
|
|
..Default::default()
|
|
},
|
|
};
|
|
active.value = ActiveValue::set(value);
|
|
active.save(db).await?.try_into_model()
|
|
}
|
|
}
|