51 lines
1.4 KiB
Rust
51 lines
1.4 KiB
Rust
|
use entity::alerts;
|
||
|
|
use migration::{Migrator, MigratorTrait};
|
||
|
|
use sea_orm::{ActiveModelTrait, Database, DatabaseConnection, DbErr, EntityTrait, Set};
|
||
|
|
use tracing::debug;
|
||
|
|
|
||
|
|
use crate::json::alert::Alert;
|
||
|
|
|
||
|
|
pub struct Db {
|
||
|
|
connection: DatabaseConnection,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Db {
|
||
|
|
pub async fn is_alert_updated(&self, alert: &Alert) -> Result<bool, DbErr> {
|
||
|
|
let db_alert = entity::alerts::Entity::find_by_id(&alert.identifier)
|
||
|
|
.one(&self.connection)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
Ok(match db_alert {
|
||
|
|
Some(db_alert) => db_alert.publish_date < alert.publish_date,
|
||
|
|
None => true,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn save_alert(&self, alert: &Alert) -> Result<(), DbErr> {
|
||
|
|
let db_alert = entity::alerts::Entity::find_by_id(&alert.identifier)
|
||
|
|
.one(&self.connection)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
let alert = alerts::ActiveModel {
|
||
|
|
id: Set(alert.identifier.to_owned()),
|
||
|
|
publish_date: Set(alert.publish_date),
|
||
|
|
};
|
||
|
|
|
||
|
|
match db_alert {
|
||
|
|
Some(_) => alert.save(&self.connection).await?,
|
||
|
|
None => alert.insert(&self.connection).await?.into(),
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn connect(connection_string: &str) -> Result<Db, DbErr> {
|
||
|
|
debug!("connecting to {connection_string}...");
|
||
|
|
|
||
|
|
let connection: DatabaseConnection = Database::connect(connection_string).await?;
|
||
|
|
Migrator::up(&connection, None).await?;
|
||
|
|
|
||
|
|
Ok(Db { connection })
|
||
|
|
}
|