Method to reply, finish stream and rpc actor.

ydb-platform · Jan 30, 2024 · 5dac868 · 5dac868
1 parent 02f3fa6
commit 5dac868
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ydb/core/grpc_services/query/rpc_execute_query.cpp b/ydb/core/grpc_services/query/rpc_execute_query.cpp
@@ -409,9 +409,7 @@ class TExecuteQueryRPC : public TActorBootstrapped<TExecuteQueryRPC> {
                 response.mutable_issues()->CopyFrom(issueMessage);
                 TString out;
                 Y_PROTOBUF_SUPPRESS_NODISCARD response.SerializeToString(&out);
-                const auto finishStreamFlag = NYdbGrpc::IRequestContextBase::EStreamCtrl::FINISH;
-                Request_->SendSerializedResult(std::move(out), record.GetYdbStatus(), finishStreamFlag);
-                this->PassAway();
+                ReplySerializedAndFinishStream(record.GetYdbStatus(), std::move(out));
             }
         }
 
@@ -434,6 +432,12 @@ class TExecuteQueryRPC : public TActorBootstrapped<TExecuteQueryRPC> {
         ReplyFinishStream(Ydb::StatusIds::INTERNAL_ERROR, issue);
     }
 
+    void ReplySerializedAndFinishStream(Ydb::StatusIds::StatusCode status, TString&& buf) {
+        const auto finishStreamFlag = NYdbGrpc::IRequestContextBase::EStreamCtrl::FINISH;
+        Request_->SendSerializedResult(std::move(buf), status, finishStreamFlag);
+        this->PassAway();
+    }
+
     void ReplyFinishStream(Ydb::StatusIds::StatusCode status, const NYql::TIssue& issue) {
         google::protobuf::RepeatedPtrField<TYdbIssueMessageType> issuesMessage;
         NYql::IssueToMessage(issue, issuesMessage.Add());