From bc7983b13e5142082445153804994be06e7106ab Mon Sep 17 00:00:00 2001
From: "Jeff Washington (jwash)"
 <75863576+jeffwashington@users.noreply.github.com>
Date: Mon, 20 Sep 2021 09:44:40 -0500
Subject: [PATCH] AcctIdx: metric for active bg threads (#20026)

---
 runtime/src/accounts_index_storage.rs  | 2 ++
 runtime/src/bucket_map_holder_stats.rs | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/runtime/src/accounts_index_storage.rs b/runtime/src/accounts_index_storage.rs
index aeafffe56e..0f31d3ef8c 100644
--- a/runtime/src/accounts_index_storage.rs
+++ b/runtime/src/accounts_index_storage.rs
@@ -109,11 +109,13 @@ impl<T: IndexValue> AccountsIndexStorage<T> {
                 break;
             }
 
+            storage.stats.active_threads.fetch_add(1, Ordering::Relaxed);
             for bucket in &in_mem {
                 bucket.flush();
             }
 
             storage.stats.report_stats();
+            storage.stats.active_threads.fetch_sub(1, Ordering::Relaxed);
         }
     }
 }
diff --git a/runtime/src/bucket_map_holder_stats.rs b/runtime/src/bucket_map_holder_stats.rs
index e9f937e3f9..a48a2b0faa 100644
--- a/runtime/src/bucket_map_holder_stats.rs
+++ b/runtime/src/bucket_map_holder_stats.rs
@@ -19,6 +19,7 @@ pub struct BucketMapHolderStats {
     pub inserts: AtomicU64,
     pub count_in_mem: AtomicU64,
     pub per_bucket_count: Vec<AtomicU64>,
+    pub active_threads: AtomicU64,
     pub get_range_us: AtomicU64,
     last_time: AtomicInterval,
 }
@@ -125,6 +126,11 @@ impl BucketMapHolderStats {
             ),
             ("inserts", self.inserts.swap(0, Ordering::Relaxed), i64),
             ("deletes", self.deletes.swap(0, Ordering::Relaxed), i64),
+            (
+                "active_threads",
+                self.active_threads.load(Ordering::Relaxed),
+                i64
+            ),
             ("items", self.items.swap(0, Ordering::Relaxed), i64),
             ("keys", self.keys.swap(0, Ordering::Relaxed), i64),
         );