datasources 0.1.5

Async datasource interfaces, middleware, registry, and test doubles
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

use futures_util::TryStreamExt;
use crate::delete_nested::DeleteNested;
use crate::enrichment::Enrichment;
use crate::repository::ICRUDRepository;
use crate::result::Result;
use crate::row::Row;
use crate::{Operator, Where};
use async_trait::async_trait;
use std::sync::Arc;

#[async_trait]
pub trait DeleteEnrichmentStrategy: Send + Sync {
    async fn delete_children(
        &self,
        writer: Arc<dyn ICRUDRepository>,
        enrichment: &Enrichment,
        parent: &Row,
        delete_nested: &dyn DeleteNested,
    ) -> Result<()>;

    async fn delete_linked(
        &self,
        writer: Arc<dyn ICRUDRepository>,
        enrichment: &Enrichment,
        parent: &Row,
    ) -> Result<()>;
}

pub(crate) async fn load(
    writer: &dyn ICRUDRepository,
    enrichment: &Enrichment,
    parent: &Row,
) -> Result<Vec<Row>> {
    let Some(key) = parent.get(&enrichment.source_property).cloned() else {
        return Ok(Vec::new());
    };
    writer
        .read(
            Some(&Where::new(Operator::equals(enrichment.related_key.clone(), key))),
            None, None
        )
        .try_collect::<Vec<_>>().await
}

pub(crate) fn children_where(
    writer: &dyn ICRUDRepository,
    enrichment: &Enrichment,
    parent: &Row,
) -> Option<Where> {
    if !writer.id_names().is_empty() {
        return None;
    }
    parent
        .get(&enrichment.source_property)
        .cloned()
        .map(|key| Where::new(Operator::equals(enrichment.related_key.clone(), key)))
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::crud_repository::CRUDRepository;
    use crate::default_write_enrichment_strategy::DefaultWriteEnrichmentStrategy;
    use crate::repository::IReadonlyRepository;
    use crate::value::Value;
    use std::sync::Arc;

    struct SkipDelete;

    #[async_trait]
    impl DeleteEnrichmentStrategy for SkipDelete {
        async fn delete_children(
            &self,
            _writer: Arc<dyn ICRUDRepository>,
            _enrichment: &Enrichment,
            _parent: &Row,
            _delete_nested: &dyn DeleteNested,
        ) -> Result<()> {
            Ok(())
        }

        async fn delete_linked(
            &self,
            _writer: Arc<dyn ICRUDRepository>,
            _enrichment: &Enrichment,
            _parent: &Row,
        ) -> Result<()> {
            Ok(())
        }
    }

    #[tokio::test]
    async fn delete_strategy_can_skip_children() {
        let write = Arc::new(DefaultWriteEnrichmentStrategy::new(
            Some(Arc::new(SkipDelete)),
            None,
            None,
            false,
        ));
        let children = Arc::new(CRUDRepository::memory(vec![]));
        let mut parents = CRUDRepository::memory_with(vec![], Some(write));
        let mut parent = Row::new();
        let created = parents.create(&mut parent).await.unwrap();
        let mut child = Row::new();
        child.insert("parent_id".into(), created.get("id").unwrap().clone());
        child.insert("label".into(), Value::from("a"));
        children.create(&mut child).await.unwrap();
        parents.add_enriched(
            Enrichment::new("children", "id", "parent_id", None, crate::EnrichmentPropertyType::List),
            children.clone(),
            None,
        );
        parents.delete(&created, None).await.unwrap();
        let left = children.read(None, None, None).try_collect::<Vec<_>>().await.unwrap();
        assert_eq!(left.len(), 1);
        assert_eq!(left[0].get("label"), Some(&Value::from("a")));
        assert!(parents.read(None, None, None).try_collect::<Vec<_>>().await.unwrap().is_empty());
    }
}