rbatis_codegen/codegen/syntax_tree_pysql/
foreach_node.rs

1use crate::codegen::syntax_tree_pysql::{Name, NodeType, ToHtml};
2
3/// Represents a `for` loop node in py_sql.
4/// It iterates over a collection and executes the nested SQL block for each item.
5/// 
6/// # Attributes
7/// 
8/// - `collection`: The expression providing the collection to iterate over.
9/// - `item`: The name of the variable to hold the current item in each iteration.
10/// - `index`: The name of the variable to hold the current key/index in each iteration.
11/// 
12/// # Example
13/// 
14/// PySQL syntax:
15/// ```py
16/// for item in ids:
17///   AND id = #{item}
18/// 
19/// for key, item in user_map:
20///   (#{key}, #{item.name})
21/// ```
22#[derive(Clone, Debug, Eq, PartialEq)]
23pub struct ForEachNode {
24    pub childs: Vec<NodeType>,
25    pub collection: String,
26    pub index: String,
27    pub item: String,
28}
29
30impl Name for ForEachNode {
31    fn name() -> &'static str {
32        "for"
33    }
34}
35
36
37impl ToHtml for ForEachNode {
38    fn as_html(&self) -> String {
39        let mut childs = String::new();
40        for x in &self.childs {
41            childs.push_str(&x.as_html());
42        }
43        format!(
44            "<foreach collection=\"{}\" index=\"{}\" item=\"{}\" >{}</foreach>",
45            self.collection, self.index, self.item, childs
46        )
47    }
48}