resource_metering: free leaked thread CPU stats (#15373)

close #15304 resource_metering: free leaked thread CPU stats Signed-off-by: Neil Shen <overvenus@gmail.com> Co-authored-by: ti-chi-bot[bot] <108142056+ti-chi-bot[bot]@users.noreply.github.com>
tikv · Aug 21, 2023 · e59ad22 · e59ad22
1 parent 2c5e7eb
commit e59ad22
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/components/resource_metering/src/recorder/sub_recorder/cpu.rs b/components/resource_metering/src/recorder/sub_recorder/cpu.rs
@@ -49,10 +49,13 @@ impl SubRecorder for CpuRecorder {
     fn cleanup(
         &mut self,
         _records: &mut RawRecords,
-        _thread_stores: &mut HashMap<Pid, LocalStorage>,
+        thread_stores: &mut HashMap<Pid, LocalStorage>,
     ) {
-        const THREAD_STAT_LEN_THRESHOLD: usize = 500;
+        // Remove thread stats that are no longer in thread_stores.
+        self.thread_stats
+            .retain(|tid, _| thread_stores.contains_key(tid));
 
+        const THREAD_STAT_LEN_THRESHOLD: usize = 500;
         if self.thread_stats.capacity() > THREAD_STAT_LEN_THRESHOLD
             && self.thread_stats.len() < THREAD_STAT_LEN_THRESHOLD / 2
         {