feat(cognitarium): handle query engine for loop join

axone-protocol · Jun 5, 2023 · f9af316 · f9af316
1 parent 65e09ee
commit f9af316
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/contracts/okp4-cognitarium/src/querier/engine.rs b/contracts/okp4-cognitarium/src/querier/engine.rs
@@ -33,7 +33,13 @@ impl<'a> QueryEngine<'a> {
             QueryNode::CartesianProductJoin { left, right } => {
                 Box::new(move |_| Box::new(iter::empty()))
             }
-            QueryNode::ForLoopJoin { left, right } => Box::new(move |_| Box::new(iter::empty())),
+            QueryNode::ForLoopJoin { left, right } => {
+                let left = self.eval_node(left);
+                let right = self.eval_node(right);
+                Box::new(move |vars| -> ResolvedVariablesIterator {
+                    Box::new(left(vars).flat_map(move |v| right(v)))
+                })
+            }
             QueryNode::Skip { child, first } => {
                 let upstream = self.eval_node(child);
                 Box::new(move |vars| -> ResolvedVariablesIterator {