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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
use backend::Backend;
use connection::Connection;
use helper_types::Limit;
use query_builder::{QueryFragment, AsQuery, QueryId};
use query_source::Queryable;
use result::{QueryResult, first_or_not_found};
use super::LimitDsl;
use types::HasSqlType;
pub trait LoadQuery<Conn, U>: LoadDsl<Conn> {
fn internal_load(self, conn: &Conn) -> QueryResult<Vec<U>>;
}
impl<Conn, T, U> LoadQuery<Conn, U> for T where
Conn: Connection,
Conn::Backend: HasSqlType<T::SqlType>,
T: AsQuery,
T::Query: QueryFragment<Conn::Backend> + QueryId,
U: Queryable<T::SqlType, Conn::Backend>,
{
fn internal_load(self, conn: &Conn) -> QueryResult<Vec<U>> {
conn.query_by_index(self)
}
}
pub trait LoadDsl<Conn>: Sized {
fn load<U>(self, conn: &Conn) -> QueryResult<Vec<U>> where
Self: LoadQuery<Conn, U>,
{
self.internal_load(conn)
}
fn get_result<U>(self, conn: &Conn) -> QueryResult<U> where
Self: LoadQuery<Conn, U>,
{
first_or_not_found(self.load(conn))
}
fn get_results<U>(self, conn: &Conn) -> QueryResult<Vec<U>> where
Self: LoadQuery<Conn, U>,
{
self.load(conn)
}
}
impl<Conn, T> LoadDsl<Conn> for T where
Conn: Connection,
Conn::Backend: HasSqlType<T::SqlType>,
T: AsQuery,
T::Query: QueryFragment<Conn::Backend> + QueryId,
{
}
pub trait FirstDsl<Conn>: LimitDsl + LoadDsl<Conn> {
fn first<U>(self, conn: &Conn) -> QueryResult<U> where
Limit<Self>: LoadQuery<Conn, U>,
{
self.limit(1).get_result(conn)
}
}
impl<Conn, T: LimitDsl + LoadDsl<Conn>> FirstDsl<Conn> for T {
}
pub trait ExecuteDsl<Conn: Connection<Backend=DB>, DB: Backend = <Conn as Connection>::Backend>: Sized {
fn execute(self, conn: &Conn) -> QueryResult<usize>;
}
impl<Conn, T> ExecuteDsl<Conn> for T where
Conn: Connection,
T: QueryFragment<Conn::Backend> + QueryId,
{
fn execute(self, conn: &Conn) -> QueryResult<usize> {
conn.execute_returning_count(&self)
}
}