persmgr/src/db/tables/records/ranks.rs

76 lines
2.1 KiB
Rust

use anyhow::Result;
use persmgr_derive::TableMeta;
use sqlx::prelude::FromRow;
use crate::db::{
CurrPool,
tables::{ForeignKey, TableMeta, assignables::ranks::Rank, user::User},
};
#[derive(Debug, Default, Clone, FromRow, TableMeta)]
#[meta(table = "records_ranks")]
pub struct RankRecord {
pub id: i64,
pub user_id: ForeignKey<User>,
pub rank_id: ForeignKey<Rank>,
pub author_id: ForeignKey<User>,
pub created_at: i64,
}
impl RankRecord {
pub async fn insert_new(&self, pool: &CurrPool) -> Result<Self> {
let session = sqlx::query_as!(
RankRecord,
r#"
INSERT INTO records_ranks (user_id, rank_id, author_id, created_at)
VALUES ($1, $2, $3, $4)
RETURNING *
"#,
*self.user_id,
*self.rank_id,
*self.author_id,
self.created_at,
)
.fetch_one(pool)
.await?;
Ok(session)
}
pub async fn get_by_id(pool: &CurrPool, id: i64) -> anyhow::Result<Self> {
let session = sqlx::query_as!(RankRecord, "SELECT * FROM records_ranks WHERE id = $1", id)
.fetch_one(pool)
.await?;
Ok(session)
}
pub async fn get_by_user_id(pool: &CurrPool, id: i64) -> anyhow::Result<Vec<Self>> {
let session = sqlx::query_as!(
RankRecord,
"SELECT * FROM records_ranks WHERE user_id = $1",
id
)
.fetch_all(pool)
.await?;
Ok(session)
}
pub async fn get_by_author_id(pool: &CurrPool, id: i64) -> anyhow::Result<Vec<Self>> {
let session = sqlx::query_as!(
RankRecord,
"SELECT * FROM records_ranks WHERE author_id = $1",
id
)
.fetch_all(pool)
.await?;
Ok(session)
}
pub async fn get_by_rank_id(pool: &CurrPool, id: i64) -> anyhow::Result<Vec<Self>> {
let session = sqlx::query_as!(
RankRecord,
"SELECT * FROM records_ranks WHERE rank_id = $1",
id
)
.fetch_all(pool)
.await?;
Ok(session)
}
}