pyroscope 0.3.1-aarch64-linux → 0.3.2-aarch64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ee2c1dba441e9d5a2b56b8c43e1dd161ad3bfdcb8182f65162e048505e2c8caf
-  data.tar.gz: 276b3292df0609fdca21471d188a4a79b6aa7dcbbcbc3e8c64674d4c5d46621a
+  metadata.gz: d961eb1511f2da0db538f6d6667bdec6450fc7984d7cd8bc3ae22215b2fa47f8
+  data.tar.gz: 2f791e0636fc5515ce490db4516a02fe82baf50341e1c9c8199a949385a65320
 SHA512:
-  metadata.gz: fcd551822fb88c6959bbb15421730af6f103c291c62e39359acf4236e1aff57a0393c3156d86071fd066dd9ebd2f7fc64e77f3259b9e81f1d3de0c400f10a735
-  data.tar.gz: 33389934d411b0440e6ad28aefbe4850e9f27d5e9cecb236c9d37845ba0fdd2e6c9acad48f4fa994b230cdeb5606f10b8f4e90a4a2ae23b80b34db7df12bd275
+  metadata.gz: c2f61b99809331a28411b914f785b2fdde2bad19bdf5d0538a5a001c837854253191ad7cf3687a41e93e55ca3e3e117476ddaa41fecdcbbbcec76f4dd50299c1
+  data.tar.gz: cb6e1f024f6e7a2b12ccb973b1ab8f5e051f9f5faab1e03a93facbef405e02f21b059e174fc16d93bc7508c27395ab8afdd022f22b583741304fa083a17be2a9

data/ext/rbspy/src/lib.rs

@@ -1,11 +1,80 @@
 use ffikit::Signal;
-use pyroscope::backend::Tag;
+use pyroscope::backend::{Report, StackFrame, Tag};
 use pyroscope::PyroscopeAgent;
 use pyroscope_rbspy::{rbspy_backend, RbspyConfig};
 use std::collections::hash_map::DefaultHasher;
 use std::ffi::CStr;
 use std::hash::Hasher;
 use std::os::raw::c_char;
+use std::env;
+
+pub fn transform_report(report: Report) -> Report {
+    let cwd = env::current_dir().unwrap();
+    let cwd = cwd.to_str().unwrap_or("");
+
+    let data = report
+        .data
+        .iter()
+        .map(|(stacktrace, count)| {
+            let new_frames = stacktrace
+                .frames
+                .iter()
+                .map(|frame| {
+                    let frame = frame.to_owned();
+                    let mut s = frame.filename.unwrap();
+                    match s.find(cwd) {
+                        Some(i) => {
+                            s = s[(i+cwd.len()+1)..].to_string();
+                        }
+                        None => {
+                            match s.find("/gems/") {
+                                Some(i) => {
+                                    s = s[(i+1)..].to_string();
+                                }
+                                None => {
+                                    match s.find("/ruby/") {
+                                        Some(i) => {
+                                            s = s[(i+6)..].to_string();
+                                            match s.find("/") {
+                                                Some(i) => {
+                                                    s = s[(i+1)..].to_string();
+                                                }
+                                                None => {
+                                                }
+                                            }
+                                        }
+                                        None => {
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+
+                    // something
+                    StackFrame::new(
+                        frame.module,
+                        frame.name,
+                        Some(s.to_string()),
+                        frame.relative_path,
+                        frame.absolute_path,
+                        frame.line,
+                    )
+                })
+                .collect();
+
+            let mut mystack = stacktrace.to_owned();
+
+            mystack.frames = new_frames;
+
+            (mystack, count.to_owned())
+        })
+        .collect();
+
+    let new_report = Report::new(data).metadata(report.metadata.clone());
+
+    new_report
+}
 
 #[no_mangle]
 pub extern "C" fn initialize_agent(
@@ -52,6 +121,7 @@ pub extern "C" fn initialize_agent(
 
     let mut agent_builder = PyroscopeAgent::builder(server_address, application_name)
         .backend(rbspy)
+        .func(transform_report)
         .tags(tags);
 
     if auth_token != "" {

data/lib/pyroscope/version.rb

@@ -1,3 +1,3 @@
 module Pyroscope
-  VERSION = '0.3.1'.freeze
+  VERSION = '0.3.2'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pyroscope
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: aarch64-linux
 authors:
 - Pyroscope Team
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-07-19 00:00:00.000000000 Z
+date: 2022-07-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi