Coverage / crates/hub/src/db/push.rs 100.0%

pub async fn register(pool: &Pool, token_id: i64, account_id: i64, fcm_token: &str, public_key: &str) -> sqlx::Result<()> {

    let now = now_ms();

    sqlx::query(

        "INSERT INTO push_devices (token_id, account_id, fcm_token, public_key, created_ms, updated_ms) VALUES ($1, $2, $3, $4, $5, $5) \

         ON CONFLICT (token_id) DO UPDATE SET fcm_token = excluded.fcm_token, public_key = excluded.public_key, updated_ms = excluded.updated_ms",

    )

    .bind(token_id)

    .bind(account_id)

    .bind(fcm_token)

    .bind(public_key)

    .bind(now)

    .execute(pool)

    .await?;

    Ok(())

}

pub async fn forget(pool: &Pool, token_id: i64) -> sqlx::Result<bool> {

    Ok(sqlx::query("DELETE FROM push_devices WHERE token_id = $1").bind(token_id).execute(pool).await?.rows_affected() > 0)

}

pub async fn forget_fcm(pool: &Pool, fcm_token: &str) -> sqlx::Result<()> {

    sqlx::query("DELETE FROM push_devices WHERE fcm_token = $1").bind(fcm_token).execute(pool).await?;

    Ok(())

}

pub async fn of_account(pool: &Pool, account_id: i64) -> sqlx::Result<Vec<Device>> {

    sqlx::query_as("SELECT token_id, account_id, fcm_token, public_key FROM push_devices WHERE account_id = $1")

        .bind(account_id)

        .fetch_all(pool)

        .await

}