use crate::{ diesel::ExpressionMethods, newtypes::LanguageId, schema::language::dsl::{code, id, language}, source::language::Language, utils::{get_conn, DbPool}, }; use diesel::{result::Error, QueryDsl}; use diesel_async::{AsyncPgConnection, RunQueryDsl}; impl Language { pub async fn read_all(pool: &DbPool) -> Result, Error> { let conn = &mut get_conn(pool).await?; Self::read_all_conn(conn).await } pub async fn read_all_conn(conn: &mut AsyncPgConnection) -> Result, Error> { language.load::(conn).await } pub async fn read_from_id(pool: &DbPool, id_: LanguageId) -> Result { let conn = &mut get_conn(pool).await?; language.filter(id.eq(id_)).first::(conn).await } /// Attempts to find the given language code and return its ID. If not found, returns none. pub async fn read_id_from_code( pool: &DbPool, code_: Option<&str>, ) -> Result, Error> { if let Some(code_) = code_ { let conn = &mut get_conn(pool).await?; Ok( language .filter(code.eq(code_)) .first::(conn) .await .map(|l| l.id) .ok(), ) } else { Ok(None) } } } #[cfg(test)] mod tests { use crate::{source::language::Language, utils::build_db_pool_for_tests}; use serial_test::serial; #[tokio::test] #[serial] async fn test_languages() { let pool = &build_db_pool_for_tests().await; let all = Language::read_all(pool).await.unwrap(); assert_eq!(184, all.len()); assert_eq!("ak", all[5].code); assert_eq!("lv", all[99].code); assert_eq!("yi", all[179].code); } }