Group row indices by partition key, preserving first-seen partition order.
(
rows: &[(String, serde_json::Value)],
partition_by: &[SqlExpr],
)
| 8 | |
| 9 | /// Group row indices by partition key, preserving first-seen partition order. |
| 10 | pub(super) fn build_partitions( |
| 11 | rows: &[(String, serde_json::Value)], |
| 12 | partition_by: &[SqlExpr], |
| 13 | ) -> Vec<Vec<usize>> { |
| 14 | if partition_by.is_empty() { |
| 15 | return vec![(0..rows.len()).collect()]; |
| 16 | } |
| 17 | |
| 18 | let mut groups: HashMap<String, Vec<usize>> = HashMap::new(); |
| 19 | let mut order = Vec::new(); |
| 20 | |
| 21 | for (i, (_id, doc)) in rows.iter().enumerate() { |
| 22 | let key: String = partition_by |
| 23 | .iter() |
| 24 | .map(|expr| eval_expr_on_json(expr, doc).to_string()) |
| 25 | .collect::<Vec<_>>() |
| 26 | .join("\x00"); |
| 27 | let entry = groups.entry(key.clone()).or_default(); |
| 28 | if entry.is_empty() { |
| 29 | order.push(key); |
| 30 | } |
| 31 | entry.push(i); |
| 32 | } |
| 33 | |
| 34 | order.iter().filter_map(|k| groups.remove(k)).collect() |
| 35 | } |
| 36 | |
| 37 | pub(super) fn set_window_col(row: &mut serde_json::Value, alias: &str, val: serde_json::Value) { |
| 38 | if let serde_json::Value::Object(map) = row { |