pyroscope 0.6.7 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2119ee2c645b4b3dc14e436eeb7d99eb6c94848a09b41eaaf6b8119d264c8427
-  data.tar.gz: 719a6942b94e5c3941e4383896b153f055e8f7368507189699cefb3543a57d9c
+  metadata.gz: 56905d725cb3ba8a3cf0b4a1342b16d66f02a878d5d7d231a66e58230d356a29
+  data.tar.gz: 53679ec8008bc732fc3dbbaf7a7f571ee15f8d6d90a81c5999b4139cb314dff3
 SHA512:
-  metadata.gz: 187a968896c5787e8c5d11055dffd1f086b52badfff9ff41fa372f362e24f8b4d3b830afe93ad7a7842cae11a7aaf9f6ad98e35120d551746db4a5bd122d4ab1
-  data.tar.gz: d0437211746dadb0e4591251cab53da33158af8ed2a1c4932729c81f0aedd7a75f33e6a26dec2b55b80d9d032b4950a4d25ac5b5b4f0ee3b89db2baba096f9c4
+  metadata.gz: 66e3661210271621c7e9d2bc58b920c00eed6f74ca25d74ca09c0866948954822f5a4e85e05729372d1a940b47a31d570cf53466c25c84b1c50a67b40f520839
+  data.tar.gz: fd206c4e0e95870a58f84bd16a23b64f8d8566f323cd61870593e44a6e87a822e8a17ba60ed1220dc0e4bf60d61bacb96beed9c528d7f2e980c14165a258eb2c

data/Gemfile.lock

@@ -1,14 +1,15 @@
 PATH
   remote: .
   specs:
-    pyroscope (0.6.7)
+    pyroscope (1.0.2)
       ffi
 
 GEM
   remote: https://rubygems.org/
   specs:
-    ffi (1.16.3)
-    rake (13.2.0)
+    ffi (1.17.4)
+    ffi (1.17.4-x86_64-linux-gnu)
+    rake (13.3.1)
 
 PLATFORMS
   ruby
@@ -20,4 +21,4 @@ DEPENDENCIES
   rake (~> 13.0)
 
 BUNDLED WITH
-   2.5.3
+  4.0.9

data/README.md

@@ -1,71 +1,24 @@
 # Pyroscope Ruby Gem
 
-**Pyroscope integration for Ruby**
+Ruby integration for [Pyroscope](https://grafana.com/oss/pyroscope/) continuous profiling platform.
 
-[![license](https://img.shields.io/badge/license-Apache2.0-blue.svg)](LICENSE) 
-![tests](https://github.com/pyroscope-io/pyroscope-rs/workflows/Tests/badge.svg)
-![build](https://github.com/pyroscope-io/pyroscope-rs/workflows/Build/badge.svg)
-[![Gem version](https://badge.fury.io/rb/pyroscope.svg)](https://badge.fury.io/rb/pyroscope)
+## Installation
 
----
-
-### What is Pyroscope
-[Pyroscope](https://github.com/pyroscope-io/pyroscope) is a tool that lets you continuously profile your applications to prevent and debug performance issues in your code. It consists of a low-overhead agent which sends data to the Pyroscope server which includes a custom-built storage engine. This allows for you to store and query any applications profiling data in an extremely efficient and cost effective way.
-
-
-### Supported platforms
-
-| Linux | macOS | Windows | Docker |
-|:-----:|:-----:|:-------:|:------:|
-|   ✅  |   ✅  |         |   ✅   |
-
-### Profiling Ruby applications
-
-Add the `pyroscope` gem to your Gemfile:
-
-```bash
-bundle add pyroscope
 ```
-
-### Basic Configuration
-
-Add the following code to your application. If you're using rails, put this into `config/initializers` directory. This code will initialize pyroscope profiler and start profiling:
-
-```ruby
-require 'pyroscope'
-
-Pyroscope.configure do |config|
-  config.application_name = "my.ruby.app" # replace this with some name for your application
-  config.server_address   = "http://my-pyroscope-server:4040" # replace this with the address of your pyroscope server
-  # config.auth_token     = "{YOUR_API_KEY}" # optionally, if authentication is enabled, specify the API key
-end
+gem install pyroscope
 ```
 
-### Tags
-
-Pyroscope ruby integration provides a number of ways to tag profiling data. For example, you can provide tags when you're initializing the profiler:
+## Usage
 
 ```ruby
 require 'pyroscope'
 
 Pyroscope.configure do |config|
   config.application_name = "my.ruby.app"
-  config.server_address   = "http://my-pyroscope-server:4040"
-
-  config.tags = {
-    "hostname" => ENV["HOSTNAME"],
-  }
-end
-```
-
-or you can dynamically tag certain parts of your code:
-
-```ruby
-Pyroscope.tag_wrapper({ "controller": "slow_controller_i_want_to_profile" }) do
-  slow_code
+  config.server_address = "http://localhost:4040"
 end
 ```
 
-### Example
+## License
 
-Check out this [example ruby project in our repository](https://github.com/pyroscope-io/pyroscope/tree/main/examples/ruby) for examples of how you can use these features.
+Apache-2.0

data/lib/pyroscope/version.rb

@@ -1,3 +1,3 @@
 module Pyroscope
-  VERSION = '0.6.7'.freeze
+  VERSION = '1.0.2'.freeze # x-release-please-version
 end

data/lib/pyroscope.rb

@@ -9,20 +9,12 @@ module Pyroscope
     extend FFI::Library
     ffi_lib File.expand_path(File.dirname(__FILE__)) + "/rbspy/rbspy.#{RbConfig::CONFIG["DLEXT"]}"
     attach_function :initialize_logging, [:int], :bool
-    attach_function :initialize_agent, [:string, :string, :string, :string, :string, :int, :bool, :bool, :bool, :bool, :string, :string, :string, :string, :string], :bool
-    attach_function :add_thread_tag, [:uint64, :string, :string], :bool
-    attach_function :remove_thread_tag, [:uint64, :string, :string], :bool
-    attach_function :add_global_tag, [:string, :string], :bool
-    attach_function :remove_global_tag, [:string, :string], :bool
+    attach_function :initialize_agent, [:string, :string, :string, :string, :int, :bool, :bool, :bool, :string, :string, :string], :bool
+    attach_function :add_thread_tag, [:string, :string], :bool
+    attach_function :remove_thread_tag, [:string, :string], :bool
     attach_function :drop_agent, [], :bool
   end
 
-  module Utils
-    extend FFI::Library
-    ffi_lib File.expand_path(File.dirname(__FILE__)) + "/thread_id/thread_id.#{RbConfig::CONFIG["DLEXT"]}"
-    attach_function :thread_id, [], :uint64
-  end
-
   if defined?(::Rails::Engine)
     class Engine < ::Rails::Engine
       config.after_initialize do
@@ -37,12 +29,10 @@ module Pyroscope
     :application_name,
     :app_name,
     :server_address,
-    :auth_token,
     :basic_auth_username,
     :basic_auth_password,
     :log_level,
     :sample_rate,
-    :detect_subprocesses,
     :oncpu,
     :report_pid,
     :report_thread_id,
@@ -58,11 +48,9 @@ module Pyroscope
       # defaults:
       self.application_name = ''
       self.server_address = 'http://localhost:4040'
-      self.auth_token = ''
       self.basic_auth_username = ''
       self.basic_auth_password = ''
       self.sample_rate = 100
-      self.detect_subprocesses = false
       self.oncpu = true
       self.report_pid = false
       self.report_thread_id = false
@@ -109,17 +97,13 @@ module Pyroscope
         # these are defaults in case user-provided values are nil:
         @config.app_name || @config.application_name || "",
         @config.server_address || "",
-        @config.auth_token || "",
         @config.basic_auth_username || "",
         @config.basic_auth_password || "",
         @config.sample_rate || 100,
-        @config.detect_subprocesses || false,
         @config.oncpu || false,
         @config.report_pid || false,
         @config.report_thread_id || false,
         tags_to_string(@config.tags || {}),
-        @config.compression || "",
-        @config.report_encoding || "pprof",
         @config.tenant_id || "",
         http_headers_to_json(@config.http_headers || {})
       )
@@ -137,12 +121,11 @@ module Pyroscope
     end
 
     def tag_wrapper(tags)
-      tid = thread_id
-      _add_tags(tid, tags)
+      _add_tags(tags)
       begin
         yield
       ensure
-        _remove_tags(tid, tags)
+        _remove_tags(tags)
       end
     end
 
@@ -154,19 +137,15 @@ module Pyroscope
       warn("deprecated. Use `Pyroscope.tag_wrapper` instead.")
     end
 
-    def thread_id
-      return Utils.thread_id
-    end
-
-    def _add_tags(thread_id, tags)
+    def _add_tags(tags)
       tags.each do |tag_name, tag_value|
-        Rust.add_thread_tag(thread_id, tag_name.to_s, tag_value.to_s)
+        Rust.add_thread_tag(tag_name.to_s, tag_value.to_s)
       end
     end
 
-    def _remove_tags(thread_id, tags)
+    def _remove_tags(tags)
       tags.each do |tag_name, tag_value|
-        Rust.remove_thread_tag(thread_id, tag_name.to_s, tag_value.to_s)
+        Rust.remove_thread_tag(tag_name.to_s, tag_value.to_s)
       end
     end
 

data/pyroscope.gemspec

@@ -14,55 +14,31 @@ Gem::Specification.new do |s|
   s.description = 'Pyroscope FFI Integration for Ruby'
   s.authors = ['Pyroscope Team']
   s.email = ['contact@pyroscope.io']
-  s.homepage = 'https://pyroscope.io'
+  s.homepage = 'https://grafana.com/oss/pyroscope/'
   s.license = 'Apache-2.0'
   s.metadata = {
-    "homepage_uri" => "https://pyroscope.io",
-    "bug_tracker_uri" => "https://github.com/pyroscope-io/pyroscope-rs/issues",
-    "documentation_uri" => "https://pyroscope.io/docs/ruby/",
-    "changelog_uri" => "https://github.com/pyroscope-io/pyroscope-rs/tree/main/pyroscope_ffi/ruby/CHANGELOG.md",
-    "source_code_uri" => "https://github.com/pyroscope-io/pyroscope-rs/tree/main/pyroscope_ffi/ruby",
+    "homepage_uri" => "https://grafana.com/oss/pyroscope/",
+    "bug_tracker_uri" => "https://github.com/grafana/pyroscope-ruby/issues",
+    "documentation_uri" => "https://grafana.com/docs/pyroscope/latest/configure-client/language-sdks/ruby/",
+    "source_code_uri" => "https://github.com/grafana/pyroscope-ruby",
   }
 
-  # Specify which files should be added to the gem when it is released.
-  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  #s.files = Dir.chdir(__dir__) do
-    #`git ls-files -z`.split("\x0").reject do |f|
-      #(f == __FILE__) || f.match(%r{\A(?:(?:bin|test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
-    #end
-  #end
-#   s.files = `git ls-files -z`.split("\0").reject { |f| f =~ /^(\.|G|spec|Rakefile)/ }
+  # TODO: s.files is incomplete — ext/rbspy/ sources are missing, so building
+  # the gem from source does not work. See https://github.com/grafana/pyroscope-ruby/issues/2
   s.files = [
     "Gemfile",
     "Gemfile.lock",
     "LICENSE",
-#     "Makefile",
     "README.md",
-#     "Rakefile",
-    "ext/rbspy/Cargo.toml",
-    "ext/rbspy/Rakefile",
-    "ext/rbspy/build.rs",
-    "ext/rbspy/cbindgen.toml",
-    "ext/rbspy/extconf.rb",
-    "ext/rbspy/include/rbspy.h",
-    "ext/rbspy/src/lib.rs",
-    "ext/thread_id/Cargo.toml",
-    "ext/thread_id/Rakefile",
-    "ext/thread_id/build.rs",
-    "ext/thread_id/cbindgen.toml",
-    "ext/thread_id/extconf.rb",
-    "ext/thread_id/include/thread_id.h",
-    "ext/thread_id/src/lib.rs",
     "lib/pyroscope.rb",
     "lib/pyroscope/version.rb",
     "pyroscope.gemspec",
-#     "scripts/tests/test.rb",
   ]
   s.platform = Gem::Platform::RUBY
 
   s.required_ruby_version = ">= 1.9.3"
 
-  s.extensions = ['ext/rbspy/extconf.rb', 'ext/thread_id/extconf.rb']
+  s.extensions = ['ext/rbspy/extconf.rb']
 
   s.add_dependency 'ffi'
 

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: pyroscope
 version: !ruby/object:Gem::Version
-  version: 0.6.7
+  version: 1.0.2
 platform: ruby
 authors:
 - Pyroscope Team
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-09-17 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -24,7 +23,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  force_ruby_platform: false
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -59,40 +57,24 @@ email:
 executables: []
 extensions:
 - ext/rbspy/extconf.rb
-- ext/thread_id/extconf.rb
 extra_rdoc_files: []
 files:
 - Gemfile
 - Gemfile.lock
 - LICENSE
 - README.md
-- ext/rbspy/Cargo.toml
-- ext/rbspy/Rakefile
-- ext/rbspy/build.rs
-- ext/rbspy/cbindgen.toml
 - ext/rbspy/extconf.rb
-- ext/rbspy/include/rbspy.h
-- ext/rbspy/src/lib.rs
-- ext/thread_id/Cargo.toml
-- ext/thread_id/Rakefile
-- ext/thread_id/build.rs
-- ext/thread_id/cbindgen.toml
-- ext/thread_id/extconf.rb
-- ext/thread_id/include/thread_id.h
-- ext/thread_id/src/lib.rs
 - lib/pyroscope.rb
 - lib/pyroscope/version.rb
 - pyroscope.gemspec
-homepage: https://pyroscope.io
+homepage: https://grafana.com/oss/pyroscope/
 licenses:
 - Apache-2.0
 metadata:
-  homepage_uri: https://pyroscope.io
-  bug_tracker_uri: https://github.com/pyroscope-io/pyroscope-rs/issues
-  documentation_uri: https://pyroscope.io/docs/ruby/
-  changelog_uri: https://github.com/pyroscope-io/pyroscope-rs/tree/main/pyroscope_ffi/ruby/CHANGELOG.md
-  source_code_uri: https://github.com/pyroscope-io/pyroscope-rs/tree/main/pyroscope_ffi/ruby
-post_install_message:
+  homepage_uri: https://grafana.com/oss/pyroscope/
+  bug_tracker_uri: https://github.com/grafana/pyroscope-ruby/issues
+  documentation_uri: https://grafana.com/docs/pyroscope/latest/configure-client/language-sdks/ruby/
+  source_code_uri: https://github.com/grafana/pyroscope-ruby
 rdoc_options: []
 require_paths:
 - lib
@@ -107,8 +89,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.27
-signing_key:
+rubygems_version: 4.0.6
 specification_version: 4
 summary: Pyroscope
 test_files: []

data/ext/rbspy/Cargo.toml

@@ -1,21 +0,0 @@
-[package]
-name = "ffiruby"
-version = "0.1.0"
-edition = "2021"
-rust-version = "1.64"
-
-[lib]
-name = "rbspy"
-crate-type = ["cdylib"]
-
-[dependencies]
-pyroscope = { path = "../../../../" }
-pyroscope_rbspy = { path = "../../../../pyroscope_backends/pyroscope_rbspy" }
-ffikit = { path = "../../../ffikit" }
-# todo remove this dependency
-pretty_env_logger = "0.5"
-log = "0.4"
-
-[build-dependencies]
-cbindgen = "0.28"
-

data/ext/rbspy/Rakefile

@@ -1,164 +0,0 @@
-# frozen_string_literal: true
-
-require "shellwords"
-
-class RbspyRakeCargoHelper
-  attr_reader :gemname
-
-  def initialize(gemname=File.basename(__dir__))
-    @gemname = gemname
-  end
-
-  def self.command?(name)
-    exts = ENV["PATHEXT"] ? ENV["PATHEXT"].split(";") : [""]
-    ENV["PATH"].split(File::PATH_SEPARATOR).any? do |path|
-      exts.any? do |ext|
-        exe = File.join(path, "#{name}#{ext}")
-        File.executable?(exe) && !File.directory?(exe)
-      end
-    end
-  end
-
-  def self.rust_toolchain
-    # return env variable if set
-    target = ENV["RUST_TARGET"]
-    return target if target
-
-    str = `rustc --version --verbose`
-    info = str.lines.map {|l| l.chomp.split(/:\s+/, 2)}.drop(1).to_h
-    info["host"]
-  end
-
-  def self.cargo_target_dir
-    return @cargo_target_dir if defined? @cargo_target_dir
-
-    str = `cargo metadata --format-version 1 --offline --no-deps --quiet`
-    begin
-      require "json"
-      dir = JSON.parse(str)["target_directory"]
-    rescue LoadError # json is usually part of the stdlib, but just in case
-      /"target_directory"\s*:\s*"(?<dir>[^"]*)"/ =~ str
-    end
-    @cargo_target_dir = dir || "target"
-  end
-
-  def self.flags
-    cc_flags = Shellwords.split(RbConfig.expand(RbConfig::MAKEFILE_CONFIG["CC"].dup))
-
-    ["-C", "linker=#{cc_flags.shift}",
-      *cc_flags.flat_map {|a| ["-C", "link-arg=#{a}"] },
-      "-L", "native=#{RbConfig::CONFIG["libdir"]}",
-      *dld_flags,
-      *platform_flags,
-    ]
-  end
-
-  def self.dld_flags
-    Shellwords.split(RbConfig::CONFIG["DLDFLAGS"]).flat_map do |arg|
-      arg = arg.gsub(/\$\((\w+)\)/) do
-        $1 == "DEFFILE" ? nil : RbConfig::CONFIG[name]
-      end.strip
-      next [] if arg.empty?
-
-      transform_flag(arg)
-    end
-  end
-
-  def self.platform_flags
-    return unless RbConfig::CONFIG["target_os"] =~ /mingw/i
-
-    [*Shellwords.split(RbConfig::CONFIG["LIBRUBYARG"]).flat_map {|arg| transform_flag(arg)},
-      "-C", "link-arg=-Wl,--dynamicbase",
-      "-C", "link-arg=-Wl,--disable-auto-image-base",
-      "-C", "link-arg=-static-libgcc"]
-  end
-
-  def self.transform_flag(arg)
-    k, v = arg.split(/(?<=..)/, 2)
-    case k
-    when "-L"
-      [k, "native=#{v}"]
-    when "-l"
-      [k, v]
-    when "-F"
-      ["-l", "framework=#{v}"]
-    else
-      ["-C", "link_arg=#{k}#{v}"]
-    end
-  end
-
-  def install_dir
-    File.expand_path(File.join("..", "..", "lib", gemname), __dir__)
-  end
-
-  def rust_name
-    prefix = "lib" unless Gem.win_platform?
-    suffix = if RbConfig::CONFIG["target_os"] =~ /darwin/i
-      ".dylib"
-    elsif Gem.win_platform?
-      ".dll"
-    else
-      ".so"
-    end
-    "#{prefix}#{gemname}#{suffix}"
-  end
-
-  def ruby_name
-    "#{gemname}.#{RbConfig::CONFIG["DLEXT"]}"
-  end
-
-end
-
-task default: [:rbspy_install, :rbspy_clean]
-task rbspy: [:rbspy_install, :rbspy_clean]
-
-desc "set dev mode for subsequent task, run like `rake dev install`"
-task :rbspy_dev do
-  @dev = true
-end
-
-desc "build gem native extension and copy to lib"
-task rbspy_install: [:rbspy_cd, :rbspy_build] do
-  helper = RbspyRakeCargoHelper.new
-  profile_dir = @dev ? "debug" : "release"
-  arch_dir = RbspyRakeCargoHelper.rust_toolchain
-  source = File.join(RbspyRakeCargoHelper.cargo_target_dir, arch_dir, profile_dir, helper.rust_name)
-  dest = File.join(helper.install_dir, helper.ruby_name)
-  mkdir_p(helper.install_dir)
-  rm(dest) if File.exist?(dest)
-  cp(source, dest)
-end
-
-desc "build gem native extension"
-task rbspy_build: [:rbspy_cargo, :rbspy_cd] do
-  sh "cargo", "rustc", *(["--locked", "--release"] unless @dev), "--target=#{RbspyRakeCargoHelper.rust_toolchain}", "--", *RbspyRakeCargoHelper.flags
-end
-
-desc "clean up release build artifacts"
-task rbspy_clean: [:rbspy_cargo, :rbspy_cd] do
-  sh "cargo clean --release"
-end
-
-desc "clean up build artifacts"
-task rbspy_clobber: [:rbspy_cargo, :rbspy_cd] do
-  sh "cargo clean"
-end
-
-desc "check for cargo"
-task :rbspy_cargo do
-  raise <<-MSG unless RbspyRakeCargoHelper.command?("cargo")
-    This gem requires a Rust compiler and the `cargo' build tool to build the
-    gem's native extension. See https://www.rust-lang.org/tools/install for
-    how to install Rust. `cargo' is usually part of the Rust installation.
-  MSG
-
-  raise <<-MSG if Gem.win_platform? && RbspyRakeCargoHelper.rust_toolchain !~ /gnu/
-    Found Rust toolchain `#{RbspyRakeCargoHelper.rust_toolchain}' but the gem native
-    extension requires the gnu toolchain on Windows.
-  MSG
-end
-
-# ensure task is running in the right dir
-task :rbspy_cd do
-  cd(__dir__) unless __dir__ == pwd
-end

data/ext/rbspy/build.rs

@@ -1,12 +0,0 @@
-extern crate cbindgen;
-
-use cbindgen::Config;
-
-fn main() {
-    let bindings = {
-        let crate_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
-        let config = Config::from_file("cbindgen.toml").unwrap();
-        cbindgen::generate_with_config(&crate_dir, config).unwrap()
-    };
-    bindings.write_to_file("include/rbspy.h");
-}

data/ext/rbspy/cbindgen.toml

@@ -1,22 +0,0 @@
-# The language to output bindings in
-language = "C"
-documentation_style = "C"
-
-style = "type"
-
-# An optional name to use as an include guard
-include_guard = "RBSPY_H_"
-# include a comment with the version of cbindgen used to generate the file
-include_version = true
-
-# An optional string of text to output at the beginning of the generated file
-header = "/* Licensed under Apache-2.0 */"
-autogen_warning = "/* Warning, this file is autogenerated by cbindgen. Don't modify this manually. */"
-
-braces = "SameLine"
-tab_width = 2
-line_length = 80
-
-[parse]
-# Do not parse dependent crates
-parse_deps = false

data/ext/rbspy/include/rbspy.h

@@ -1,43 +0,0 @@
-/* Licensed under Apache-2.0 */
-
-#ifndef RBSPY_H_
-#define RBSPY_H_
-
-/* Generated with cbindgen:0.28.0 */
-
-/* Warning, this file is autogenerated by cbindgen. Don't modify this manually. */
-
-#include <stdarg.h>
-#include <stdbool.h>
-#include <stdint.h>
-#include <stdlib.h>
-
-bool initialize_logging(uint32_t logging_level);
-
-bool initialize_agent(const char *application_name,
-                      const char *server_address,
-                      const char *auth_token,
-                      const char *basic_auth_user,
-                      const char *basic_auth_password,
-                      uint32_t sample_rate,
-                      bool detect_subprocesses,
-                      bool oncpu,
-                      bool report_pid,
-                      bool report_thread_id,
-                      const char *tags,
-                      const char *compression,
-                      const char *_report_encoding,
-                      const char *tenant_id,
-                      const char *http_headers_json);
-
-bool drop_agent(void);
-
-bool add_thread_tag(uint64_t thread_id, const char *key, const char *value);
-
-bool remove_thread_tag(uint64_t thread_id, const char *key, const char *value);
-
-bool add_global_tag(const char *key, const char *value);
-
-bool remove_global_tag(const char *key, const char *value);
-
-#endif  /* RBSPY_H_ */

data/ext/rbspy/src/lib.rs

@@ -1,358 +0,0 @@
-use std::collections::hash_map::DefaultHasher;
-use std::env;
-use std::ffi::CStr;
-use std::hash::Hasher;
-use std::os::raw::c_char;
-use std::str::FromStr;
-
-use ffikit::Signal;
-use pyroscope_rbspy::{rbspy_backend, RbspyConfig};
-
-use pyroscope;
-use pyroscope::{pyroscope::Compression, PyroscopeAgent};
-use pyroscope::backend::{Report, StackFrame, Tag};
-use pyroscope::pyroscope::ReportEncoding;
-
-const LOG_TAG: &str = "Pyroscope::rbspy::ffi";
-
-
-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_logging(logging_level: u32) -> bool {
-    // Force rustc to display the log messages in the console.
-    match logging_level {
-        50 => {
-            std::env::set_var("RUST_LOG", "error");
-        }
-        40 => {
-            std::env::set_var("RUST_LOG", "warn");
-        }
-        30 => {
-            std::env::set_var("RUST_LOG", "info");
-        }
-        20 => {
-            std::env::set_var("RUST_LOG", "debug");
-        }
-        10 => {
-            std::env::set_var("RUST_LOG", "trace");
-        }
-        _ => {
-            std::env::set_var("RUST_LOG", "debug");
-        }
-    }
-
-    // Initialize the logger.
-    pretty_env_logger::init_timed();
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn initialize_agent(
-    application_name: *const c_char,
-    server_address: *const c_char,
-    auth_token: *const c_char,
-    basic_auth_user: *const c_char,
-    basic_auth_password: *const c_char,
-    sample_rate: u32,
-    detect_subprocesses: bool,
-    oncpu: bool,
-    report_pid: bool,
-    report_thread_id: bool,
-    tags: *const c_char,
-    compression: *const c_char,
-    _report_encoding: *const c_char,
-    tenant_id: *const c_char,
-    http_headers_json: *const c_char,
-) -> bool {
-    // Initialize FFIKit
-    let recv = ffikit::initialize_ffi().unwrap();
-
-    let application_name = unsafe { CStr::from_ptr(application_name) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let mut server_address = unsafe { CStr::from_ptr(server_address) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let adhoc_server_address = std::env::var("PYROSCOPE_ADHOC_SERVER_ADDRESS");
-    if let Ok(adhoc_server_address) = adhoc_server_address {
-        server_address = adhoc_server_address
-    }
-
-    let auth_token = unsafe { CStr::from_ptr(auth_token) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let basic_auth_user = unsafe { CStr::from_ptr(basic_auth_user) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let basic_auth_password = unsafe { CStr::from_ptr(basic_auth_password) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let tags_string = unsafe { CStr::from_ptr(tags) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let compression_string = unsafe { CStr::from_ptr(compression) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let tenant_id = unsafe { CStr::from_ptr(tenant_id) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let http_headers_json = unsafe { CStr::from_ptr(http_headers_json) }
-        .to_str()
-        .unwrap()
-        .to_string();
-
-    let compression = Compression::from_str(&compression_string);
-
-    let pid = std::process::id();
-
-    let rbspy_config = RbspyConfig::new(pid.try_into().unwrap())
-        .sample_rate(sample_rate)
-        .lock_process(false)
-        .detect_subprocesses(detect_subprocesses)
-        .oncpu(oncpu)
-        .report_pid(report_pid)
-        .report_thread_id(report_thread_id);
-
-    let tags_ref = tags_string.as_str();
-    let tags = string_to_tags(tags_ref);
-    let rbspy = rbspy_backend(rbspy_config);
-
-    let mut agent_builder = PyroscopeAgent::builder(server_address, application_name)
-        .backend(rbspy)
-        .func(transform_report)
-        .tags(tags)
-        .report_encoding(ReportEncoding::PPROF);
-
-    if auth_token != "" {
-        agent_builder = agent_builder.auth_token(auth_token);
-    } else if basic_auth_user != "" && basic_auth_password != "" {
-        agent_builder = agent_builder.basic_auth(basic_auth_user, basic_auth_password);
-    }
-
-    if tenant_id != "" {
-        agent_builder = agent_builder.tenant_id(tenant_id);
-    }
-
-    let http_headers = pyroscope::pyroscope::parse_http_headers_json(http_headers_json);
-    match http_headers {
-        Ok(http_headers) => {
-            agent_builder = agent_builder.http_headers(http_headers);
-        }
-        Err(e) => {
-            match e {
-                pyroscope::PyroscopeError::Json(e) => {
-                    log::error!(target: LOG_TAG, "parse_http_headers_json error {}", e);
-                }
-                pyroscope::PyroscopeError::AdHoc(e) => {
-                    log::error!(target: LOG_TAG, "parse_http_headers_json {}", e);
-                }
-                _ => {}
-            }
-        }
-    }
-
-    if let Ok(compression) = compression {
-        agent_builder = agent_builder.compression(compression);
-    }
-
-    let agent = agent_builder.build().unwrap();
-
-    let agent_running = agent.start().unwrap();
-
-    std::thread::spawn(move || {
-        while let Ok(signal) = recv.recv() {
-            match signal {
-                Signal::Kill => {
-                    agent_running.stop().unwrap();
-                    break;
-                }
-                Signal::AddGlobalTag(name, value) => {
-                    agent_running.add_global_tag(Tag::new(name, value)).unwrap();
-                }
-                Signal::RemoveGlobalTag(name, value) => {
-                    agent_running
-                        .remove_global_tag(Tag::new(name, value))
-                        .unwrap();
-                }
-                Signal::AddThreadTag(thread_id, key, value) => {
-                    let tag = Tag::new(key, value);
-                    agent_running.add_thread_tag(thread_id, tag).unwrap();
-                }
-                Signal::RemoveThreadTag(thread_id, key, value) => {
-                    let tag = Tag::new(key, value);
-                    agent_running.remove_thread_tag(thread_id, tag).unwrap();
-                }
-            }
-        }
-    });
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn drop_agent() -> bool {
-    // Send Kill signal to the FFI merge channel.
-    ffikit::send(ffikit::Signal::Kill).unwrap();
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn add_thread_tag(thread_id: u64, key: *const c_char, value: *const c_char) -> bool {
-    let key = unsafe { CStr::from_ptr(key) }.to_str().unwrap().to_owned();
-    let value = unsafe { CStr::from_ptr(value) }
-        .to_str()
-        .unwrap()
-        .to_owned();
-
-    let pid = std::process::id();
-    let mut hasher = DefaultHasher::new();
-    hasher.write_u64(thread_id % pid as u64);
-    let id = hasher.finish();
-
-    ffikit::send(ffikit::Signal::AddThreadTag(id, key, value)).unwrap();
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn remove_thread_tag(
-    thread_id: u64, key: *const c_char, value: *const c_char,
-) -> bool {
-    let key = unsafe { CStr::from_ptr(key) }.to_str().unwrap().to_owned();
-    let value = unsafe { CStr::from_ptr(value) }
-        .to_str()
-        .unwrap()
-        .to_owned();
-
-    let pid = std::process::id();
-    let mut hasher = DefaultHasher::new();
-    hasher.write_u64(thread_id % pid as u64);
-    let id = hasher.finish();
-
-    ffikit::send(ffikit::Signal::RemoveThreadTag(id, key, value)).unwrap();
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn add_global_tag(key: *const c_char, value: *const c_char) -> bool {
-    let key = unsafe { CStr::from_ptr(key) }.to_str().unwrap().to_owned();
-    let value = unsafe { CStr::from_ptr(value) }
-        .to_str()
-        .unwrap()
-        .to_owned();
-
-    ffikit::send(ffikit::Signal::AddGlobalTag(key, value)).unwrap();
-
-    true
-}
-
-#[no_mangle]
-pub extern "C" fn remove_global_tag(key: *const c_char, value: *const c_char) -> bool {
-    let key = unsafe { CStr::from_ptr(key) }.to_str().unwrap().to_owned();
-    let value = unsafe { CStr::from_ptr(value) }
-        .to_str()
-        .unwrap()
-        .to_owned();
-
-    ffikit::send(ffikit::Signal::RemoveGlobalTag(key, value)).unwrap();
-
-    true
-}
-
-// Convert a string of tags to a Vec<(&str, &str)>
-fn string_to_tags<'a>(tags: &'a str) -> Vec<(&'a str, &'a str)> {
-    let mut tags_vec = Vec::new();
-    // check if string is empty
-    if tags.is_empty() {
-        return tags_vec;
-    }
-
-    for tag in tags.split(',') {
-        let mut tag_split = tag.split('=');
-        let key = tag_split.next().unwrap();
-        let value = tag_split.next().unwrap();
-        tags_vec.push((key, value));
-    }
-
-    tags_vec
-}

data/ext/thread_id/Cargo.toml

@@ -1,15 +0,0 @@
-[package]
-name = "thread_id"
-version = "0.1.0"
-edition = "2021"
-rust-version = "1.64"
-
-[lib]
-name = "thread_id"
-crate-type = ["cdylib"]
-
-[dependencies]
-libc = "*"
-
-[build-dependencies]
-cbindgen = "0.28.0"

data/ext/thread_id/Rakefile

@@ -1,163 +0,0 @@
-# frozen_string_literal: true
-
-require "shellwords"
-
-class ThreadIdRakeCargoHelper
-  attr_reader :gemname
-
-  def initialize(gemname=File.basename(__dir__))
-    @gemname = gemname
-  end
-
-  def self.command?(name)
-    exts = ENV["PATHEXT"] ? ENV["PATHEXT"].split(";") : [""]
-    ENV["PATH"].split(File::PATH_SEPARATOR).any? do |path|
-      exts.any? do |ext|
-        exe = File.join(path, "#{name}#{ext}")
-        File.executable?(exe) && !File.directory?(exe)
-      end
-    end
-  end
-
-  def self.rust_toolchain
-    # return env variable if set
-    target = ENV["RUST_TARGET"]
-    return target if target
-
-    str = `rustc --version --verbose`
-    info = str.lines.map {|l| l.chomp.split(/:\s+/, 2)}.drop(1).to_h
-    info["host"]
-  end
-
-  def self.cargo_target_dir
-    return @cargo_target_dir if defined? @cargo_target_dir
-
-    str = `cargo metadata --format-version 1 --offline --no-deps --quiet`
-    begin
-      require "json"
-      dir = JSON.parse(str)["target_directory"]
-    rescue LoadError # json is usually part of the stdlib, but just in case
-      /"target_directory"\s*:\s*"(?<dir>[^"]*)"/ =~ str
-    end
-    @cargo_target_dir = dir || "target"
-  end
-
-  def self.flags
-    cc_flags = Shellwords.split(RbConfig.expand(RbConfig::MAKEFILE_CONFIG["CC"].dup))
-
-    ["-C", "linker=#{cc_flags.shift}",
-      *cc_flags.flat_map {|a| ["-C", "link-arg=#{a}"] },
-      "-L", "native=#{RbConfig::CONFIG["libdir"]}",
-      *dld_flags,
-      *platform_flags]
-  end
-
-  def self.dld_flags
-    Shellwords.split(RbConfig::CONFIG["DLDFLAGS"]).flat_map do |arg|
-      arg = arg.gsub(/\$\((\w+)\)/) do
-        $1 == "DEFFILE" ? nil : RbConfig::CONFIG[name]
-      end.strip
-      next [] if arg.empty?
-
-      transform_flag(arg)
-    end
-  end
-
-  def self.platform_flags
-    return unless RbConfig::CONFIG["target_os"] =~ /mingw/i
-
-    [*Shellwords.split(RbConfig::CONFIG["LIBRUBYARG"]).flat_map {|arg| transform_flag(arg)},
-      "-C", "link-arg=-Wl,--dynamicbase",
-      "-C", "link-arg=-Wl,--disable-auto-image-base",
-      "-C", "link-arg=-static-libgcc"]
-  end
-
-  def self.transform_flag(arg)
-    k, v = arg.split(/(?<=..)/, 2)
-    case k
-    when "-L"
-      [k, "native=#{v}"]
-    when "-l"
-      [k, v]
-    when "-F"
-      ["-l", "framework=#{v}"]
-    else
-      ["-C", "link_arg=#{k}#{v}"]
-    end
-  end
-
-  def install_dir
-    File.expand_path(File.join("..", "..", "lib", gemname), __dir__)
-  end
-
-  def rust_name
-    prefix = "lib" unless Gem.win_platform?
-    suffix = if RbConfig::CONFIG["target_os"] =~ /darwin/i
-      ".dylib"
-    elsif Gem.win_platform?
-      ".dll"
-    else
-      ".so"
-    end
-    "#{prefix}#{gemname}#{suffix}"
-  end
-
-  def ruby_name
-    "#{gemname}.#{RbConfig::CONFIG["DLEXT"]}"
-  end
-
-end
-
-task default: [:thread_id_install, :thread_id_clean]
-task thread_id: [:thread_id_install, :thread_id_clean]
-
-desc "set dev mode for subsequent task, run like `rake dev install`"
-task :thread_id_dev do
-  @dev = true
-end
-
-desc "build gem native extension and copy to lib"
-task thread_id_install: [:thread_id_cd, :thread_id_build] do
-  helper = ThreadIdRakeCargoHelper.new
-  profile_dir = @dev ? "debug" : "release"
-  arch_dir = RbspyRakeCargoHelper.rust_toolchain
-  source = File.join(ThreadIdRakeCargoHelper.cargo_target_dir, arch_dir, profile_dir, helper.rust_name)
-  dest = File.join(helper.install_dir, helper.ruby_name)
-  mkdir_p(helper.install_dir)
-  rm(dest) if File.exist?(dest)
-  cp(source, dest)
-end
-
-desc "build gem native extension"
-task thread_id_build: [:thread_id_cargo, :thread_id_cd] do
-  sh "cargo", "rustc", *(["--locked", "--release"] unless @dev), "--target=#{RbspyRakeCargoHelper.rust_toolchain}", "--", *RbspyRakeCargoHelper.flags
-end
-
-desc "clean up release build artifacts"
-task thread_id_clean: [:thread_id_cargo, :thread_id_cd] do
-  sh "cargo clean --release"
-end
-
-desc "clean up build artifacts"
-task thread_id_clobber: [:thread_id_cargo, :thread_id_cd] do
-  sh "cargo clean"
-end
-
-desc "check for cargo"
-task :thread_id_cargo do
-  raise <<-MSG unless ThreadIdRakeCargoHelper.command?("cargo")
-    This gem requires a Rust compiler and the `cargo' build tool to build the
-    gem's native extension. See https://www.rust-lang.org/tools/install for
-    how to install Rust. `cargo' is usually part of the Rust installation.
-  MSG
-
-  raise <<-MSG if Gem.win_platform? && ThreadIdRakeCargoHelper.rust_toolchain !~ /gnu/
-    Found Rust toolchain `#{ThreadIdRakeCargoHelper.rust_toolchain}' but the gem native
-    extension requires the gnu toolchain on Windows.
-  MSG
-end
-
-# ensure task is running in the right dir
-task :thread_id_cd do
-  cd(__dir__) unless __dir__ == pwd
-end

data/ext/thread_id/build.rs

@@ -1,12 +0,0 @@
-extern crate cbindgen;
-
-use cbindgen::Config;
-
-fn main() {
-    let bindings = {
-        let crate_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
-        let config = Config::from_file("cbindgen.toml").unwrap();
-        cbindgen::generate_with_config(&crate_dir, config).unwrap()
-    };
-    bindings.write_to_file("include/thread_id.h");
-}

data/ext/thread_id/cbindgen.toml

@@ -1,22 +0,0 @@
-# The language to output bindings in
-language = "C"
-documentation_style = "C"
-
-style = "type"
-
-# An optional name to use as an include guard
-include_guard = "RBSPY_H_"
-# include a comment with the version of cbindgen used to generate the file
-include_version = true
-
-# An optional string of text to output at the beginning of the generated file
-header = "/* Licensed under Apache-2.0 */"
-autogen_warning = "/* Warning, this file is autogenerated by cbindgen. Don't modify this manually. */"
-
-braces = "SameLine"
-tab_width = 2
-line_length = 80
-
-[parse]
-# Do not parse dependent crates
-parse_deps = false

data/ext/thread_id/extconf.rb

@@ -1,11 +0,0 @@
-require 'mkmf'
-require 'rake'
-
-create_makefile('thread_id')
-
-app = Rake.application
-app.init
-app.add_import 'Rakefile'
-app.load_rakefile
-
-app['default'].invoke

data/ext/thread_id/include/thread_id.h

@@ -1,17 +0,0 @@
-/* Licensed under Apache-2.0 */
-
-#ifndef RBSPY_H_
-#define RBSPY_H_
-
-/* Generated with cbindgen:0.28.0 */
-
-/* Warning, this file is autogenerated by cbindgen. Don't modify this manually. */
-
-#include <stdarg.h>
-#include <stdbool.h>
-#include <stdint.h>
-#include <stdlib.h>
-
-uint64_t thread_id(void);
-
-#endif  /* RBSPY_H_ */

data/ext/thread_id/src/lib.rs

@@ -1,4 +0,0 @@
-#[no_mangle]
-pub extern "C" fn thread_id() -> u64 {
-    unsafe { libc::pthread_self() as u64 }
-}