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

use super::support::*;

/// Rewrites every query onto the unqualified `wide` table.
pub(super) struct RewriteToWide;

impl PartitionResolver for RewriteToWide {
    fn resolve(
        &self,
        _context: &QueryContext<'_>,
    ) -> Result<Option<PartitionName>, ExtensionError> {
        Ok(Some(PartitionName::new("wide")?))
    }
}

/// Rejects SQL that already carries a partition rewrite.
pub(super) struct RejectWide;

impl SqlInterceptor for RejectWide {
    fn intercept(
        &self,
        _context: &SqlContext,
        sql: &str,
    ) -> Result<Option<String>, ExtensionError> {
        assert!(sql.contains("FROM \"wide\""), "{sql}");
        Err(ExtensionError::rejected(
            "test",
            "partitioned queries are disabled",
        ))
    }
}

pub(super) struct DropTitle;

impl ResultFilter for DropTitle {
    fn filter(
        &self,
        _context: &ResultContext,
        _row_index: usize,
        row: &mut sift_query_sqlx::RowMap,
    ) -> Result<ResultFilterDecision, ExtensionError> {
        if row.get("title") == Some(&RowValue::Text("beta".to_owned())) {
            Ok(ResultFilterDecision::Discard)
        } else {
            Ok(ResultFilterDecision::Keep)
        }
    }
}

pub(super) struct NullPartition;

impl PartitionResolver for NullPartition {
    fn resolve(
        &self,
        _context: &QueryContext<'_>,
    ) -> Result<Option<PartitionName>, ExtensionError> {
        Ok(None)
    }
}