produce inputRange

arangod/Aql/EnumerateListExecutor.cpp

@@ -25,6 +25,8 @@
 
 #include "EnumerateListExecutor.h"
 
+#include "Aql/AqlCall.h"
+#include "Aql/AqlItemBlockInputRange.h"
 #include "Aql/AqlValue.h"
 #include "Aql/ExecutorInfos.h"
 #include "Aql/InputAqlItemRow.h"
@@ -45,7 +47,7 @@ void throwArrayExpectedException(AqlValue const& value) {
               " as operand to FOR loop; you provided a value of type '") +
           value.getTypeString() + std::string("'"));
 }
-} // namespace
+}  // namespace
 
 constexpr bool EnumerateListExecutor::Properties::preservesOrder;
 constexpr BlockPassthrough EnumerateListExecutor::Properties::allowsBlockPassthrough;
@@ -154,6 +156,60 @@ std::pair<ExecutionState, NoStats> EnumerateListExecutor::produceRows(OutputAqlI
   }
 }
 
+std::tuple<ExecutorState, NoStats, AqlCall> EnumerateListExecutor::produceRows(
+    size_t limit, AqlItemBlockInputRange& inputRange, OutputAqlItemRow& output) {
+  while (inputRange.hasMore() && limit > 0) {
+    initialize();
+    auto const& [state, input] = inputRange.next();
+    TRI_ASSERT(input.isInitialized());
+    // HIT in first run, because pos and length are initiliazed
+    // both with 0
+
+    AqlValue const& inputList = input.getValue(_infos.getInputRegister());
+
+    while (true) {
+      if (_inputArrayPosition == 0) {
+        // store the length into a local variable
+        // so we don't need to calculate length every time
+        if (inputList.isDocvec()) {
+          _inputArrayLength = inputList.docvecSize();
+        } else {
+          if (!inputList.isArray()) {
+            throwArrayExpectedException(inputList);
+          }
+          _inputArrayLength = inputList.length();
+        }
+      }
+
+      if (_inputArrayLength == 0) {
+        continue;
+      } else if (_inputArrayLength == _inputArrayPosition) {
+        // we reached the end, forget all state
+        break;
+      } else {
+        bool mustDestroy;
+        AqlValue innerValue = getAqlValue(inputList, _inputArrayPosition, mustDestroy);
+        AqlValueGuard guard(innerValue, mustDestroy);
+
+        TRI_IF_FAILURE("EnumerateListBlock::getSome") {
+          THROW_ARANGO_EXCEPTION(TRI_ERROR_DEBUG);
+        }
+
+        output.moveValueInto(_infos.getOutputRegister(), input, guard);
+        output.advanceRow();
+        limit--;
+
+        // set position to +1 for next iteration after new fetchRow
+        _inputArrayPosition++;
+      }
+    }
+  }
+
+  AqlCall upstreamCall{};
+  upstreamCall.softLimit = limit;
+  return {inputRange.peek().first, NoStats{}, upstreamCall};
+}
+
 void EnumerateListExecutor::initialize() {
   _inputArrayLength = 0;
   _inputArrayPosition = 0;

arangod/Aql/EnumerateListExecutor.h

@@ -41,6 +41,8 @@ class Methods;
 
 namespace aql {
 
+struct AqlCall;
+class AqlItemBlockInputRange;
 class ExecutorInfos;
 class OutputAqlItemRow;
 class NoStats;
@@ -95,6 +97,15 @@ class EnumerateListExecutor {
    */
   std::pair<ExecutionState, Stats> produceRows(OutputAqlItemRow& output);
 
+  /**
+   * @brief produce the next Row of Aql Values.
+   *
+   * @return ExecutorState, the stats, and a new Call that needs to be send to upstream
+   */
+  std::tuple<ExecutorState, Stats, AqlCall> produceRows(size_t atMost,
+                                                        AqlItemBlockInputRange& input,
+                                                        OutputAqlItemRow& output);
+
  private:
   AqlValue getAqlValue(AqlValue const& inVarReg, size_t const& pos, bool& mustDestroy);
   void initialize();