]> Untitled Git - lemmy.git/blob - crates/db_schema/src/impls/language.rs
Fix data fetching from Peertube (fixes #2689) (#2703)
[lemmy.git] / crates / db_schema / src / impls / language.rs
1 use crate::{
2   diesel::ExpressionMethods,
3   newtypes::LanguageId,
4   schema::language::dsl::{code, id, language},
5   source::language::Language,
6   utils::{get_conn, DbPool},
7 };
8 use diesel::{result::Error, QueryDsl};
9 use diesel_async::{AsyncPgConnection, RunQueryDsl};
10
11 impl Language {
12   pub async fn read_all(pool: &DbPool) -> Result<Vec<Language>, Error> {
13     let conn = &mut get_conn(pool).await?;
14     Self::read_all_conn(conn).await
15   }
16
17   pub async fn read_all_conn(conn: &mut AsyncPgConnection) -> Result<Vec<Language>, Error> {
18     language.load::<Self>(conn).await
19   }
20
21   pub async fn read_from_id(pool: &DbPool, id_: LanguageId) -> Result<Language, Error> {
22     let conn = &mut get_conn(pool).await?;
23     language.filter(id.eq(id_)).first::<Self>(conn).await
24   }
25
26   /// Attempts to find the given language code and return its ID. If not found, returns none.
27   pub async fn read_id_from_code(
28     pool: &DbPool,
29     code_: Option<&str>,
30   ) -> Result<Option<LanguageId>, Error> {
31     if let Some(code_) = code_ {
32       let conn = &mut get_conn(pool).await?;
33       Ok(
34         language
35           .filter(code.eq(code_))
36           .first::<Self>(conn)
37           .await
38           .map(|l| l.id)
39           .ok(),
40       )
41     } else {
42       Ok(None)
43     }
44   }
45 }
46
47 #[cfg(test)]
48 mod tests {
49   use crate::{source::language::Language, utils::build_db_pool_for_tests};
50   use serial_test::serial;
51
52   #[tokio::test]
53   #[serial]
54   async fn test_languages() {
55     let pool = &build_db_pool_for_tests().await;
56
57     let all = Language::read_all(pool).await.unwrap();
58
59     assert_eq!(184, all.len());
60     assert_eq!("ak", all[5].code);
61     assert_eq!("lv", all[99].code);
62     assert_eq!("yi", all[179].code);
63   }
64 }