Support result set extract from TDataQueryResult (#1388)

2 files changed, 5 insertions, 0 deletions

diff --git a/ydb/public/sdk/cpp/client/ydb_table/table.cpp b/ydb/public/sdk/cpp/client/ydb_table/table.cpp
index 2cf00b8ad6..647cae8973 100644
--- 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
index 3c5b3db53f..ee1a176ee7 100644
--- a/ydb/public/sdk/cpp/client/ydb_table/table.h
+++ b/ydb/public/sdk/cpp/client/ydb_table/table.h
@@ -1818,6 +1818,7 @@ public:
         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;