34 lines
674 B
Rust
34 lines
674 B
Rust
|
//! All logic for snapshots.
|
||
|
|
|
||
|
|
use {
|
||
|
|
color_eyre::Result,
|
||
|
|
sea_orm::{prelude::*, QueryOrder},
|
||
|
|
};
|
||
|
|
|
||
|
|
use crate::entities::snapshot;
|
||
|
|
|
||
|
|
mod create;
|
||
|
|
|
||
|
|
pub use create::create;
|
||
|
|
|
||
|
|
/// Get a snapshot for a given date.
|
||
|
|
pub async fn get_by_date(
|
||
|
|
db: &DatabaseConnection,
|
||
|
|
date: ChronoDate,
|
||
|
|
) -> Result<Option<snapshot::Model>> {
|
||
|
|
let existing = snapshot::Entity::find()
|
||
|
|
.filter(snapshot::Column::Date.eq(date))
|
||
|
|
.order_by_desc(snapshot::Column::Date)
|
||
|
|
.one(db)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
Ok(existing)
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Get all snapshots.
|
||
|
|
pub async fn get_all(db: &DatabaseConnection) -> Result<Vec<snapshot::Model>> {
|
||
|
|
let snapshots = snapshot::Entity::find().all(db).await?;
|
||
|
|
|
||
|
|
Ok(snapshots)
|
||
|
|
}
|