blob: e1c3a55ab04670801e31dda498e20e0fdf30babd (
plain) (
blame)
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
use std::env;
use diesel::prelude::*;
use diesel::r2d2::{ConnectionManager, ManageConnection};
use super::{Result, Error};
pub use self::models::*;
use self::schema::*;
mod schema;
mod models;
lazy_static! {
static ref DB_URL: String = env::var("DATABASE_URL").expect("no database url in environment").into();
static ref CONN_MGR: ConnectionManager<PgConnection> = ConnectionManager::new(DB_URL.clone());
}
pub fn connection() -> Result<PgConnection> {
CONN_MGR.connect().map_err(Error::from)
}
pub trait AssociatedData {
type Associated;
fn associated_data(&self, conn: &PgConnection) -> Result<Self::Associated>;
}
pub fn find_text(conn: &PgConnection, search: String) -> Result<TextMeme> {
let format_search = format!("%{}%", search);
text_memes::table
.filter(text_memes::title.ilike(&format_search).or(text_memes::content.ilike(&format_search)))
.limit(1)
.first::<TextMeme>(conn)
.map_err(Error::from)
}
pub fn find_audio(conn: &PgConnection, search: String) -> Result<AudioMeme> {
let format_search = format!("%{}%", search);
audio_memes::table
.filter(audio_memes::title.ilike(format_search))
.limit(1)
.first::<AudioMeme>(conn)
.map_err(Error::from)
}
pub fn find_image(conn: &PgConnection, search: String) -> Result<ImageMeme> {
let format_search = format!("%{}%", search);
image_memes::table
.filter(image_memes::title.ilike(format_search))
.limit(1)
.first::<ImageMeme>(conn)
.map_err(Error::from)
}
pub fn rand_text(conn: &PgConnection) -> Result<TextMeme> {
text_memes::table
.order(random.desc())
.first::<TextMeme>(conn)
.map_err(Error::from)
}
pub fn rand_image(conn: &PgConnection) -> Result<ImageMeme> {
image_memes::table
.order(random.desc())
.first::<ImageMeme>(conn)
.map_err(Error::from)
}
pub fn rand_audio(conn: &PgConnection) -> Result<AudioMeme> {
audio_memes::table
.order(random.desc())
.first::<AudioMeme>(conn)
.map_err(Error::from)
}
use diesel::sql_types;
no_arg_sql_function!(random, sql_types::Double, "SQL random() function");
|
