Fix unseen closed shard notification

quickwit-oss · Mar 8, 2024 · 5ea63a6 · 5ea63a6
1 parent b290a10
commit 5ea63a6
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/quickwit/quickwit-ingest/src/ingest_v2/models.rs b/quickwit/quickwit-ingest/src/ingest_v2/models.rs
@@ -59,7 +59,9 @@ impl IngesterShard {
         now: Instant,
     ) -> Self {
         let shard_status = (shard_state, replication_position_inclusive.clone());
-        let (shard_status_tx, shard_status_rx) = watch::channel(shard_status);
+        let (shard_status_tx, mut shard_status_rx) = watch::channel(shard_status);
+        shard_status_rx.mark_changed();
+
         Self {
             shard_type: IngesterShardType::Primary { follower_id },
             shard_state,
@@ -79,7 +81,9 @@ impl IngesterShard {
         now: Instant,
     ) -> Self {
         let shard_status = (shard_state, replication_position_inclusive.clone());
-        let (shard_status_tx, shard_status_rx) = watch::channel(shard_status);
+        let (shard_status_tx, mut shard_status_rx) = watch::channel(shard_status);
+        shard_status_rx.mark_changed();
+
         Self {
             shard_type: IngesterShardType::Replica { leader_id },
             shard_state,
@@ -98,7 +102,9 @@ impl IngesterShard {
         now: Instant,
     ) -> Self {
         let shard_status = (shard_state, replication_position_inclusive.clone());
-        let (shard_status_tx, shard_status_rx) = watch::channel(shard_status);
+        let (shard_status_tx, mut shard_status_rx) = watch::channel(shard_status);
+        shard_status_rx.mark_changed();
+
         Self {
             shard_type: IngesterShardType::Solo,
             shard_state,