From 77d2ed95fff721f19375c8f5696df9c9f710d10f Mon Sep 17 00:00:00 2001
From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com>
Date: Thu, 4 Feb 2021 15:12:23 -0800
Subject: [PATCH] Add ref count from storage (#15078) (#15092)

(cherry picked from commit e5225b7e68c35099dc890d39f973ea856dbed415)

Co-authored-by: sakridge <sakridge@gmail.com>
---
 runtime/src/accounts_db.rs    | 2 +-
 runtime/src/accounts_index.rs | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/runtime/src/accounts_db.rs b/runtime/src/accounts_db.rs
index 432225bc27..799a83fbd2 100644
--- a/runtime/src/accounts_db.rs
+++ b/runtime/src/accounts_db.rs
@@ -4629,7 +4629,7 @@ impl AccountsDB {
         roots.sort();
         info!("{}: accounts_index roots: {:?}", label, roots,);
         for (pubkey, account_entry) in self.accounts_index.account_maps.read().unwrap().iter() {
-            info!("  key: {}", pubkey);
+            info!("  key: {} ref_count: {}", pubkey, account_entry.ref_count(),);
             info!(
                 "      slots: {:?}",
                 *account_entry.slot_list.read().unwrap()
diff --git a/runtime/src/accounts_index.rs b/runtime/src/accounts_index.rs
index 5143a2896b..14e50620c0 100644
--- a/runtime/src/accounts_index.rs
+++ b/runtime/src/accounts_index.rs
@@ -78,6 +78,12 @@ pub struct AccountMapEntryInner<T> {
     pub slot_list: RwLock<SlotList<T>>,
 }
 
+impl<T> AccountMapEntryInner<T> {
+    pub fn ref_count(&self) -> u64 {
+        self.ref_count.load(Ordering::Relaxed)
+    }
+}
+
 #[self_referencing]
 pub struct ReadAccountMapEntry<T: 'static> {
     owned_entry: AccountMapEntry<T>,