mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-26 09:10:32 +00:00
c9f1407429
* Initial commit to bump diesel to 2.0.0-rc.0 and see what happens * Add chrono feature from diesel * db_schema crate is close to building? * Upgrade diesel-derive-newtype * Mostly modifying references to connections to be mutable ones; also used new way to do migrations as suggested by the migration guide; a lot more compiles now, though I can't figure out this tricky ToSql issue at the moment * Running clippy --fix * Trying to fix drone clippy 1 * Fix clippy * Upgrade clux-musl * Trying to fix drone clippy 2 * Trying to fix drone clippy 3 * Trying to fix drone clippy 5 * Adding diesel table aliases, removing sql view hack. Fixes #2101 Co-authored-by: Steven Chu <stevenc1@gmail.com> Co-authored-by: Nutomic <me@nutomic.com>
55 lines
1.5 KiB
Rust
55 lines
1.5 KiB
Rust
use crate::{newtypes::LanguageId, source::language::Language};
|
|
use diesel::{result::Error, PgConnection, RunQueryDsl, *};
|
|
|
|
impl Language {
|
|
pub fn read_all(conn: &mut PgConnection) -> Result<Vec<Language>, Error> {
|
|
use crate::schema::language::dsl::*;
|
|
language.load::<Self>(conn)
|
|
}
|
|
|
|
pub fn read_from_id(conn: &mut PgConnection, id_: LanguageId) -> Result<Language, Error> {
|
|
use crate::schema::language::dsl::*;
|
|
language.filter(id.eq(id_)).first::<Self>(conn)
|
|
}
|
|
|
|
pub fn read_id_from_code(conn: &mut PgConnection, code_: &str) -> Result<LanguageId, Error> {
|
|
use crate::schema::language::dsl::*;
|
|
Ok(language.filter(code.eq(code_)).first::<Self>(conn)?.id)
|
|
}
|
|
|
|
pub fn read_id_from_code_opt(
|
|
conn: &mut PgConnection,
|
|
code_: Option<&str>,
|
|
) -> Result<Option<LanguageId>, Error> {
|
|
if let Some(code_) = code_ {
|
|
Ok(Some(Language::read_id_from_code(conn, code_)?))
|
|
} else {
|
|
Ok(None)
|
|
}
|
|
}
|
|
|
|
pub fn read_undetermined(conn: &mut PgConnection) -> Result<LanguageId, Error> {
|
|
use crate::schema::language::dsl::*;
|
|
Ok(language.filter(code.eq("und")).first::<Self>(conn)?.id)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use crate::{source::language::Language, utils::establish_unpooled_connection};
|
|
use serial_test::serial;
|
|
|
|
#[test]
|
|
#[serial]
|
|
fn test_languages() {
|
|
let conn = &mut establish_unpooled_connection();
|
|
|
|
let all = Language::read_all(conn).unwrap();
|
|
|
|
assert_eq!(184, all.len());
|
|
assert_eq!("ak", all[5].code);
|
|
assert_eq!("lv", all[99].code);
|
|
assert_eq!("yi", all[179].code);
|
|
}
|
|
}
|