src/db/mod.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

use std::env;

use diesel::prelude::*;

use super::{Result, Error};
pub use self::models::*;

mod schema;
mod models;

fn connection() -> Result<PgConnection> {
    let database_url = env::var("DATABASE_URL")?;
    PgConnection::establish(&database_url).map_err(Error::from)
}

pub fn find_text(search: String) -> Result<TextMeme> {
    use self::schema::text_memes::dsl::*;

    let format_search = format!("%{}%", search);

    let conn = connection()?;
    text_memes
        .filter(title.ilike(&format_search).or(content.ilike(&format_search)))
        .limit(1)
        .first::<TextMeme>(&conn)
        .map_err(Error::from)
}

pub fn find_audio(search: String) -> Result<AudioMeme> {
    use self::schema::audio_memes::dsl::*;

    let format_search = format!("%{}%", search);

    let conn = connection()?;
    audio_memes
        .filter(title.ilike(format_search))
        .limit(1)
        .first::<AudioMeme>(&conn)
        .map_err(Error::from)
}

pub fn rand_audio() -> Result<AudioMeme> {
    use self::schema::audio_memes::dsl::*;

    let conn = connection()?;
    audio_memes
        .order(random.desc())
        .first::<AudioMeme>(&conn)
        .map_err(Error::from)
}

pub fn rand_text() -> Result<TextMeme> {
    use self::schema::text_memes::dsl::*;

    let conn = connection()?;
    text_memes
        .order(random.desc())
        .first::<TextMeme>(&conn)
        .map_err(Error::from)
}

use diesel::sql_types;
no_arg_sql_function!(random, sql_types::Double, "SQL random() function");