app_profiler 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94eb2e555af032b1b54f392d98d3fa5f6b6fa1cff7aa76fc2092f24b00cdfe9f
-  data.tar.gz: 94e0901199de37232236a09ce66b6adaf61e2a23374c2eed736f603a004e8990
+  metadata.gz: 5a4f543410317515564af0f46a5f437e33af8a7c66334e6cfd42830e37dfb89c
+  data.tar.gz: bb37127c11ae94eade6d736d14d3609e7e99854709659dd9b7bb081e8cbb86b3
 SHA512:
-  metadata.gz: ec60f80896dff0a664faa2d807c09afd154494325a68c6e418e4f9b48765f42b8452f5f6a44253e46e9747513fd80b3835a762fb6a3fdf8cac0ac8086ec3099c
-  data.tar.gz: 4b88a7df76dfaac8d918674675c5dc7a5acae867e7bc4e62ba496aa13952bf02d450a26e1942e394914b98483ff4ebe4afebc7903b04cd669e9485aaac33382c
+  metadata.gz: 5a6da942b841feddb9b78d8a212b4c546b89e4966130419c87ad167f5c698301753fa4c14c4a6548338b341b217fed8418a71e9ceb2dc3a9289211436c036bd8
+  data.tar.gz: bdb22060ba83526a177c34302e731960d4eff6c563efe313f19e8282535632fa61c71ba269ce90700066166b6102244b1b1cdf6463b97d877fc45deb27828127

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|

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.8"
 end

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.8
 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-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport