diesel/pg/expression/
operators.rs

Help
1use crate::backend::{Backend, DieselReserveSpecialization};
2use crate::expression::expression_types::NotSelectable;
3use crate::expression::{TypedExpressionType, ValidGrouping};
4use crate::pg::Pg;
5use crate::query_builder::update_statement::changeset::AssignmentTarget;
6use crate::query_builder::{AstPass, QueryFragment, QueryId};
7use crate::sql_types::{
8    Array, Bigint, Bool, DieselNumericOps, Inet, Integer, Jsonb, SqlType, Text,
9};
10use crate::{Column, QueryResult};
11
12__diesel_infix_operator!(IsDistinctFrom, " IS DISTINCT FROM ", ConstantNullability Bool, backend: Pg);
13__diesel_infix_operator!(IsNotDistinctFrom, " IS NOT DISTINCT FROM ", ConstantNullability Bool, backend: Pg);
14infix_operator!(OverlapsWith, " && ", backend: Pg);
15infix_operator!(Contains, " @> ", backend: Pg);
16infix_operator!(IsContainedBy, " <@ ", backend: Pg);
17infix_operator!(ILike, " ILIKE ", backend: Pg);
18infix_operator!(NotILike, " NOT ILIKE ", backend: Pg);
19infix_operator!(SimilarTo, " SIMILAR TO ", backend: Pg);
20infix_operator!(NotSimilarTo, " NOT SIMILAR TO ", backend: Pg);
21postfix_operator!(NullsFirst, " NULLS FIRST", NotSelectable, backend: Pg);
22postfix_operator!(NullsLast, " NULLS LAST", NotSelectable, backend: Pg);
23infix_operator!(ContainsNet, " >> ", backend: Pg);
24infix_operator!(ContainsNetLoose, " >>= ", backend: Pg);
25infix_operator!(IsContainedByNet, " << ", backend: Pg);
26infix_operator!(IsContainedByNetLoose, " <<= ", backend: Pg);
27infix_operator!(AndNet, " & ", Inet, backend: Pg);
28infix_operator!(OrNet, " | ", Inet, backend: Pg);
29infix_operator!(DifferenceNet, " - ", Bigint, backend: Pg);
30infix_operator!(HasKeyJsonb, " ? ", backend: Pg);
31infix_operator!(HasAnyKeyJsonb, " ?| ", backend: Pg);
32infix_operator!(HasAllKeysJsonb, " ?& ", backend: Pg);
33infix_operator!(RemoveFromJsonb, " - ", Jsonb, backend: Pg);
34__diesel_infix_operator!(
35    RetrieveAsObjectJson,
36    " -> ",
37    __diesel_internal_SameResultAsInput,
38    backend: Pg
39);
40infix_operator!(RetrieveAsTextJson, " ->> ", Text, backend: Pg);
41__diesel_infix_operator!(
42    RetrieveByPathAsObjectJson,
43    " #> ",
44    __diesel_internal_SameResultAsInput,
45    backend: Pg
46);
47infix_operator!(RetrieveByPathAsTextJson, " #>> ", Text, backend: Pg);
48infix_operator!(RemoveByPathFromJsonb, " #-", Jsonb, backend: Pg);
49
50#[derive(Debug, Clone, Copy, QueryId, DieselNumericOps, ValidGrouping)]
51#[doc(hidden)]
52pub struct ArrayIndex<L, R> {
53    pub(crate) array_expr: L,
54    pub(crate) index_expr: R,
55}
56
57impl<L, R> ArrayIndex<L, R> {
58    pub fn new(array_expr: L, index_expr: R) -> Self {
59        Self {
60            array_expr,
61            index_expr,
62        }
63    }
64}
65
66impl<L, R, ST> crate::expression::Expression for ArrayIndex<L, R>
67where
68    L: crate::expression::Expression<SqlType = Array<ST>>,
69    R: crate::expression::Expression<SqlType = Integer>,
70    ST: SqlType + TypedExpressionType,
71{
72    type SqlType = ST;
73}
74
75impl_selectable_expression!(ArrayIndex<L, R>);
76
77impl<L, R> QueryFragment<Pg> for ArrayIndex<L, R>
78where
79    L: QueryFragment<Pg>,
80    R: QueryFragment<Pg>,
81{
82    fn walk_ast<'b>(
83        &'b self,
84        mut out: crate::query_builder::AstPass<'_, 'b, Pg>,
85    ) -> crate::result::QueryResult<()> {
86        self.array_expr.walk_ast(out.reborrow())?;
87        out.push_sql("[");
88        self.index_expr.walk_ast(out.reborrow())?;
89        out.push_sql("]");
90        Ok(())
91    }
92}
93
94impl<L, R> AssignmentTarget for ArrayIndex<L, R>
95where
96    L: Column,
97{
98    type Table = <L as Column>::Table;
99    type QueryAstNode = ArrayIndex<UncorrelatedColumn<L>, R>;
100
101    fn into_target(self) -> Self::QueryAstNode {
102        ArrayIndex::new(UncorrelatedColumn(self.array_expr), self.index_expr)
103    }
104}
105
106/// Signifies that this column should be rendered without its 'correlation'
107/// (i.e. table name prefix). For update statements, fully qualified column
108/// names aren't allowed.
109#[derive(Debug)]
110pub struct UncorrelatedColumn<C>(C);
111
112impl<C, DB> QueryFragment<DB> for UncorrelatedColumn<C>
113where
114    C: Column,
115    DB: Backend + DieselReserveSpecialization,
116{
117    fn walk_ast<'b>(&'b self, mut out: AstPass<'_, 'b, DB>) -> QueryResult<()> {
118        out.push_identifier(C::NAME)
119    }
120}
diesel/pg/expression/operators.rs

diesel/pg/expression/
operators.rs
