summaryrefslogtreecommitdiffstats
path: root/compiler/rustc_span/src/profiling.rs
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/rustc_span/src/profiling.rs')
-rw-r--r--compiler/rustc_span/src/profiling.rs16
1 files changed, 5 insertions, 11 deletions
diff --git a/compiler/rustc_span/src/profiling.rs b/compiler/rustc_span/src/profiling.rs
index 0ab890b9f..66e5369da 100644
--- a/compiler/rustc_span/src/profiling.rs
+++ b/compiler/rustc_span/src/profiling.rs
@@ -1,3 +1,5 @@
+use crate::source_map::SourceMap;
+
use std::borrow::Borrow;
use rustc_data_structures::profiling::EventArgRecorder;
@@ -11,25 +13,17 @@ pub trait SpannedEventArgRecorder {
///
/// Note: when self-profiling with costly event arguments, at least one argument
/// needs to be recorded. A panic will be triggered if that doesn't happen.
- fn record_arg_with_span<A>(&mut self, event_arg: A, span: crate::Span)
+ fn record_arg_with_span<A>(&mut self, source_map: &SourceMap, event_arg: A, span: crate::Span)
where
A: Borrow<str> + Into<String>;
}
impl SpannedEventArgRecorder for EventArgRecorder<'_> {
- fn record_arg_with_span<A>(&mut self, event_arg: A, span: crate::Span)
+ fn record_arg_with_span<A>(&mut self, source_map: &SourceMap, event_arg: A, span: crate::Span)
where
A: Borrow<str> + Into<String>,
{
self.record_arg(event_arg);
-
- let span_arg = crate::with_session_globals(|session_globals| {
- if let Some(source_map) = &*session_globals.source_map.borrow() {
- source_map.span_to_embeddable_string(span)
- } else {
- format!("{span:?}")
- }
- });
- self.record_arg(span_arg);
+ self.record_arg(source_map.span_to_embeddable_string(span));
}
}