From c98bfd62e226514bf3239bdee852bb1670da34be Mon Sep 17 00:00:00 2001
From: Remi Dettai <rdettai@gmail.com>
Date: Tue, 17 Sep 2024 15:24:28 +0200
Subject: [PATCH] Change strong_count to Weak

---
 .../src/source/queue_sources/shared_state.rs              | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/quickwit/quickwit-indexing/src/source/queue_sources/shared_state.rs b/quickwit/quickwit-indexing/src/source/queue_sources/shared_state.rs
index 1ddfea7b9a3..9322433edaf 100644
--- a/quickwit/quickwit-indexing/src/source/queue_sources/shared_state.rs
+++ b/quickwit/quickwit-indexing/src/source/queue_sources/shared_state.rs
@@ -18,7 +18,7 @@
 // along with this program. If not, see <http://www.gnu.org/licenses/>.
 
 use std::collections::BTreeMap;
-use std::sync::Arc;
+use std::sync::{Arc, Weak};
 use std::time::Duration;
 
 use anyhow::{bail, Context};
@@ -61,7 +61,7 @@ impl QueueSharedState {
             max_age,
             max_count,
             pruning_interval,
-            cleanup_handle.clone(),
+            Arc::downgrade(&cleanup_handle),
         ));
         Self {
             metastore,
@@ -77,7 +77,7 @@ impl QueueSharedState {
         max_age: Option<Duration>,
         max_count: Option<u32>,
         pruning_interval: Duration,
-        owner_handle: Arc<()>,
+        owner_handle: Weak<()>,
     ) {
         if max_count.is_none() && max_age.is_none() {
             return;
@@ -91,7 +91,7 @@ impl QueueSharedState {
             let mut interval = tokio::time::interval(pruning_interval);
             loop {
                 interval.tick().await;
-                if Arc::strong_count(&owner_handle) == 1 {
+                if owner_handle.upgrade().is_none() {
                     break;
                 }
                 let result: Result<_, _> = metastore