Subscribe to pipe state (#5685)

ydb-platform · Jun 18, 2024 · 9b24426 · 9b24426
1 parent 40bcf99
commit 9b24426
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ydb/core/tx/replication/service/table_writer_impl.h b/ydb/core/tx/replication/service/table_writer_impl.h
@@ -100,7 +100,7 @@ class TTablePartitionWriter: public TActorBootstrapped<TTablePartitionWriter<TCh
             event->Record.SetSource(source);
         }
 
-        this->Send(LeaderPipeCache, new TEvPipeCache::TEvForward(event.Release(), TabletId, false));
+        this->Send(LeaderPipeCache, new TEvPipeCache::TEvForward(event.Release(), TabletId, true, ++SubscribeCookie));
         this->Become(&TThis::StateWaitingStatus);
     }
 
@@ -140,7 +140,7 @@ class TTablePartitionWriter: public TActorBootstrapped<TTablePartitionWriter<TCh
     }
 
     void Handle(TEvPipeCache::TEvDeliveryProblem::TPtr& ev) {
-        if (TabletId == ev->Get()->TabletId) {
+        if (TabletId == ev->Get()->TabletId && ev->Cookie == SubscribeCookie) {
             Leave();
         }
     }
@@ -194,7 +194,9 @@ class TTablePartitionWriter: public TActorBootstrapped<TTablePartitionWriter<TCh
     mutable TMaybe<TString> LogPrefix;
 
     TActorId LeaderPipeCache;
+    ui64 SubscribeCookie = 0;
     TChangeRecordBuilderContextTrait<TChangeRecord> BuilderContext;
+
 }; // TTablePartitionWriter
 
 template <class TChangeRecord>
@@ -421,7 +423,7 @@ class TLocalTableWriter
         return new TTablePartitionWriter<TChangeRecord>(this->SelfId(), partitionId, TTableId(this->PathId, Schema->Version));
     }
 
-    const TVector<TKeyDesc::TPartitionInfo>& GetPartitions() const override { return KeyDesc->GetPartitions(); };
+    const TVector<TKeyDesc::TPartitionInfo>& GetPartitions() const override { return KeyDesc->GetPartitions(); }
     const TVector<NScheme::TTypeInfo>& GetSchema() const override { return KeyDesc->KeyColumnTypes; }
     NKikimrSchemeOp::ECdcStreamFormat GetStreamFormat() const override { return TChangeRecord::StreamType; }