feat(cognitarium): handle engine limit query node

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