halo-sqlbuilder 1.0.0

Composable SQL builder and argument collector
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

//! Injection: inject extra SQL fragments at predefined markers.
#![allow(dead_code)]

use std::collections::HashMap;

pub(crate) type InjectionMarker = usize;

#[derive(Debug, Default, Clone)]
pub(crate) struct Injection {
    marker_sqls: HashMap<InjectionMarker, Vec<String>>,
}

impl Injection {
    pub(crate) fn new() -> Self {
        Self::default()
    }

    pub(crate) fn sql(&mut self, marker: InjectionMarker, sql: impl Into<String>) {
        self.marker_sqls.entry(marker).or_default().push(sql.into());
    }

    pub(crate) fn at(&self, marker: InjectionMarker) -> &[String] {
        self.marker_sqls
            .get(&marker)
            .map(|v| v.as_slice())
            .unwrap_or(&[])
    }
}