Add get_type_sizes to CodeStats

Author: blyxyas

compiler/rustc_session/src/code_stats.rs

@@ -2,7 +2,7 @@ use std::cmp;
 
 use rustc_abi::{Align, Size};
 use rustc_data_structures::fx::FxHashSet;
-use rustc_data_structures::sync::Lock;
+use rustc_data_structures::sync::{Lock, LockGuard};
 use rustc_span::Symbol;
 
 #[derive(Clone, PartialEq, Eq, Hash, Debug)]
@@ -106,6 +106,13 @@ impl CodeStats {
         self.type_sizes.borrow_mut().insert(info);
     }
 
+    /// Get type sizes if possible, returns `None` if internal lock is locked,
+    /// or couldn't be locked
+    #[inline]
+    pub fn get_type_sizes(&self) -> Option<LockGuard<'_, FxHashSet<TypeSizeInfo>>> {
+        self.type_sizes.try_lock()
+    }
+
     pub fn print_type_sizes(&self) {
         let type_sizes = self.type_sizes.borrow();
         // We will soon sort, so the initial order does not matter.