app_profiler 0.1.7 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94eb2e555af032b1b54f392d98d3fa5f6b6fa1cff7aa76fc2092f24b00cdfe9f
-  data.tar.gz: 94e0901199de37232236a09ce66b6adaf61e2a23374c2eed736f603a004e8990
+  metadata.gz: 6411e547488321b7d7de4fb7cc2d19c48ac462559331ac79b1e161ac11339fe7
+  data.tar.gz: 8f78938dfcdf37ee184371c6772db42cbca73e547aaab2e904ec51135cb4b69a
 SHA512:
-  metadata.gz: ec60f80896dff0a664faa2d807c09afd154494325a68c6e418e4f9b48765f42b8452f5f6a44253e46e9747513fd80b3835a762fb6a3fdf8cac0ac8086ec3099c
-  data.tar.gz: 4b88a7df76dfaac8d918674675c5dc7a5acae867e7bc4e62ba496aa13952bf02d450a26e1942e394914b98483ff4ebe4afebc7903b04cd669e9485aaac33382c
+  metadata.gz: 992474693598d4ff50a51d703143d593cb32156fa12f1e9d65fedd6489d5ff5a8d4d6a823a608dca814a64d505a85dc74a36000a07c0c4803377b9ba28cf5457
+  data.tar.gz: f184c40a906cc4cf2989becfd0cf5e8dab28c17a3f2a7cf6ad51da3e244a69d58a60bfc8c8b5011d3bcf955602cce24f36dd2122990104e1d2645b3e76a4fc01

data/lib/app_profiler/railtie.rb

@@ -37,6 +37,9 @@ module AppProfiler
       AppProfiler.upload_queue_max_length = app.config.app_profiler.upload_queue_max_length || 10
       AppProfiler.upload_queue_interval_secs = app.config.app_profiler.upload_queue_interval_secs || 5
       AppProfiler.profile_file_prefix = app.config.app_profiler.profile_file_prefix || DefaultProfilePrefix
+      AppProfiler.profile_enqueue_success = app.config.app_profiler.profile_enqueue_success
+      AppProfiler.profile_enqueue_failure = app.config.app_profiler.profile_enqueue_failure
+      AppProfiler.after_process_queue = app.config.app_profiler.after_process_queue
     end
 
     initializer "app_profiler.add_middleware" do |app|
@@ -60,7 +63,7 @@ module AppProfiler
     private
 
     def default_middleware_action
-      if Rails.env.development?
+      if Rails.env.development? || Rails.env.test?
         Middleware::ViewAction
       else
         Middleware::UploadAction

data/lib/app_profiler/storage/google_cloud_storage.rb

@@ -37,8 +37,10 @@ module AppProfiler
             @queue ||= init_queue
             begin
               @queue.push(profile, true) # non-blocking push, raises ThreadError if queue is full
+              AppProfiler.profile_enqueue_success&.call
             rescue ThreadError
               AppProfiler.logger.info("[AppProfiler] upload queue is full, profile discarded")
+              AppProfiler.profile_enqueue_failure&.call(profile)
             end
           end
         end
@@ -80,7 +82,12 @@ module AppProfiler
 
           return unless queue
 
-          queue.size.times { queue.pop(false).upload }
+          num_success = 0
+          num_failures = 0
+          queue.size.times do
+            queue.pop(false).upload ? num_success += 1 : num_failures += 1
+          end
+          AppProfiler.after_process_queue&.call(num_success, num_failures)
         end
 
         def gcs_filename(profile)

data/lib/app_profiler/version.rb

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

data/lib/app_profiler/viewer/speedscope_remote_viewer/base_middleware.rb

@@ -1,12 +1,13 @@
 # frozen_string_literal: true
 
+gem "rails-html-sanitizer", ">= 1.6.0"
 require "rails-html-sanitizer"
 
 module AppProfiler
   module Viewer
     class SpeedscopeRemoteViewer < BaseViewer
       class BaseMiddleware
-        class Sanitizer < Rails::Html::SafeListSanitizer
+        class Sanitizer < Rails::HTML::Sanitizer.best_supported_vendor.safe_list_sanitizer
           self.allowed_tags = Set.new([
             "strong", "em", "b", "i", "p", "code", "pre", "tt", "samp", "kbd", "var", "sub",
             "sup", "dfn", "cite", "big", "small", "address", "hr", "br", "div", "span", "h1",

data/lib/app_profiler.rb

@@ -55,6 +55,9 @@ module AppProfiler
   mattr_accessor :upload_queue_max_length, default: 10
   mattr_accessor :upload_queue_interval_secs, default: 5
   mattr_accessor :profile_file_prefix, default: DefaultProfilePrefix
+  mattr_reader :profile_enqueue_success, default: nil
+  mattr_reader :profile_enqueue_failure, default: nil
+  mattr_reader :after_process_queue, default: nil
 
   class << self
     def run(*args, &block)
@@ -88,6 +91,30 @@ module AppProfiler
       @@profile_url_formatter = block # rubocop:disable Style/ClassVars
     end
 
+    def profile_enqueue_success=(handler)
+      if handler && (!handler.is_a?(Proc) || (handler.lambda? && handler.arity != 0))
+        raise ArgumentError, "profile_enqueue_success must be proc or a lambda that accepts no argument"
+      end
+
+      @@profile_enqueue_success = handler # rubocop:disable Style/ClassVars
+    end
+
+    def profile_enqueue_failure=(handler)
+      if handler && (!handler.is_a?(Proc) || (handler.lambda? && handler.arity != 1))
+        raise ArgumentError, "profile_enqueue_failure must be a proc or a lambda that accepts one argument"
+      end
+
+      @@profile_enqueue_failure = handler # rubocop:disable Style/ClassVars
+    end
+
+    def after_process_queue=(handler)
+      if handler && (!handler.is_a?(Proc) || (handler.lambda? && handler.arity != 2))
+        raise ArgumentError, "after_process_queue must be a proc or a lambda that accepts two arguments"
+      end
+
+      @@after_process_queue = handler # rubocop:disable Style/ClassVars
+    end
+
     def profile_url(upload)
       return unless AppProfiler.profile_url_formatter
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: app_profiler
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.9
 platform: ruby
 authors:
 - Gannon McGibbon
@@ -13,7 +13,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-20 00:00:00.000000000 Z
+date: 2023-12-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -175,7 +175,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.21
+rubygems_version: 3.4.22
 signing_key: 
 specification_version: 4
 summary: Collect performance profiles for your Rails application.