YQ-4235 supported runtime results for script executions (#17421)

Author: GrigoriyPA

.github/last_commit.txt

@@ -1 +1 @@
-ec36c2f7151e1e7718c2f0c0b2292383cab01e64
+04801c0ba7d7f9f55f7bd7681dd03a87c221d8c2

include/ydb-cpp-sdk/client/proto/accessor.h

@@ -1,5 +1,7 @@
 #pragma once
 
+#include <src/api/protos/draft/ydb_replication.pb.h>
+#include <src/api/protos/draft/ydb_view.pb.h>
 #include <src/api/protos/ydb_coordination.pb.h>
 #include <src/api/protos/ydb_export.pb.h>
 #include <src/api/protos/ydb_import.pb.h>
@@ -8,6 +10,8 @@
 #include <src/api/protos/ydb_topic.pb.h>
 #include <src/api/protos/ydb_value.pb.h>
 
+#include <ydb-cpp-sdk/client/draft/ydb_replication.h>
+#include <ydb-cpp-sdk/client/draft/ydb_view.h>
 #include <ydb-cpp-sdk/client/coordination/coordination.h>
 #include <ydb-cpp-sdk/client/export/export.h>
 #include <ydb-cpp-sdk/client/import/import.h>
@@ -45,6 +49,8 @@ class TProtoAccessor {
     static const Ydb::Topic::DescribeConsumerResult& GetProto(const NYdb::NTopic::TConsumerDescription& consumerDescription);
     static const Ydb::Monitoring::SelfCheckResult& GetProto(const NYdb::NMonitoring::TSelfCheckResult& selfCheckResult);
     static const Ydb::Coordination::DescribeNodeResult& GetProto(const NYdb::NCoordination::TNodeDescription &describeNodeResult);
+    static const Ydb::Replication::DescribeReplicationResult& GetProto(const NYdb::NReplication::TDescribeReplicationResult& desc);
+    static const Ydb::View::DescribeViewResult& GetProto(const NYdb::NView::TDescribeViewResult& desc);
 
     static NTable::TQueryStats FromProto(const Ydb::TableStats::QueryStats& queryStats);
     static NTable::TTableDescription FromProto(const Ydb::Table::CreateTableRequest& request);

include/ydb-cpp-sdk/client/query/query.h

@@ -124,15 +124,21 @@ class TResultSetMeta {
 public:
     TResultSetMeta() = default;
 
-    explicit TResultSetMeta(const std::vector<TColumn>& columns)
+    explicit TResultSetMeta(const std::vector<TColumn>& columns, uint64_t rowsCount = 0, bool finished = false)
         : Columns(columns)
+        , RowsCount(rowsCount)
+        , Finished(finished)
     {}
 
-    explicit TResultSetMeta(std::vector<TColumn>&& columns)
+    explicit TResultSetMeta(std::vector<TColumn>&& columns, uint64_t rowsCount = 0, bool finished = false)
         : Columns(std::move(columns))
+        , RowsCount(rowsCount)
+        , Finished(finished)
     {}
 
     std::vector<TColumn> Columns;
+    uint64_t RowsCount = 0;
+    bool Finished = false;
 };
 
 class TScriptExecutionOperation : public TOperation {

src/api/protos/ydb_query.proto

@@ -186,6 +186,8 @@ message ExecuteQueryRequest {
 
 message ResultSetMeta {
     repeated Ydb.Column columns = 1;
+    uint64 number_rows = 2;
+    bool finished = 3;
 }
 
 message ExecuteQueryResponsePart {

src/client/query/query.cpp

@@ -53,7 +53,7 @@ TScriptExecutionOperation::TScriptExecutionOperation(TStatus&& status, Ydb::Oper
         for (const auto& column : resultSetMeta.columns()) {
             columns.emplace_back(column.name(), column.type());
         }
-        Metadata_.ResultSetsMeta.emplace_back(std::move(columns)); 
+        Metadata_.ResultSetsMeta.emplace_back(std::move(columns), resultSetMeta.number_rows(), resultSetMeta.finished()); 
     }
 
     if (metadata.has_script_content()) {