Add booking dao implementations

dao_impl/src/booking.rs

@@ -0,0 +1,109 @@
+use std::sync::Arc;
+
+use crate::ResultDbErrorExt;
+use async_trait::async_trait;
+use dao::{
+    booking::{BookingDao, BookingEntity},
+    DaoError,
+};
+use sqlx::{query, query_as};
+use time::{format_description::well_known::Iso8601, PrimitiveDateTime};
+use uuid::Uuid;
+
+struct BookingDb {
+    id: Vec<u8>,
+    sales_person_id: Vec<u8>,
+    slot_id: Vec<u8>,
+    calendar_week: i64,
+    year: i64,
+    created: String,
+    deleted: Option<String>,
+    update_version: Vec<u8>,
+}
+impl TryFrom<&BookingDb> for BookingEntity {
+    type Error = DaoError;
+    fn try_from(booking: &BookingDb) -> Result<Self, Self::Error> {
+        Ok(Self {
+            id: Uuid::from_slice(booking.id.as_ref()).unwrap(),
+            sales_person_id: Uuid::from_slice(booking.sales_person_id.as_ref()).unwrap(),
+            slot_id: Uuid::from_slice(booking.slot_id.as_ref()).unwrap(),
+            calendar_week: booking.calendar_week as i32,
+            year: booking.year as u32,
+            created: PrimitiveDateTime::parse(&booking.created, &Iso8601::DATE_TIME)?,
+            deleted: booking
+                .deleted
+                .as_ref()
+                .map(|deleted| PrimitiveDateTime::parse(deleted, &Iso8601::DATE_TIME))
+                .transpose()?,
+            version: Uuid::from_slice(&booking.update_version).unwrap(),
+        })
+    }
+}
+
+pub struct BookingDaoImpl {
+    pub pool: Arc<sqlx::SqlitePool>,
+}
+impl BookingDaoImpl {
+    pub fn new(pool: Arc<sqlx::SqlitePool>) -> Self {
+        Self { pool }
+    }
+}
+
+#[async_trait]
+impl BookingDao for BookingDaoImpl {
+    async fn all(&self) -> Result<Arc<[BookingEntity]>, DaoError> {
+        Ok(query_as!(
+            BookingDb,
+            "SELECT id, sales_person_id, slot_id, calendar_week, year, created, deleted, update_version FROM booking WHERE deleted IS NULL"
+        )
+            .fetch_all(self.pool.as_ref())
+            .await
+            .map_db_error()?
+            .iter()
+            .map(BookingEntity::try_from)
+            .collect::<Result<Arc<[BookingEntity]>, DaoError>>()?
+        )
+    }
+    async fn find_by_id(&self, id: Uuid) -> Result<Option<BookingEntity>, DaoError> {
+        let id_vec = id.as_bytes().to_vec();
+        Ok(query_as!(
+            BookingDb,
+            "SELECT id, sales_person_id, slot_id, calendar_week, year, created, deleted, update_version FROM booking WHERE id = ?",
+            id_vec,
+        )
+        .fetch_optional(self.pool.as_ref())
+        .await
+        .map_db_error()?
+        .as_ref()
+        .map(BookingEntity::try_from)
+        .transpose()?)
+    }
+    async fn create(&self, entity: &BookingEntity, process: &str) -> Result<(), DaoError> {
+        let id_vec = entity.id.as_bytes().to_vec();
+        let sales_person_id_vec = entity.sales_person_id.as_bytes().to_vec();
+        let slot_id_vec = entity.slot_id.as_bytes().to_vec();
+        let created = entity.created.to_string();
+        let deleted = entity.deleted.as_ref().map(|deleted| deleted.to_string());
+        let version_vec = entity.version.as_bytes().to_vec();
+        query!("INSERT INTO booking (id, sales_person_id, slot_id, calendar_week, year, created, deleted, update_version, update_process) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)",
+            id_vec, sales_person_id_vec, slot_id_vec, entity.calendar_week, entity.year, created, deleted, version_vec, process
+        ).execute(self.pool.as_ref()).await.map_db_error()?;
+        Ok(())
+    }
+    async fn update(&self, entity: &BookingEntity, process: &str) -> Result<(), DaoError> {
+        let id_vec = entity.id.as_bytes().to_vec();
+        let version_vec = entity.version.as_bytes().to_vec();
+        let deleted = entity.deleted.as_ref().map(|deleted| deleted.to_string());
+        query!(
+            "UPDATE booking SET deleted = ?, update_version = ?, update_process = ? WHERE id = ?",
+            deleted,
+            version_vec,
+            process,
+            id_vec
+        )
+        .execute(self.pool.as_ref())
+        .await
+        .map_db_error()?;
+        Ok(())
+    }
+}

dao_impl/src/lib.rs

@@ -4,6 +4,7 @@ use async_trait::async_trait;
 use dao::DaoError;
 use sqlx::{query, query_as, SqlitePool};
 
+pub mod booking;
 pub mod sales_person;
 pub mod slot;