nyanoblog/src/data/account/pg.rs

use serde;
use sqlx::{Executor, PgPool};

use crate::core::*;
use crate::model::account::{Account, NewAccount};

pub struct PgAccountDataSource {
    pool: PgPool,
}

impl PgAccountDataSource {
    pub fn new(pool: PgPool) -> PgAccountDataSource {
        PgAccountDataSource { pool }
    }

    pub async fn by_id<I>(&self, id: I) -> Result<Account>
    where
        I: Into<Id> + Send,
    {
        let id: Id = id.into();
        let account: Account = sqlx::query_as("SELECT * FROM accounts WHERE id = $1")
            .bind(id)
            .fetch_one(&self.pool)
            .await?;
        Ok(account)
    }

    pub async fn create<T>(&self, new: T) -> Result<Account>
    where
        T: Into<NewAccount> + Send,
    {
        let new = new.into();
        let account: Account =
            sqlx::query_as("INSERT INTO accounts (name, domain, display_name) VALUES ($1, $2, $3)")
                .bind(new.name)
                .bind(new.domain)
                .bind(new.display_name)
                .fetch_one(&self.pool)
                .await?;
        Ok(account)
    }
}