57 lines
1.5 KiB
Rust
57 lines
1.5 KiB
Rust
use std::ops::DerefMut;
|
|
|
|
use chrono::NaiveDate;
|
|
use rocket::serde::{Deserialize, Serialize};
|
|
use sqlx::{FromRow, Sqlite, SqlitePool, Transaction};
|
|
|
|
#[derive(FromRow, Debug, Serialize, Deserialize, PartialEq, Clone)]
|
|
pub struct Weather {
|
|
pub id: i64,
|
|
pub day: NaiveDate,
|
|
pub max_temp: f64,
|
|
pub wind_gust: f64,
|
|
pub rain_mm: f64,
|
|
}
|
|
|
|
impl Weather {
|
|
pub async fn find_by_day(db: &SqlitePool, day: NaiveDate) -> Option<Self> {
|
|
sqlx::query_as!(Self, "SELECT * FROM weather WHERE day = ?", day)
|
|
.fetch_one(db)
|
|
.await
|
|
.ok()
|
|
}
|
|
pub async fn find_by_id_tx(db: &mut Transaction<'_, Sqlite>, day: NaiveDate) -> Option<Self> {
|
|
sqlx::query_as!(Self, "SELECT * FROM weather WHERE day = ?", day)
|
|
.fetch_one(db.deref_mut())
|
|
.await
|
|
.ok()
|
|
}
|
|
|
|
pub async fn create(
|
|
db: &mut Transaction<'_, Sqlite>,
|
|
day: NaiveDate,
|
|
max_temp: f64,
|
|
wind_gust: f64,
|
|
rain_mm: f64,
|
|
) -> Result<(), String> {
|
|
sqlx::query!(
|
|
"INSERT INTO weather(day, max_temp, wind_gust, rain_mm) VALUES (?,?,?,?)",
|
|
day,
|
|
max_temp,
|
|
wind_gust,
|
|
rain_mm
|
|
)
|
|
.execute(db.deref_mut())
|
|
.await
|
|
.map_err(|e| e.to_string())?;
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn delete_all(db: &mut Transaction<'_, Sqlite>) {
|
|
sqlx::query!("DELETE FROM weather;")
|
|
.execute(db.deref_mut())
|
|
.await
|
|
.unwrap();
|
|
}
|
|
}
|