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
use crate::{ColumnAsExpr, ColumnTrait, DbBackend, IntoIdentity, QuerySelect, Statement};
use sea_query::QueryStatementBuilder;
/// A Trait for any type performing queries on a Model or ActiveModel
pub trait QueryTrait {
/// Constrain the QueryStatement to [QueryStatementBuilder] trait
type QueryStatement: QueryStatementBuilder;
/// Get a mutable ref to the query builder
fn query(&mut self) -> &mut Self::QueryStatement;
/// Get an immutable ref to the query builder
fn as_query(&self) -> &Self::QueryStatement;
/// Take ownership of the query builder
fn into_query(self) -> Self::QueryStatement;
/// Build the query as [`Statement`]
fn build(&self, db_backend: DbBackend) -> Statement {
let query_builder = db_backend.get_query_builder();
Statement::from_string_values_tuple(
db_backend,
self.as_query().build_any(query_builder.as_ref()),
)
}
/// Apply an operation on the [QueryTrait::QueryStatement] if the given `Option<T>` is `Some(_)`
///
/// # Example
///
/// ```
/// use sea_orm::{entity::*, query::*, tests_cfg::cake, DbBackend};
///
/// assert_eq!(
/// cake::Entity::find()
/// .apply_if(Some(3), |mut query, v| {
/// query.filter(cake::Column::Id.eq(v))
/// })
/// .apply_if(Some(100), QuerySelect::limit)
/// .apply_if(None, QuerySelect::offset::<Option<u64>>) // no-op
/// .build(DbBackend::Postgres)
/// .to_string(),
/// r#"SELECT "cake"."id", "cake"."name" FROM "cake" WHERE "cake"."id" = 3 LIMIT 100"#
/// );
/// ```
fn apply_if<T, F>(self, val: Option<T>, if_some: F) -> Self
where
Self: Sized,
F: FnOnce(Self, T) -> Self,
{
if let Some(val) = val {
if_some(self, val)
} else {
self
}
}
}
/// Select specific column for partial model queries
pub trait SelectColumns {
/// Add a select column
///
/// For more detail, please visit [QuerySelect::column]
fn select_column<C: ColumnTrait>(self, col: C) -> Self;
/// Add a select column with alias
///
/// For more detail, please visit [QuerySelect::column_as]
fn select_column_as<C, I>(self, col: C, alias: I) -> Self
where
C: ColumnAsExpr,
I: IntoIdentity;
}
impl<S> SelectColumns for S
where
S: QuerySelect,
{
fn select_column<C: ColumnTrait>(self, col: C) -> Self {
QuerySelect::column(self, col)
}
fn select_column_as<C, I>(self, col: C, alias: I) -> Self
where
C: ColumnAsExpr,
I: IntoIdentity,
{
QuerySelect::column_as(self, col, alias)
}
}