Don't pass expired failure injections to the child process (#17004)

1 files changed, 15 insertions, 13 deletions

diff --git a/ydb/library/yql/providers/dq/task_runner/tasks_runner_pipe.cpp b/ydb/library/yql/providers/dq/task_runner/tasks_runner_pipe.cpp
index f1942a2a457..07d9c02a57a 100644
--- a/ydb/library/yql/providers/dq/task_runner/tasks_runner_pipe.cpp
+++ b/ydb/library/yql/providers/dq/task_runner/tasks_runner_pipe.cpp
@@ -585,7 +585,7 @@ public:
         if (protocolVersion <= 1) {
             return std::numeric_limits<i64>::max();
         }
-        
+
         if (protocolVersion < 6) {
             NDqProto::TCommandHeader header;
             header.SetVersion(2);
@@ -614,7 +614,7 @@ public:
         header.SetChannelId(ChannelId);
         header.Save(&Output);
 
-        i64 written = 0; 
+        i64 written = 0;
         TCountingOutput countingOutput(&Output);
         data.Proto.Save(&countingOutput);
         if (data.IsOOB()) {
@@ -1175,7 +1175,7 @@ public:
     const TDqOutputStats& GetPushStats() const override {
         return PushStats;
     }
-    
+
     const TDqAsyncOutputBufferStats& GetPopStats() const override {
         return PopStats;
     }
@@ -1386,15 +1386,17 @@ public:
 
         auto state = TFailureInjector::GetCurrentState();
         for (auto& [k, v]: state) {
-            NDqProto::TCommandHeader header;
-            header.SetVersion(1);
-            header.SetCommand(NDqProto::TCommandHeader::CONFIGURE_FAILURE_INJECTOR);
-            header.Save(&Output);
-            NYql::NDqProto::TConfigureFailureInjectorRequest request;
-            request.SetName(k);
-            request.SetSkip(v.Skip);
-            request.SetFail(v.CountOfFails);
-            request.Save(&Output);
+            if (v.CountOfFails) {
+                NDqProto::TCommandHeader header;
+                header.SetVersion(1);
+                header.SetCommand(NDqProto::TCommandHeader::CONFIGURE_FAILURE_INJECTOR);
+                header.Save(&Output);
+                NYql::NDqProto::TConfigureFailureInjectorRequest request;
+                request.SetName(k);
+                request.SetSkip(v.Skip);
+                request.SetFail(v.CountOfFails);
+                request.Save(&Output);
+            }
         }
 
         return ret;
@@ -1542,7 +1544,7 @@ private:
 
         {
             auto guard = BindAllocator({});
-            ProgramNode = DeserializeRuntimeNode(Task.GetProgram().GetRaw(), GetTypeEnv()); 
+            ProgramNode = DeserializeRuntimeNode(Task.GetProgram().GetRaw(), GetTypeEnv());
         }
 
         auto& programStruct = static_cast<TStructLiteral&>(*ProgramNode.GetNode());