add custom demangler

Signed-off-by: YangKeao <[email protected]>

Author: YangKeao

src/frames.rs

@@ -104,6 +104,10 @@ impl Symbol {
     }
 
     pub fn name(&self) -> String {
+        self.name_with_demangle(demangle)
+    }
+
+    pub fn name_with_demangle<T>(&self, demangle: T) -> String where T: Fn(&str)->Cow<str> + 'static {
         demangle(&String::from_utf8_lossy(self.raw_name())).into_owned()
     }
 

src/report.rs

@@ -1,9 +1,11 @@
 // Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
 
+use std::borrow::Cow;
 use std::collections::HashMap;
 use std::fmt::{Debug, Formatter};
 
 use parking_lot::RwLock;
+use symbolic_demangle::demangle;
 
 use crate::frames::{Frames, UnresolvedFrames};
 use crate::profiler::Profiler;
@@ -32,6 +34,8 @@ pub struct UnresolvedReport {
 /// A builder of `Report` and `UnresolvedReport`. It builds report from a running `Profiler`.
 pub struct ReportBuilder<'a> {
     frames_post_processor: Option<Box<dyn Fn(&mut Frames)>>,
+    demangle: Box<dyn Fn(&str)->Cow<str>>,
+
     profiler: &'a RwLock<Result<Profiler>>,
     timing: ReportTiming,
 }
@@ -40,6 +44,8 @@ impl<'a> ReportBuilder<'a> {
     pub(crate) fn new(profiler: &'a RwLock<Result<Profiler>>, timing: ReportTiming) -> Self {
         Self {
             frames_post_processor: None,
+            demangle: Box::new(demangle),
+
             profiler,
             timing,
         }
@@ -57,6 +63,15 @@ impl<'a> ReportBuilder<'a> {
         self
     }
 
+    pub fn demangler<T, S>(&mut self, demangler: T) -> &mut Self
+    where
+        T: Fn(&str) -> Cow<str> + 'static,
+    {
+        self.demangle = Box::new(demangler);
+
+        self
+    }
+
     /// Build an `UnresolvedReport`
     pub fn build_unresolved(&self) -> Result<UnresolvedReport> {
         let mut hash_map = HashMap::new();
@@ -232,7 +247,7 @@ mod protobuf {
                 dedup_str.insert(key.thread_name_or_id());
                 for frame in key.frames.iter() {
                     for symbol in frame {
-                        dedup_str.insert(symbol.name());
+                        dedup_str.insert(symbol.name_with_demangle(&self.demangle));
                         dedup_str.insert(symbol.sys_name().into_owned());
                         dedup_str.insert(symbol.filename().into_owned());
                     }
@@ -260,7 +275,7 @@ mod protobuf {
                 let mut locs = vec![];
                 for frame in key.frames.iter() {
                     for symbol in frame {
-                        let name = symbol.name();
+                        let name = symbol.name_with_demangle(&self.demangle);
                         if let Some(loc_idx) = functions.get(&name) {
                             locs.push(*loc_idx);
                             continue;