feat(cognitarium): handle query engine skip node

axone-protocol · Jun 5, 2023 · 65e09ee · 65e09ee
1 parent 9fa3b9d
commit 65e09ee
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
@@ -34,7 +34,12 @@ impl<'a> QueryEngine<'a> {
                 Box::new(move |_| Box::new(iter::empty()))
             }
             QueryNode::ForLoopJoin { left, right } => Box::new(move |_| Box::new(iter::empty())),
-            QueryNode::Skip { child, first } => Box::new(move |_| Box::new(iter::empty())),
+            QueryNode::Skip { child, first } => {
+                let upstream = self.eval_node(child);
+                Box::new(move |vars| -> ResolvedVariablesIterator {
+                    Box::new(upstream(vars).skip(first))
+                })
+            }
             QueryNode::Limit { child, first } => {
                 let upstream = self.eval_node(child);
                 Box::new(move |vars| -> ResolvedVariablesIterator {