app_profiler 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 443e88eecba7bccc081c2b44d2a0840570c54793025251e960242bd76dc86090
-  data.tar.gz: fcc47ff7544b73faa76160212b1ef0134b2f7239de758ec8af86887abe41ef59
+  metadata.gz: 5a4f543410317515564af0f46a5f437e33af8a7c66334e6cfd42830e37dfb89c
+  data.tar.gz: bb37127c11ae94eade6d736d14d3609e7e99854709659dd9b7bb081e8cbb86b3
 SHA512:
-  metadata.gz: 7fda25b4c3e449d7f8c35341649c96679e299dbc3110e44911dcd1e970bce64b0abf2aaf3fe4830ba729fdac3780477ff626b6451579a74bdd2ceac6cf9c866b
-  data.tar.gz: 3f5020c4c4a5c817c4692a622001fa7ad851316b20cfdb0ad0057a04ab2e0d6736d1dae17e397d5a7d6aa9b2eab464c1dc39f3e0e653dc46530e96ec7eaf86c6
+  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/server.rb

@@ -225,8 +225,10 @@ module AppProfiler
         end
 
         def stop
-          File.unlink(@socket_file) if File.exist?(@socket_file) && File.socket?(@socket_file)
           @socket.close
+          File.unlink(@socket_file) if File.exist?(@socket_file) && File.socket?(@socket_file)
+          ObjectSpace.undefine_finalizer(self)
+          nil
         end
 
         def abandon

data/lib/app_profiler/storage/google_cloud_storage.rb

@@ -25,6 +25,8 @@ module AppProfiler
               content_type: "application/json",
               content_encoding: "gzip",
             )
+          ensure
+            profile.file.unlink
           end
         end
 
@@ -35,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
@@ -78,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.6"
+  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.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Gannon McGibbon
@@ -13,7 +13,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-10-18 00:00:00.000000000 Z
+date: 2023-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport