Move thread_mem_usage module into measure/

core/src/banking_stage.rs

@@ -6,7 +6,7 @@ use crate::{
     packet::{limited_deserialize, Packet, Packets, PACKETS_PER_BATCH},
     poh_recorder::{PohRecorder, PohRecorderError, WorkingBankEntry},
     poh_service::PohService,
-    thread_mem_usage,
+    result::{Error, Result},
 };
 use crossbeam_channel::{Receiver as CrossbeamReceiver, RecvTimeoutError};
 use itertools::Itertools;
@@ -16,7 +16,7 @@ use solana_ledger::{
     entry::hash_transactions,
     leader_schedule_cache::LeaderScheduleCache,
 };
-use solana_measure::measure::Measure;
+use solana_measure::{measure::Measure, thread_mem_usage};
 use solana_metrics::{inc_new_counter_debug, inc_new_counter_info, inc_new_counter_warn};
 use solana_perf::{cuda_runtime::PinnedVec, perf_libs};
 use solana_runtime::{

core/src/cluster_info.rs

@@ -24,7 +24,6 @@ use crate::{
     repair_service::RepairType,
     result::{Error, Result},
     sendmmsg::{multicast, send_mmsg},
-    thread_mem_usage,
     weighted_shuffle::{weighted_best, weighted_shuffle},
 };
 use bincode::{serialize, serialized_size};
@@ -32,6 +31,7 @@ use core::cmp;
 use itertools::Itertools;
 use rand::{thread_rng, Rng};
 use solana_ledger::{bank_forks::BankForks, blocktree::Blocktree, staking_utils};
+use solana_measure::thread_mem_usage;
 use solana_metrics::{datapoint_debug, inc_new_counter_debug, inc_new_counter_error};
 use solana_net_utils::{
     bind_common, bind_common_in_range, bind_in_range, find_available_port_in_range,

core/src/fetch_stage.rs

@@ -5,7 +5,7 @@ use crate::packet::PacketsRecycler;
 use crate::poh_recorder::PohRecorder;
 use crate::result::{Error, Result};
 use crate::streamer::{self, PacketReceiver, PacketSender};
-use crate::thread_mem_usage;
+use solana_measure::thread_mem_usage;
 use solana_metrics::{inc_new_counter_debug, inc_new_counter_info};
 use solana_perf::recycler::Recycler;
 use solana_sdk::clock::DEFAULT_TICKS_PER_SLOT;

core/src/lib.rs

@@ -12,7 +12,6 @@ pub mod chacha_cuda;
 pub mod cluster_info_vote_listener;
 pub mod commitment;
 pub mod shred_fetch_stage;
-pub mod thread_mem_usage;
 #[macro_use]
 pub mod contact_info;
 pub mod archiver;
@@ -84,10 +83,3 @@ extern crate solana_metrics;
 #[cfg(test)]
 #[macro_use]
 extern crate matches;
-
-#[cfg(unix)]
-extern crate jemallocator;
-
-#[cfg(unix)]
-#[global_allocator]
-static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;

core/src/replay_stage.rs

@@ -7,7 +7,6 @@ use crate::{
     poh_recorder::PohRecorder,
     result::{Error, Result},
     rpc_subscriptions::RpcSubscriptions,
-    thread_mem_usage,
 };
 use solana_ledger::entry::EntryVerificationStatus;
 use solana_ledger::{
@@ -20,6 +19,7 @@ use solana_ledger::{
     snapshot_package::SnapshotPackageSender,
 };
 use solana_measure::measure::Measure;
+use solana_measure::thread_mem_usage;
 use solana_metrics::inc_new_counter_info;
 use solana_runtime::bank::Bank;
 use solana_sdk::{

core/src/streamer.rs

@@ -3,7 +3,8 @@
 
 use crate::packet::{self, send_to, Packets, PacketsRecycler, PACKETS_PER_BATCH};
 use crate::recvmmsg::NUM_RCVMMSGS;
-use crate::thread_mem_usage;
+use crate::result::{Error, Result};
+use solana_measure::thread_mem_usage;
 use solana_sdk::timing::duration_as_ms;
 use std::net::UdpSocket;
 use std::sync::atomic::{AtomicBool, Ordering};

core/src/thread_mem_usage.rs

@@ -1,39 +0,0 @@
-#[cfg(unix)]
-use jemalloc_ctl::thread;
-
-pub fn datapoint(_name: &'static str) {
-    #[cfg(unix)]
-    {
-        let allocated = thread::allocatedp::mib().unwrap();
-        let allocated = allocated.read().unwrap();
-        let mem = allocated.get();
-        solana_metrics::datapoint_debug!("thread-memory", (_name, mem as i64, i64));
-    }
-}
-
-pub struct Allocatedp {
-    #[cfg(unix)]
-    allocated: thread::ThreadLocal<u64>,
-}
-
-impl Allocatedp {
-    pub fn default() -> Self {
-        #[cfg(unix)]
-        {
-            let allocated = thread::allocatedp::mib().unwrap();
-            let allocated = allocated.read().unwrap();
-            Self { allocated }
-        }
-        #[cfg(not(unix))]
-        Self {}
-    }
-
-    pub fn get(&self) -> u64 {
-        #[cfg(unix)]
-        {
-            self.allocated.get()
-        }
-        #[cfg(not(unix))]
-        0
-    }
-}