polodb_core 5.3.0

An embedded document database
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

mod document_path;
mod set_operator;
mod inc_operator;
mod mul_operator;
mod rename_operator;
mod unset_operator;
mod push_operator;
mod pop_operator;
mod min_operator;
mod max_operator;

use bson::{Bson, Document};
use crate::Result;

#[derive(Debug, Default)]
pub(crate) struct UpdateResult {
    #[allow(dead_code)]
    pub(crate) updated: bool,
}

pub(crate) trait UpdateOperator {
    fn name(&self) -> &str;
    fn update(&self, value: &mut Bson) -> Result<UpdateResult>;
}

impl dyn UpdateOperator {

    pub(crate) fn validate_key(doc: &Document) -> Result<()> {
        document_path::validate_update_paths(doc.keys().map(String::as_str))
    }

    pub(crate) fn validate_update_document(update: &Document) -> Result<()> {
        let mut paths = Vec::new();
        for (operator, value) in update {
            if !matches!(
                operator.as_str(),
                "$inc"
                    | "$set"
                    | "$max"
                    | "$min"
                    | "$mul"
                    | "$rename"
                    | "$unset"
                    | "$push"
                    | "$pop"
            ) {
                continue;
            }

            let Some(fields) = value.as_document() else {
                continue;
            };
            paths.extend(fields.keys().map(String::as_str));

            if operator == "$rename" {
                paths.extend(fields.values().filter_map(Bson::as_str));
            }
        }
        document_path::validate_update_paths(paths)
    }

}

pub(crate) use set_operator::SetOperator;
pub(crate) use inc_operator::IncOperator;
pub(crate) use mul_operator::MulOperator;
pub(crate) use rename_operator::RenameOperator;
pub(crate) use unset_operator::UnsetOperator;
pub(crate) use push_operator::PushOperator;
pub(crate) use pop_operator::PopOperator;
pub(crate) use min_operator::MinOperator;
pub(crate) use max_operator::MaxOperator;