Support result set extract from TDataQueryResult (ydb-platform#1388)

molotkov-and · Jan 29, 2024 · 8a154cb · 8a154cb
1 parent 6aac9f0
commit 8a154cb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ydb/public/sdk/cpp/client/ydb_table/table.cpp b/ydb/public/sdk/cpp/client/ydb_table/table.cpp
@@ -2084,6 +2084,10 @@ const TVector<TResultSet>& TDataQueryResult::GetResultSets() const {
     return ResultSets_;
 }
 
+TVector<TResultSet> TDataQueryResult::ExtractResultSets() && {
+    return std::move(ResultSets_);
+}
+
 TResultSet TDataQueryResult::GetResultSet(size_t resultIndex) const {
     if (resultIndex >= ResultSets_.size()) {
         RaiseError(TString("Requested index out of range\n"));

diff --git a/ydb/public/sdk/cpp/client/ydb_table/table.h b/ydb/public/sdk/cpp/client/ydb_table/table.h
@@ -1818,6 +1818,7 @@ class TDataQueryResult : public TStatus {
         const TMaybe<TDataQuery>& dataQuery, bool fromCache, const TMaybe<TQueryStats>& queryStats);
 
     const TVector<TResultSet>& GetResultSets() const;
+    TVector<TResultSet> ExtractResultSets() &&;
     TResultSet GetResultSet(size_t resultIndex) const;
 
     TResultSetParser GetResultSetParser(size_t resultIndex) const;