app_profiler 0.2.9 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88fa4e71be7c319cba26e669752c8a7a46c9b7c56881894944a8ae4b8def6c6f
-  data.tar.gz: 9daa41d818e4fbd0a113186c20e7a18814a3c039bc7dc91d52d26efd7a3ab5c6
+  metadata.gz: e78035c478029d4ad4de636efab44240957c2bd05ea761dc9fe28b753b4fa1d3
+  data.tar.gz: c4d6659a31eaa7440c6c8a2550c9cf41fbadb567426cae7c6bd0cd4442e8039a
 SHA512:
-  metadata.gz: 8471c04250be1ae8b393a804ee83553ccc5c374008f343ab0213464b130658ee69fa77626a0f557e7423350e60d0ac6c803228a656e48be739def63400d34fe3
-  data.tar.gz: 216c13be42bb571a535c894f543889fa114f97ff49aa84fb59d7ab2a39e8b7f2815c5e75399a3d98b7cbcf645376e37bb4899d8e5c63761f3fac6f1ca2483848
+  metadata.gz: 45610912af0b6952b2c104d27bfc3bf2b37b92f91d2c873bdf9e2bfc43498529decf9568af63b9ee47d84794dd921dbf6312de11aac272be296f99416d5f2b31
+  data.tar.gz: 29ee1a661cc8c81274fbdb7f6123ce90d7dbeaa351ca3b16a8a6bfe933ec3f68385bfe0f2d0afb2ae6c49ba59fa318d733bd23d6cf7724a3d03f1246e36b65ca

data/lib/app_profiler/middleware.rb

@@ -7,6 +7,11 @@ require "app_profiler/middleware/view_action"
 
 module AppProfiler
   class Middleware
+    OTEL_PROFILE_ID = "profile.id"
+    OTEL_PROFILE_BACKEND = "profile.profiler"
+    OTEL_PROFILE_MODE = "profile.mode"
+    OTEL_PROFILE_CONTEXT = "profile.context"
+
     class_attribute :action,   default: UploadAction
     class_attribute :disabled, default: false
 
@@ -29,8 +34,11 @@ module AppProfiler
       return yield unless app_profiler_params
 
       params_hash = app_profiler_params.to_h
+
       return yield unless before_profile(env, params_hash)
 
+      add_otel_instrumentation(env, params_hash) if AppProfiler.otel_instrumentation_enabled
+
       profile = AppProfiler.run(**params_hash) do
         response = yield
       end
@@ -47,6 +55,27 @@ module AppProfiler
       response
     end
 
+    def add_otel_instrumentation(env, params)
+      rack_span = OpenTelemetry::Instrumentation::Rack.current_span
+      return unless rack_span.recording?
+
+      metadata = params[:metadata]
+      profile_id = if metadata[:id].present?
+        metadata[:id]
+      else
+        AppProfiler::ProfileId.current
+      end
+
+      attributes = {
+        OTEL_PROFILE_ID => profile_id,
+        OTEL_PROFILE_BACKEND => params[:backend].to_s,
+        OTEL_PROFILE_MODE => params[:mode].to_s,
+        OTEL_PROFILE_CONTEXT => AppProfiler.context,
+      }
+
+      rack_span.add_attributes(attributes)
+    end
+
     def profile_params(params)
       return params if params.valid?
       return unless AppProfiler.profile_sampler_enabled

data/lib/app_profiler/railtie.rb

@@ -51,6 +51,8 @@ module AppProfiler
       AppProfiler.profile_sampler_enabled = app.config.app_profiler.profile_sampler_enabled || false
       AppProfiler.profile_sampler_config = app.config.app_profiler.profile_sampler_config ||
         AppProfiler::Sampler::Config.new
+
+      AppProfiler.otel_instrumentation_enabled = app.config.app_profiler.otel_instrumentation_enabled || false
     end
 
     initializer "app_profiler.add_middleware" do |app|

data/lib/app_profiler/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AppProfiler
-  VERSION = "0.2.9"
+  VERSION = "0.3.0"
 end

data/lib/app_profiler.rb

@@ -76,6 +76,8 @@ module AppProfiler
   mattr_reader :profile_file_name
 
   class << self
+    attr_reader :otel_instrumentation_enabled
+
     def deprecator # :nodoc:
       @deprecator ||= ActiveSupport::Deprecation.new("in future releases", "app_profiler")
     end
@@ -162,6 +164,15 @@ module AppProfiler
       false
     end
 
+    def otel_instrumentation_enabled=(value)
+      if value
+        gem("opentelemetry-instrumentation-rack")
+        require("opentelemetry/instrumentation/rack")
+      end
+
+      @otel_instrumentation_enabled = value
+    end
+
     def backend_for(backend_name)
       if vernier_supported? &&
           backend_name&.to_sym == AppProfiler::VernierProfile::BACKEND_NAME

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: app_profiler
 version: !ruby/object:Gem::Version
-  version: 0.2.9
+  version: 0.3.0
 platform: ruby
 authors:
 - Gannon McGibbon
@@ -12,7 +12,7 @@ authors:
 - Scott Francis
 bindir: bin
 cert_chain: []
-date: 2025-02-10 00:00:00.000000000 Z
+date: 2025-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -112,6 +112,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: opentelemetry-instrumentation-rack
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement