crates/db_schema/src/impls/language.rs

   1 use crate::{diesel::ExpressionMethods, newtypes::LanguageId, source::language::Language};
   2 use diesel::{result::Error, PgConnection, QueryDsl, RunQueryDsl};
   3
   4 impl Language {
   5   pub fn read_all(conn: &mut PgConnection) -> Result<Vec<Language>, Error> {
   6     use crate::schema::language::dsl::*;
   7     language.load::<Self>(conn)
   8   }
   9
  10   pub fn read_from_id(conn: &mut PgConnection, id_: LanguageId) -> Result<Language, Error> {
  11     use crate::schema::language::dsl::*;
  12     language.filter(id.eq(id_)).first::<Self>(conn)
  13   }
  14
  15   pub fn read_id_from_code(conn: &mut PgConnection, code_: &str) -> Result<LanguageId, Error> {
  16     use crate::schema::language::dsl::*;
  17     Ok(language.filter(code.eq(code_)).first::<Self>(conn)?.id)
  18   }
  19
  20   pub fn read_id_from_code_opt(
  21     conn: &mut PgConnection,
  22     code_: Option<&str>,
  23   ) -> Result<Option<LanguageId>, Error> {
  24     if let Some(code_) = code_ {
  25       Ok(Some(Language::read_id_from_code(conn, code_)?))
  26     } else {
  27       Ok(None)
  28     }
  29   }
  30 }
  31
  32 #[cfg(test)]
  33 mod tests {
  34   use crate::{source::language::Language, utils::establish_unpooled_connection};
  35   use serial_test::serial;
  36
  37   #[test]
  38   #[serial]
  39   fn test_languages() {
  40     let conn = &mut establish_unpooled_connection();
  41
  42     let all = Language::read_all(conn).unwrap();
  43
  44     assert_eq!(184, all.len());
  45     assert_eq!("ak", all[5].code);
  46     assert_eq!("lv", all[99].code);
  47     assert_eq!("yi", all[179].code);
  48   }
  49 }