Add ResultSetTableToken::row.

Signed-off-by: ienkovich <[email protected]>

Author: ienkovich

omniscidb/ResultSetRegistry/ResultSetTableToken.cpp

@@ -67,6 +67,19 @@ std::shared_ptr<arrow::Table> ResultSetTableToken::toArrow() const {
   return res;
 }
 
+std::vector<TargetValue> ResultSetTableToken::row(size_t row_idx,
+                                                  bool translate_strings,
+                                                  bool decimal_to_double) const {
+  for (size_t rs_idx = 0; rs_idx < resultSetCount(); ++rs_idx) {
+    auto rs = resultSet(rs_idx);
+    if (rs->rowCount() > row_idx) {
+      return rs->getRowAt(row_idx, translate_strings, decimal_to_double);
+    }
+    row_idx -= rs->rowCount();
+  }
+  throw std::runtime_error("Out-of-bound row index.");
+}
+
 std::string ResultSetTableToken::description() const {
   auto first_rs = resultSet(0);
   auto last_rs = resultSet(resultSetCount() - 1);

omniscidb/ResultSetRegistry/ResultSetTableToken.h

@@ -59,6 +59,10 @@ class ResultSetTableToken : public std::enable_shared_from_this<ResultSetTableTo
 
   std::shared_ptr<arrow::Table> toArrow() const;
 
+  std::vector<TargetValue> row(size_t row_idx,
+                               bool translate_strings,
+                               bool decimal_to_double) const;
+
   std::string toString() const {
     return "ResultSetTableToken(" + std::to_string(dbId()) + ":" +
            std::to_string(tableId()) + ")";

python/pyhdk/_sql.pxd

@@ -12,7 +12,7 @@ from pyarrow.lib cimport CTable as CArrowTable
 
 from pyhdk._common cimport CConfig, CType
 from pyhdk._storage cimport CSchemaProvider, CSchemaProviderPtr, CDataProvider, CDataMgr, CBufferProvider
-from pyhdk._execute cimport CExecutor, CResultSetPtr, CCompilationOptions, CExecutionOptions, CTargetMetaInfo
+from pyhdk._execute cimport CExecutor, CResultSetPtr, CCompilationOptions, CExecutionOptions, CTargetMetaInfo, CTargetValue
 
 cdef extern from "omniscidb/QueryEngine/ExtensionFunctionsWhitelist.h":
   cdef cppclass CExtensionFunction "ExtensionFunction":
@@ -60,6 +60,8 @@ cdef extern from "omniscidb/ResultSetRegistry/ResultSetTableToken.h":
     string memoryDescription()
     string contentToString(bool)
 
+    vector[CTargetValue] row(size_t, bool, bool) except +
+
 ctypedef shared_ptr[const CResultSetTableToken] CResultSetTableTokenPtr
 
 cdef extern from "omniscidb/QueryEngine/Descriptors/RelAlgExecutionDescriptor.h":
@@ -68,7 +70,6 @@ cdef extern from "omniscidb/QueryEngine/Descriptors/RelAlgExecutionDescriptor.h"
     CExecutionResult(const CExecutionResult&)
     CExecutionResult(CExecutionResult&&)
 
-    const CResultSetPtr& getRows()
     const vector[CTargetMetaInfo]& getTargetsMeta()
     string getExplanation()
     const string& tableName()

python/pyhdk/_sql.pyx

@@ -109,7 +109,7 @@ cdef class ExecutionResult:
 
   def row(self, row_id):
     res = []
-    cdef vector[CTargetValue] vals = self.c_result.getRows().get().getRowAt(row_id, True, True)
+    cdef vector[CTargetValue] vals = self.c_result.getToken().get().row(row_id, True, True)
     cdef vector[CTargetValue].const_iterator it = vals.const_begin()
     cdef const CScalarTargetValue *scalar
     cdef const CArrayTargetValue *array