Graceful stop on "session close"

1 files changed, 3 insertions, 3 deletions

diff --git a/ydb/public/lib/ydb_cli/commands/topic_workload/topic_workload_writer.cpp b/ydb/public/lib/ydb_cli/commands/topic_workload/topic_workload_writer.cpp
index c75b437dfc..3cf13077c4 100644
--- a/ydb/public/lib/ydb_cli/commands/topic_workload/topic_workload_writer.cpp
+++ b/ydb/public/lib/ydb_cli/commands/topic_workload/topic_workload_writer.cpp
@@ -109,7 +109,7 @@ void TTopicWorkloadWriterWorker::Process() {
         } else
             WRITE_LOG(Params.Log, ELogPriority::TLOG_DEBUG, TStringBuilder() << "No WaitEvent");
 
-        while (true) {
+        while (!*Params.ErrorFlag) {
             auto events = WriteSession->GetEvents(false);
             WRITE_LOG(Params.Log, ELogPriority::TLOG_DEBUG, TStringBuilder() << "Got " << events.size() << " events.");
 
@@ -205,9 +205,9 @@ bool TTopicWorkloadWriterWorker::ProcessReadyToAcceptEvent(
 
 bool TTopicWorkloadWriterWorker::ProcessSessionClosedEvent(
     const NYdb::NTopic::TSessionClosedEvent& event) {
-    WRITE_LOG(Params.Log, ELogPriority::TLOG_INFO, TStringBuilder() << "Got close event: " << event.DebugString());
+    WRITE_LOG(Params.Log, ELogPriority::TLOG_EMERG, TStringBuilder() << "Got close event: " << event.DebugString());
     //! Session is closed, stop any work with it.
-    Y_FAIL("session closed");
+    *Params.ErrorFlag = 1;
     return false;
 }