Merge b44dd91 into b0abac7

Author: GrigoriyPA

ydb/core/kqp/proxy_service/kqp_script_executions.cpp

@@ -1056,6 +1056,8 @@ class TForgetScriptExecutionOperationActor : public TActorBootstrapped<TForgetSc
 };
 
 class TGetScriptExecutionOperationQueryActor : public TQueryBase {
+    static constexpr i32 MAX_RESPONSE_SIZE_BYTES = 50_MB;
+
 public:
     TGetScriptExecutionOperationQueryActor(const TString& database, const TString& executionId)
         : TQueryBase(__func__, executionId)
@@ -1182,6 +1184,7 @@ class TGetScriptExecutionOperationQueryActor : public TQueryBase {
 
         const TMaybe<TString> issuesSerialized = result.ColumnParser("issues").GetOptionalJsonDocument();
         if (issuesSerialized) {
+            ResponseSizeBytes += issuesSerialized->Size();
             Issues = DeserializeIssues(*issuesSerialized);
         }
 
@@ -1231,13 +1234,31 @@ class TGetScriptExecutionOperationQueryActor : public TQueryBase {
     }
 
     void OnFinish(Ydb::StatusIds::StatusCode status, NYql::TIssues&& issues) override {
+        ResponseSizeBytes += Metadata.ByteSizeLong();
+        CheckResponseSize();
+
         if (OperationStatus) {
             Send(Owner, new TEvGetScriptExecutionOperationQueryResponse(true, LeaseExpired, FinalizationStatus, RunScriptActorId, ExecutionId, *OperationStatus, std::move(Issues), std::move(Metadata)));
         } else {
             Send(Owner, new TEvGetScriptExecutionOperationQueryResponse(false, LeaseExpired, FinalizationStatus, RunScriptActorId, ExecutionId, status, std::move(issues), std::move(Metadata)));
         }
     }
 
+private:
+    void CheckResponseSize() {
+        if (ResponseSizeBytes <= MAX_RESPONSE_SIZE_BYTES) {
+            return;
+        }
+
+        NYql::TIssue astTruncatedIssue(TStringBuilder() << "Query ast was truncated, because get operation response size is too large");
+        astTruncatedIssue.SetCode(NYql::DEFAULT_ERROR, NYql::TSeverityIds::S_INFO);
+        Issues.AddIssue(astTruncatedIssue);
+
+        auto* ast = Metadata.mutable_exec_stats()->mutable_query_ast();
+        ui64 toRemove = std::min(ResponseSizeBytes - MAX_RESPONSE_SIZE_BYTES, ast->size());
+        *ast = ast->substr(0, ast->size() - toRemove) + "...\n(TRUNCATED)";
+    }
+
 private:
     TString Database;
     TString ExecutionId;
@@ -1248,6 +1269,7 @@ class TGetScriptExecutionOperationQueryActor : public TQueryBase {
     TActorId RunScriptActorId;
     NYql::TIssues Issues;
     Ydb::Query::ExecuteScriptMetadata Metadata;
+    ui64 ResponseSizeBytes = 0;
 };
 
 class TGetScriptExecutionOperationActor : public TCheckLeaseStatusActorBase {