pyroscope 0.0.14 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 619c59b1c6e494e5e1b4b47889ea9de7eed5c44264ec0bd9f1372ea87c6a94d0
-  data.tar.gz: d4c3632f2ee191e8bb60529bd5699e11100500cc3ca1cbb4d119a183c38e279d
+  metadata.gz: 743ff4c97e82300fb18a63cb6ebe626f241ce77a1bf0fd034198db698b5c5aca
+  data.tar.gz: 76d3705d25132cb5779b9e430eb2972894597d4bd4732a30c28a8995ec18fcaa
 SHA512:
-  metadata.gz: 98d120335fbe70f9527e4f9eeb3678dd8db7dfac0aefb1e44fb639c499911ddec9370bb91e9f22bd9ecda519cb3fa92e6982dd9eda135e7be9efe3f26042295d
-  data.tar.gz: '09f603d60123a92e983ac22eeefff6468e047edc99f5cffd4278dfa94e51fc837522daa669838430459bf0009f14eef4d31c1aaae8c647b0196bcd4bedeb0cb6'
+  metadata.gz: 4bc08bbdc5fa0081806d4e2030eef6a6032c1e5fb68d53784fc5e0565f7ed4d57a85171d87ff91213d3fc0209967d47279593ac2025739a611588771d36e8c0d
+  data.tar.gz: 5d27f0e63129001f60bb11fab6f19e837b82a350bc746b6ac7a1bd0e072c1ea7546b3d0b9ea32ff4d864c8feb1b71c718f2a67e9d606d98b2b518c689b9e49fc

data/README.md

@@ -1,3 +1,3 @@
 # pyroscope-ruby
 
-This is a repo for pyroscope ruby integration. It is currently under development and is not yet released.
+This is a repo for pyroscope ruby integration. See [examples/ruby](https://github.com/pyroscope-io/pyroscope/tree/main/examples/ruby) in the main repo for examples.

data/Rakefile

@@ -42,16 +42,8 @@ task :build   => [:clean, :compile]
 task :default => [:build, :spec]
 
 task :test do
-  # system "rsync -a /home/dmitry/rbspy/ /home/dmitry/pyroscope/third_party/local-clones/rbspy"
-  # # system "cd ../pyroscope && DOCKER_BUILDKIT=1 docker build -f Dockerfile.static-libs --output type=local,dest=out ." if RUBY_PLATFORM.include?("linux")
-  # system "cd ../pyroscope && rm out/*.a"
-  # system "cd ../pyroscope && ENABLED_SPIES=rbspy make build-rust-dependencies"
-  # system "cd ../pyroscope && make build-rbspy-static-library"
-  # system "cd ../pyroscope && cp third_party/rustdeps/target/x86_64-unknown-linux-gnu/release/librustdeps.a ./out/librustdeps.a"
-  extra_env = "PYROSCOPE_RUBY_LOCAL=1"
-  extra_env = ""
-
-  system "#{extra_env} rake build && #{extra_env} sudo -E gem install pkg/pyroscope-#{Pyroscope::VERSION}.gem && sudo -E ruby test.rb"
+  sudo = RUBY_PLATFORM =~ /darwin/ ? "sudo -E" : ""
+  system "rake build && #{sudo} gem install pkg/pyroscope-#{Pyroscope::VERSION}.gem && #{sudo} ruby test.rb"
 end
 
 task :test_exec do

data/ext/pyroscope/extconf.rb

@@ -9,7 +9,7 @@ HEADER_DIRS = [INCLUDEDIR]
 
 LIB_DIRS = [LIBDIR, File.expand_path(File.join(File.dirname(__FILE__), "lib"))]
 
-COMMIT = "a263249"
+COMMIT = "df45c48"
 
 # TODO: this is not very accurate, but it works for now
 OS = RUBY_PLATFORM.include?("darwin") ? "mac" : "linux"
@@ -19,10 +19,18 @@ PREFIX = "/static-libs/#{COMMIT}/#{OS}-#{ARCH}"
 
 ROOT = File.expand_path("..", __FILE__)
 
+def run(str)
+  system(str) || fail("command failed: #{str}")
+end
+
 if ENV["PYROSCOPE_RUBY_LOCAL"]
   puts "PYROSCOPE_RUBY_LOCAL yes"
-  system "cp #{ENV["HOME"]}/pyroscope/out/libpyroscope.rbspy.a #{File.join(ROOT, "lib/libpyroscope.rbspy.a")}"
-  system "cp #{ENV["HOME"]}/pyroscope/out/librustdeps.a #{File.join(ROOT, "lib/librustdeps.a")}"
+  # TODO: make configurable
+  pyroscope_path = "#{ENV["HOME"]}/pyroscope"
+  run("rm -rf #{File.join(ROOT, "build")}")
+  run("cd #{pyroscope_path} && make build-rust-dependencies-docker")
+  run("cp #{pyroscope_path}/out/libpyroscope.rbspy.a #{File.join(ROOT, "lib/libpyroscope.rbspy.a")}")
+  run("cp #{pyroscope_path}/out/librustdeps.a #{File.join(ROOT, "lib/librustdeps.a")}")
 else
   Net::HTTP.start("dl.pyroscope.io", 443, :use_ssl => true) do |http|
     ["libpyroscope.rbspy.a", "librustdeps.a"].each do |name|

data/ext/pyroscope/pyroscope.c

@@ -6,46 +6,82 @@ static VALUE rb_Pyroscope;
 #include <stdio.h>
 #include <string.h>
 
-int Start(char*, int, char*, char*);
-int Stop(int);
-int ChangeName(char*, int);
+int Start(char*, char*, char*, char*, int, int, char*);
+int Stop();
+int ChangeName(char*);
+int SetTag(char*, char*);
+int TestLogger();
+char* BuildSummary();
 
 static VALUE
-pyroscope_start(VALUE self, VALUE appName, VALUE pid, VALUE serverAddress) {
+pyroscope_start(VALUE self, VALUE appName, VALUE serverAddress, VALUE authToken, VALUE sampleRate, VALUE withSubprocesses, VALUE logLevel) {
   VALUE r_appName = StringValue(appName);
   char *c_appName = RSTRING_PTR(r_appName);
 
-  int c_pid = FIX2INT(pid);
+  int c_sampleRate = FIX2INT(sampleRate);
+  int c_withSubprocesses = FIX2INT(withSubprocesses);
 
   VALUE r_serverAddress = StringValue(serverAddress);
   char *c_serverAddress = RSTRING_PTR(r_serverAddress);
 
-  int res = Start(c_appName, c_pid, "rbspy", c_serverAddress);
+  VALUE r_authToken = StringValue(authToken);
+  char *c_authToken = RSTRING_PTR(r_authToken);
+
+  VALUE r_logLevel = StringValue(logLevel);
+  char *c_logLevel = RSTRING_PTR(r_logLevel);
+
+  int res = Start(c_appName, "rbspy", c_serverAddress, c_authToken, c_sampleRate, c_withSubprocesses, c_logLevel);
 
   return INT2FIX(res);
 }
 
 static VALUE
-pyroscope_stop(VALUE self, VALUE pid) {
-  int c_pid = FIX2INT(pid);
-  int res = Stop(c_pid);
+pyroscope_stop(VALUE self) {
+  int res = Stop();
   return INT2FIX(res);
 }
 
 static VALUE
-pyroscope_change_name(VALUE self, VALUE appName, VALUE pid) {
+pyroscope_change_name(VALUE self, VALUE appName) {
   VALUE r_appName = StringValue(appName);
   char *c_appName = RSTRING_PTR(r_appName);
-  int c_pid = FIX2INT(pid);
 
-  int res = ChangeName(c_appName, c_pid);
+  int res = ChangeName(c_appName);
+  return INT2FIX(res);
+}
+
+static VALUE
+pyroscope_set_tag(VALUE self, VALUE key, VALUE val) {
+  VALUE r_key = StringValue(key);
+  char *c_key = RSTRING_PTR(r_key);
+  VALUE r_val = StringValue(val);
+  char *c_val = RSTRING_PTR(r_val);
+
+  int res = SetTag(c_key, c_val);
   return INT2FIX(res);
 }
 
+static VALUE
+pyroscope_test_logger(VALUE self) {
+  int res = TestLogger();
+  return INT2FIX(res);
+}
+
+static VALUE
+pyroscope_build_summary(VALUE self) {
+  char *c_summary = BuildSummary();
+  VALUE r_summary = rb_str_new_cstr(c_summary);
+  free(c_summary);
+  return r_summary;
+}
+
 void
 Init_pyroscope_c() {
   rb_Pyroscope = rb_define_module("Pyroscope");
-  rb_define_module_function(rb_Pyroscope, "_start", pyroscope_start, 3);
-  rb_define_module_function(rb_Pyroscope, "_stop", pyroscope_stop, 1);
-  rb_define_module_function(rb_Pyroscope, "_change_name", pyroscope_change_name, 2);
+  rb_define_module_function(rb_Pyroscope, "_start", pyroscope_start, 6);
+  rb_define_module_function(rb_Pyroscope, "_stop", pyroscope_stop, 0);
+  rb_define_module_function(rb_Pyroscope, "_change_name", pyroscope_change_name, 1);
+  rb_define_module_function(rb_Pyroscope, "_set_tag", pyroscope_set_tag, 2);
+  rb_define_module_function(rb_Pyroscope, "_test_logger", pyroscope_test_logger, 0);
+  rb_define_module_function(rb_Pyroscope, "_build_summary", pyroscope_build_summary, 0);
 }

data/lib/pyroscope/version.rb

@@ -1,3 +1,3 @@
 module Pyroscope
-  VERSION = "0.0.14".freeze
+  VERSION = "0.1.1".freeze
 end

data/lib/pyroscope.rb

@@ -2,23 +2,59 @@ require "pyroscope/version"
 require "pyroscope_c"
 
 module Pyroscope
-  Config = Struct.new(:app_name, :server_address)
-
-  def self.configure
-    @configuration = Config.new
-    yield @configuration
-    _start(
-      @configuration.app_name,
-      Process.pid,
-      @configuration.server_address,
-    )
-  end
+  Config = Struct.new(:app_name, :server_address, :auth_token, :sample_rate, :with_subprocesses, :log_level, :tags)
 
-  def self.stop
-    _stop(Process.pid)
-  end
+  class << self
+    def configure
+      @configuration = Config.new
+      yield @configuration
+      _start(
+        @configuration.app_name,
+        @configuration.server_address,
+        @configuration.auth_token || "",
+        @configuration.sample_rate || 100,
+        @configuration.with_subprocesses || 0,
+        @configuration.log_level || "error",
+      )
+      tag(@configuration.tags) if @configuration.tags
+    end
+
+    def stop
+      _stop
+    end
+
+    def change_name(new_name)
+      _change_name(new_name)
+    end
+
+    def tag_wrapper(tags)
+      tag(tags)
+
+      begin
+        yield
+      ensure
+        remove_tags(*tags.keys)
+      end
+    end
+
+    def tag(tags)
+      tags.each_pair do |key, val|
+        _set_tag(key.to_s, val.to_s)
+      end
+    end
+
+    def remove_tags(*keys)
+      keys.each do |key|
+        _set_tag(key.to_s, "")
+      end
+    end
+
+    def test_logger
+      _test_logger
+    end
 
-  def self.change_name(new_name)
-    _change_name(new_name, Process.pid)
+    def build_summary
+      _build_summary
+    end
   end
 end

data/test.rb

@@ -1,16 +1,14 @@
-require 'pyroscope'
-
-puts "prestart #{Process.pid}"
+require "pyroscope"
 
 Pyroscope.configure do |config|
-  config.app_name = "test.ruby.app{}"
-  config.server_address = "http://localhost:4040/"
+  config.app_name = "test.ruby.app"
+  config.server_address = "http://pyroscope:4040/"
+  config.tags = {
+    :region => "us-east-1",
+    :hostname => ENV["hostname"]
+  }
 end
 
-puts "start"
-iteration=0
-st = Time.new
-
 def work(n)
   i = 0
   while i < n
@@ -18,34 +16,17 @@ def work(n)
   end
 end
 
-def job_0
-  work(rand()*1_000_000)
-end
-
-def job_1
-  work(rand()*2_000_000)
+def fast_function
+  work(20000)
 end
 
-def sleep_job
-  sleep(rand()*10)
+def slow_function
+  work(80000)
 end
 
+Pyroscope.tag({ "region" => "us-east-1" })
 
 while true
-  iteration+=1
-
-  r = rand
-  if r < 0.1
-    sleep_job
-  elsif r < 0.5
-    puts(" * test.ruby.app{job=0}")
-    Pyroscope.change_name("test.ruby.app{job=0}")
-    job_0
-    Pyroscope.change_name("test.ruby.app{}")
-  else
-    puts(" * test.ruby.app{job=1}")
-    Pyroscope.change_name("test.ruby.app{job=1}")
-    job_1
-    Pyroscope.change_name("test.ruby.app{}")
-  end
+  fast_function
+  slow_function
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pyroscope
 version: !ruby/object:Gem::Version
-  version: 0.0.14
+  version: 0.1.1
 platform: ruby
 authors:
 - Pyroscope Team
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-07-30 00:00:00.000000000 Z
+date: 2021-09-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -107,7 +107,7 @@ homepage: http://rubygems.org/gems/pyroscope
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -123,8 +123,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.15
-signing_key: 
+rubygems_version: 3.2.22
+signing_key:
 specification_version: 4
 summary: pyroscope
 test_files: []