sift-query-sqlx 0.1.0

PostgreSQL renderer and sqlx executor for SiftQuery
Documentation 
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use sift_query_core::{
    CursorValue, PagingMode, ParamError, QueryPlan, SearchEntity, SearchProjection,
};

use crate::{PgSearchError, PgSiftQuery, RowMap};

impl PgSiftQuery {
    pub(crate) fn entity_cursor<E: SearchEntity>(
        &self,
        plan: &QueryPlan<E>,
        row: Option<&E>,
    ) -> Result<Option<String>, PgSearchError> {
        if !matches!(plan.paging.mode, PagingMode::Cursor(_)) || plan.paging.size == 0 {
            return Ok(None);
        }
        let Some(row) = row else {
            return Ok(None);
        };
        let codec = self
            .policy
            .cursor_codec()
            .ok_or_else(|| ParamError::InvalidValue {
                field: self.policy.names().cursor.clone(),
                reason: "cursor pagination is not configured".to_owned(),
            })?;
        let values = plan
            .order
            .iter()
            .map(|order| {
                row.cursor_value(order.field)
                    .ok_or_else(|| ParamError::InvalidValue {
                        field: self.policy.names().cursor.clone(),
                        reason: "entity cannot extract a cursor ordering value".to_owned(),
                    })
            })
            .collect::<Result<Vec<_>, _>>()?;
        codec
            .encode::<E>(&plan.order, &values)
            .map(Some)
            .map_err(PgSearchError::Param)
    }

    pub(crate) fn map_cursor<E: SearchEntity>(
        &self,
        plan: &QueryPlan<E>,
        row: Option<&RowMap>,
    ) -> Result<Option<String>, PgSearchError> {
        if !matches!(plan.paging.mode, PagingMode::Cursor(_)) || plan.paging.size == 0 {
            return Ok(None);
        }
        let Some(row) = row else {
            return Ok(None);
        };
        let codec = self
            .policy
            .cursor_codec()
            .ok_or_else(|| ParamError::InvalidValue {
                field: self.policy.names().cursor.clone(),
                reason: "cursor pagination is not configured".to_owned(),
            })?;
        let values = plan
            .order
            .iter()
            .map(|order| {
                let field = E::META
                    .field(order.field)
                    .ok_or_else(|| ParamError::UnknownField(format!("#{}", order.field.0)))?;
                let value = row
                    .get(field.public_name)
                    .ok_or_else(|| ParamError::InvalidValue {
                        field: self.policy.names().cursor.clone(),
                        reason: format!("cursor field {} is not projected", field.public_name),
                    })?;
                CursorValue::from_row_value(value)
            })
            .collect::<Result<Vec<_>, _>>()?;
        codec
            .encode::<E>(&plan.order, &values)
            .map(Some)
            .map_err(PgSearchError::Param)
    }

    pub(crate) fn projection_cursor<E, P>(
        &self,
        plan: &QueryPlan<E>,
        row: Option<&P>,
    ) -> Result<Option<String>, PgSearchError>
    where
        E: SearchEntity,
        P: SearchProjection<E>,
    {
        if !matches!(plan.paging.mode, PagingMode::Cursor(_)) || plan.paging.size == 0 {
            return Ok(None);
        }
        let Some(row) = row else {
            return Ok(None);
        };
        let codec = self
            .policy
            .cursor_codec()
            .ok_or_else(|| ParamError::InvalidValue {
                field: self.policy.names().cursor.clone(),
                reason: "cursor pagination is not configured".to_owned(),
            })?;
        let values = plan
            .order
            .iter()
            .map(|order| {
                row.cursor_value(order.field)
                    .ok_or_else(|| ParamError::InvalidValue {
                        field: self.policy.names().cursor.clone(),
                        reason: "projection cannot extract a cursor ordering value".to_owned(),
                    })
            })
            .collect::<Result<Vec<_>, _>>()?;
        codec
            .encode::<E>(&plan.order, &values)
            .map(Some)
            .map_err(PgSearchError::Param)
    }
}
#[cfg(test)]
mod tests;