mf_model/ops/
bitand.rs

1use std::ops::BitAnd;
2
3use crate::{
4    error::{error_helpers, PoolResult},
5};
6
7use super::{MarkRef, NodeRef};
8
9/// 为 NodeRef 实现自定义的 & 运算符，用于过滤特定类型的子节点
10/// 当使用 & 运算符时，会保留指定类型的子节点，移除其他类型的子节点
11impl<'a> BitAnd<String> for NodeRef<'a> {
12    type Output = PoolResult<NodeRef<'a>>;
13    fn bitand(
14        self,
15        node_type: String,
16    ) -> Self::Output {
17        let children =
18            self.tree.children(&self.key.clone()).unwrap_or_default();
19        let mut nodes_to_remove = Vec::new();
20
21        for child_id in children {
22            if let Some(node) = self.tree.get_node(&child_id) {
23                if node.r#type.to_string() != node_type {
24                    nodes_to_remove.push(child_id);
25                }
26            }
27        }
28
29        // 移除不匹配的节点
30        if !nodes_to_remove.is_empty() {
31            self.tree.remove_node(&self.key.clone(), nodes_to_remove)?;
32        }
33
34        Ok(NodeRef::new(self.tree, self.key.clone()))
35    }
36}
37
38/// 为 NodeRef 实现自定义的 & 运算符，用于保留多个指定类型的子节点
39/// 当使用 & 运算符时，会保留匹配任一指定类型的子节点，移除其他类型的子节点
40impl<'a> BitAnd<Vec<String>> for NodeRef<'a> {
41    type Output = PoolResult<NodeRef<'a>>;
42    fn bitand(
43        self,
44        node_types: Vec<String>,
45    ) -> Self::Output {
46        let children =
47            self.tree.children(&self.key.clone()).unwrap_or_default();
48        let mut nodes_to_remove = Vec::new();
49
50        for child_id in children {
51            if let Some(node) = self.tree.get_node(&child_id) {
52                let node_type_str = node.r#type.to_string();
53                if !node_types.contains(&node_type_str) {
54                    nodes_to_remove.push(child_id);
55                }
56            }
57        }
58
59        // 移除不匹配的节点
60        if !nodes_to_remove.is_empty() {
61            self.tree.remove_node(&self.key.clone(), nodes_to_remove)?;
62        }
63
64        Ok(NodeRef::new(self.tree, self.key.clone()))
65    }
66}
67
68/// 为 MarkRef 实现自定义的 & 运算符，用于保留指定名称的标记
69/// 当使用 & 运算符时，会保留指定名称的标记，移除其他标记
70impl<'a> BitAnd<String> for MarkRef<'a> {
71    type Output = PoolResult<MarkRef<'a>>;
72    fn bitand(
73        self,
74        mark_name: String,
75    ) -> Self::Output {
76        let noderef = self.tree.get_node(&self.key.clone());
77        match noderef {
78            Some(_) => {
79                self.tree.remove_mark(&self.key.clone(), &[mark_name])?;
80                Ok(MarkRef::new(self.tree, self.key.clone()))
81            },
82            None => Err(error_helpers::node_not_found(self.key.clone())),
83        }
84    }
85}
86
87/// 为 MarkRef 实现自定义的 & 运算符，用于保留多个指定名称的标记
88/// 当使用 & 运算符时，会保留匹配任一指定名称的标记，移除其他标记
89impl<'a> BitAnd<Vec<String>> for MarkRef<'a> {
90    type Output = PoolResult<MarkRef<'a>>;
91    fn bitand(
92        self,
93        mark_names: Vec<String>,
94    ) -> Self::Output {
95        let noderef = self.tree.get_node(&self.key.clone());
96        match noderef {
97            Some(_) => {
98                self.tree.remove_mark(&self.key.clone(), &mark_names)?;
99                Ok(MarkRef::new(self.tree, self.key.clone()))
100            },
101            None => Err(error_helpers::node_not_found(self.key.clone())),
102        }
103    }
104}
mf_model/ops/bitand.rs

mf_model/ops/
bitand.rs
