Add dump methods to ResultSetTableToken.

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

Author: ienkovich

omniscidb/ResultSetRegistry/ResultSetTableToken.cpp

@@ -67,4 +67,57 @@ std::shared_ptr<arrow::Table> ResultSetTableToken::toArrow() const {
   return res;
 }
 
+std::string ResultSetTableToken::description() const {
+  auto first_rs = resultSet(0);
+  auto last_rs = resultSet(resultSetCount() - 1);
+  size_t total_entries = first_rs->entryCount();
+  for (size_t rs_idx = 1; rs_idx < resultSetCount(); ++rs_idx) {
+    total_entries += resultSet(rs_idx)->entryCount();
+  }
+  std::ostringstream oss;
+  oss << "Result Set Table Info" << std::endl;
+  oss << "\tFragments: " << resultSetCount() << std::endl;
+  oss << "\tLayout: " << first_rs->getQueryMemDesc().queryDescTypeToString() << std::endl;
+  oss << "\tColumns: " << first_rs->colCount() << std::endl;
+  oss << "\tRows: " << rowCount() << std::endl;
+  oss << "\tEntry count: " << total_entries << std::endl;
+  const std::string did_output_columnar =
+      first_rs->didOutputColumnar() ? "True" : "False;";
+  oss << "\tColumnar: " << did_output_columnar << std::endl;
+  oss << "\tLazy-fetched columns: " << first_rs->getNumColumnsLazyFetched() << std::endl;
+  const std::string is_direct_columnar_conversion_possible =
+      first_rs->isDirectColumnarConversionPossible() ? "True" : "False";
+  oss << "\tDirect columnar conversion possible: "
+      << is_direct_columnar_conversion_possible << std::endl;
+
+  size_t num_columns_zero_copy_columnarizable{0};
+  for (size_t col_idx = 0; col_idx < first_rs->colCount(); col_idx++) {
+    if (first_rs->isZeroCopyColumnarConversionPossible(col_idx)) {
+      num_columns_zero_copy_columnarizable++;
+    }
+  }
+  oss << "\tZero-copy columnar conversion columns: "
+      << num_columns_zero_copy_columnarizable << std::endl;
+
+  oss << "\tHas permutation: "
+      << (first_rs->isPermutationBufferEmpty() ? "False" : "True") << std::endl;
+  auto limit =
+      last_rs->getLimit() ? row_count_ - last_rs->rowCount() + last_rs->getLimit() : 0;
+  oss << "\tLimit: " << limit << std::endl;
+  oss << "\tOffset: " << first_rs->getOffset() << std::endl;
+  return oss.str();
+}
+
+std::string ResultSetTableToken::memoryDescription() const {
+  return resultSet(0)->toString();
+}
+
+std::string ResultSetTableToken::contentToString(bool header) const {
+  std::string res = resultSet(0)->contentToString(header);
+  for (size_t rs_idx = 1; rs_idx < resultSetCount(); ++rs_idx) {
+    res += resultSet(rs_idx)->contentToString(false);
+  }
+  return res;
+}
+
 }  // namespace hdk

omniscidb/ResultSetRegistry/ResultSetTableToken.h

@@ -64,6 +64,10 @@ class ResultSetTableToken : public std::enable_shared_from_this<ResultSetTableTo
            std::to_string(tableId()) + ")";
   }
 
+  std::string description() const;
+  std::string memoryDescription() const;
+  std::string contentToString(bool header) const;
+
  private:
   void reset();
 

python/pyhdk/_sql.pxd

@@ -56,6 +56,9 @@ cdef extern from "omniscidb/ResultSetRegistry/ResultSetTableToken.h":
   cdef cppclass CResultSetTableToken "hdk::ResultSetTableToken":
     size_t rowCount()
     shared_ptr[CArrowTable] toArrow() except +
+    string description()
+    string memoryDescription()
+    string contentToString(bool)
 
 ctypedef shared_ptr[const CResultSetTableToken] CResultSetTableTokenPtr
 

python/pyhdk/_sql.pyx

@@ -87,11 +87,13 @@ cdef class ExecutionResult:
 
   @property
   def desc(self):
-    return self.c_result.getRows().get().summaryToString()
+    cdef CResultSetTableTokenPtr c_token = self.c_result.getToken()
+    return c_token.get().description()
 
   @property
   def memory_desc(self):
-    return self.c_result.getRows().get().toString()
+    cdef CResultSetTableTokenPtr c_token = self.c_result.getToken()
+    return c_token.get().memoryDescription()
 
   @property
   def table_name(self):
@@ -152,7 +154,7 @@ cdef class ExecutionResult:
     for key, type_str in self.schema.items():
       res += f"  {key}: {type_str}\n"
     res += "Data:\n"
-    res += self.c_result.getRows().get().contentToString(False)
+    res += self.c_result.getToken().get().contentToString(False)
     return res
 
   def __repr__(self):