Moved commit "Replaced proto types to SDK types in ExecuteScript settings" from ydb repo (#257)

Author: Gazizonoki

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

@@ -2,8 +2,6 @@
 
 #include "stats.h"
 
-#include <ydb-cpp-sdk/src/api/protos/ydb_query.pb.h>
-
 #include <ydb-cpp-sdk/client/result/result.h>
 #include <ydb-cpp-sdk/client/types/fluent_settings_helpers.h>
 #include <ydb-cpp-sdk/client/types/operation/operation.h>
@@ -95,9 +93,9 @@ using TAsyncBeginTransactionResult = NThreading::TFuture<TBeginTransactionResult
 using TAsyncCommitTransactionResult = NThreading::TFuture<TCommitTransactionResult>;
 
 struct TExecuteScriptSettings : public TOperationRequestSettings<TExecuteScriptSettings> {
-    FLUENT_SETTING_DEFAULT(Ydb::Query::Syntax, Syntax, Ydb::Query::SYNTAX_YQL_V1);
-    FLUENT_SETTING_DEFAULT(Ydb::Query::ExecMode, ExecMode, Ydb::Query::EXEC_MODE_EXECUTE);
-    FLUENT_SETTING_DEFAULT(Ydb::Query::StatsMode, StatsMode, Ydb::Query::STATS_MODE_NONE);
+    FLUENT_SETTING_DEFAULT(ESyntax, Syntax, ESyntax::YqlV1);
+    FLUENT_SETTING_DEFAULT(EExecMode, ExecMode, EExecMode::Execute);
+    FLUENT_SETTING_DEFAULT(EStatsMode, StatsMode, EStatsMode::None);
     FLUENT_SETTING(TDuration, ResultsTtl);
 };
 
@@ -114,6 +112,21 @@ class TQueryContent {
     ESyntax Syntax = ESyntax::Unspecified;
 };
 
+class TResultSetMeta {
+public:
+    TResultSetMeta() = default;
+
+    explicit TResultSetMeta(const std::vector<TColumn>& columns)
+        : Columns(columns)
+    {}
+
+    explicit TResultSetMeta(std::vector<TColumn>&& columns)
+        : Columns(std::move(columns))
+    {}
+
+    std::vector<TColumn> Columns;
+};
+
 class TScriptExecutionOperation : public TOperation {
 public:
     struct TMetadata {
@@ -122,8 +135,8 @@ class TScriptExecutionOperation : public TOperation {
         EExecMode ExecMode = EExecMode::Unspecified;
 
         TQueryContent ScriptContent;
-        Ydb::TableStats::QueryStats ExecStats;
-        std::vector<Ydb::Query::ResultSetMeta> ResultSetsMeta;
+        TExecStats ExecStats;
+        std::vector<TResultSetMeta> ResultSetsMeta;
     };
 
     using TOperation::TOperation;

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

@@ -22,6 +22,8 @@ class TExecStats {
     friend class NYdb::TProtoAccessor;
 
 public:
+    TExecStats() = default;
+
     explicit TExecStats(Ydb::TableStats::QueryStats&& proto);
     explicit TExecStats(const Ydb::TableStats::QueryStats& proto);
 

src/client/query/client.cpp

@@ -87,9 +87,9 @@ class TQueryClient::TImpl: public TClientImplCommon<TQueryClient::TImpl>, public
     NThreading::TFuture<TScriptExecutionOperation> ExecuteScript(const std::string& script, const std::optional<TParams>& params, const TExecuteScriptSettings& settings) {
         using namespace Ydb::Query;
         auto request = MakeOperationRequest<ExecuteScriptRequest>(settings);
-        request.set_exec_mode(settings.ExecMode_);
-        request.set_stats_mode(settings.StatsMode_);
-        request.mutable_script_content()->set_syntax(settings.Syntax_);
+        request.set_exec_mode(::Ydb::Query::ExecMode(settings.ExecMode_));
+        request.set_stats_mode(::Ydb::Query::StatsMode(settings.StatsMode_));
+        request.mutable_script_content()->set_syntax(::Ydb::Query::Syntax(settings.Syntax_));
         request.mutable_script_content()->set_text(script);
         SetDuration(settings.ResultsTtl_, *request.mutable_results_ttl());
 

src/client/query/impl/client_session.h

@@ -3,6 +3,8 @@
 #include <ydb-cpp-sdk/client/query/client.h>
 #include <src/client/impl/ydb_internal/kqp_session_common/kqp_session_common.h>
 
+#include <src/api/grpc/ydb_query_v1.grpc.pb.h>
+
 namespace NYdb::NQuery {
 
 class TSession::TImpl : public TKqpSessionCommon {

src/client/query/query.cpp

@@ -1,5 +1,7 @@
 #include <ydb-cpp-sdk/client/query/query.h>
 
+#include <src/api/grpc/ydb_query_v1.grpc.pb.h>
+
 namespace NYdb::NQuery {
 
 std::optional<EStatsMode> ParseStatsMode(std::string_view statsMode) {
@@ -42,8 +44,17 @@ TScriptExecutionOperation::TScriptExecutionOperation(TStatus&& status, Ydb::Oper
     Metadata_.ExecutionId = metadata.execution_id();
     Metadata_.ExecMode = static_cast<EExecMode>(metadata.exec_mode());
     Metadata_.ExecStatus = static_cast<EExecStatus>(metadata.exec_status());
-    Metadata_.ExecStats = metadata.exec_stats();
-    Metadata_.ResultSetsMeta.insert(Metadata_.ResultSetsMeta.end(), metadata.result_sets_meta().begin(), metadata.result_sets_meta().end());
+    Metadata_.ExecStats = TExecStats(std::move(*metadata.mutable_exec_stats()));
+
+    Metadata_.ResultSetsMeta.reserve(metadata.result_sets_meta_size());
+    for (const auto& resultSetMeta : metadata.result_sets_meta()) {
+        std::vector<TColumn> columns;
+        columns.reserve(resultSetMeta.columns_size());
+        for (const auto& column : resultSetMeta.columns()) {
+            columns.emplace_back(column.name(), column.type());
+        }
+        Metadata_.ResultSetsMeta.emplace_back(std::move(columns)); 
+    }
 
     if (metadata.has_script_content()) {
         Metadata_.ScriptContent.Syntax = static_cast<ESyntax>(metadata.script_content().syntax());