Add async-trait and sea-orm dependencies; implement SettingsService for configuration management
This commit is contained in:
1
apps/api/src/services.rs
Normal file
1
apps/api/src/services.rs
Normal file
@@ -0,0 +1 @@
|
||||
pub mod settings;
|
||||
89
apps/api/src/services/settings.rs
Normal file
89
apps/api/src/services/settings.rs
Normal file
@@ -0,0 +1,89 @@
|
||||
use std::sync::Arc;
|
||||
|
||||
use database::generated::entities::config::{self, ActiveModel as ConfigActiveModel};
|
||||
|
||||
use sea_orm::{
|
||||
ActiveModelTrait, ActiveValue, ColumnTrait, DatabaseConnection, DbErr, EntityTrait,
|
||||
IntoActiveModel, QueryFilter,
|
||||
};
|
||||
|
||||
use crate::errors::service_error::{IntoServiceError, ServiceError};
|
||||
|
||||
#[async_trait::async_trait]
|
||||
pub trait SettingsStore: Send + Sync {
|
||||
async fn get_setting(&self, key: &str) -> Result<String, ServiceError>;
|
||||
async fn set_setting(&self, key: &str, value: String) -> Result<(), ServiceError>;
|
||||
}
|
||||
|
||||
pub struct SettingsService {
|
||||
connection: Arc<DatabaseConnection>,
|
||||
}
|
||||
|
||||
impl SettingsService {
|
||||
pub fn new(connection: Arc<DatabaseConnection>) -> Self {
|
||||
Self { connection }
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait::async_trait]
|
||||
impl SettingsStore for SettingsService {
|
||||
async fn get_setting(&self, key: &str) -> Result<String, ServiceError> {
|
||||
let setting = config::Entity::find()
|
||||
.filter(config::Column::Key.eq(key))
|
||||
.one(&*self.connection)
|
||||
.await;
|
||||
|
||||
match setting {
|
||||
Err(err) => Err(err.into_service_error()),
|
||||
Ok(None) => Err(
|
||||
DbErr::RecordNotFound(format!("Setting with key '{}' not found", key))
|
||||
.into_service_error(),
|
||||
),
|
||||
Ok(Some(record)) => Ok(record.value),
|
||||
}
|
||||
}
|
||||
|
||||
async fn set_setting(&self, key: &str, value: String) -> Result<(), ServiceError> {
|
||||
let existing = config::Entity::find()
|
||||
.filter(config::Column::Key.eq(key))
|
||||
.one(&*self.connection)
|
||||
.await;
|
||||
|
||||
let handle_not_found = async |key: String, value: String| {
|
||||
let new_record = ConfigActiveModel {
|
||||
key: ActiveValue::Set(key),
|
||||
value: ActiveValue::Set(value),
|
||||
created_at: ActiveValue::Set(chrono::Utc::now()),
|
||||
updated_at: ActiveValue::Set(chrono::Utc::now()),
|
||||
};
|
||||
new_record
|
||||
.insert(&*self.connection)
|
||||
.await
|
||||
.map_err(|err| err.into_service_error())
|
||||
};
|
||||
|
||||
match existing {
|
||||
Err(err) => match err {
|
||||
DbErr::RecordNotFound(_) => {
|
||||
handle_not_found(key.to_string(), value).await?;
|
||||
}
|
||||
_ => {
|
||||
return Err(Box::new(err));
|
||||
}
|
||||
},
|
||||
Ok(None) => {
|
||||
handle_not_found(key.to_string(), value).await?;
|
||||
}
|
||||
Ok(Some(mut record)) => {
|
||||
record.value = value;
|
||||
record
|
||||
.into_active_model()
|
||||
.update(&*self.connection)
|
||||
.await
|
||||
.map_err(|err| err.into_service_error())?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user