sidekiq-amigo 1.8.0 → 1.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a32299bc84d87b6c497414b4fcb6b89278e2993543eff1ba07bcf1a9fb75c59f
-  data.tar.gz: 27acf41f1e0235536da93d7dfd0b2a5ad1f65e4e993f428967e8072e7213f83f
+  metadata.gz: 330ead7f8face652362d632c9c127aa57b298d5587d1da392769b0458d1f8915
+  data.tar.gz: 7c0b0b9b1658eecfd202ef294580417233421edf4233eef522b747ce9c411f52
 SHA512:
-  metadata.gz: 47e75ba71d67411a80e7d5ae20f06c0b3bb4e0fb056356e63fb47ec3cd84cdb72f0deb6e2bd00f5cc0cd091267dc30725e4607ae018b7f29a31093c4c9e82ebd
-  data.tar.gz: 761f7febdba270fe365f39cb740f734615118bd8c160485ac9296119f00124458f1970e159dbdd0ed8a6fbecd4c66df8bfe07f86803a14e2982e738fbe69115c
+  metadata.gz: 898c24b82de6313c33e50b58a2c4d64ca365918a0aee2b223b452fea9428cc4d265b15975ebc2f7c7e689f547898e65b7c52d3e50edbe9bcdb062f3b25f3f0b6
+  data.tar.gz: a5ffcb482859fcddd6727a9869bef3ed00c8edb5b1b44f0cde84b31eddb44c6f84fbac6b45d8b8fccf06c712f6792a8ce3e211b4a49083fe8c0bab76474a5d38

data/lib/amigo/autoscaler/heroku.rb

@@ -81,7 +81,6 @@ module Amigo
         app_id_or_app_name: ENV.fetch("HEROKU_APP_NAME"),
         formation_id_or_formation_type: "worker"
       )
-
         @heroku = heroku
         @max_additional_workers = max_additional_workers
         @app_id_or_app_name = app_id_or_app_name

data/lib/amigo/autoscaler.rb

@@ -92,6 +92,13 @@ module Amigo
     # Proc/callable called with (level, message, params={}).
     # By default, use +Amigo.log+ (which logs to the Sidekiq logger).
     attr_reader :log
+    # Proc called with an exception that occurs while the thread is running.
+    # If the handler returns +true+, then the thread will keep going.
+    # All other values will kill the thread, which breaks autoscaling.
+    # Note that Amigo automatically logs unhandled exceptions at :error level.
+    # If you use an error reporter like Sentry, you can pass in something like:
+    #   -> (e) { Sentry.capture_exception(e) }
+    attr_reader :on_unhandled_exception
 
     def initialize(
       poll_interval: 20,
@@ -101,9 +108,9 @@ module Amigo
       alert_interval: 120,
       latency_restored_threshold: latency_threshold,
       latency_restored_handlers: [:log],
-      log: ->(level, message, params={}) { Amigo.log(nil, level, message, params) }
+      log: ->(level, message, params={}) { Amigo.log(nil, level, message, params) },
+      on_unhandled_exception: nil
     )
-
       raise ArgumentError, "latency_threshold must be > 0" if
         latency_threshold <= 0
       raise ArgumentError, "latency_restored_threshold must be >= 0" if
@@ -118,8 +125,10 @@ module Amigo
       @latency_restored_threshold = latency_restored_threshold
       @latency_restored_handlers = latency_restored_handlers.freeze
       @log = log
+      @on_unhandled_exception = on_unhandled_exception
     end
 
+    # @return [Thread]
     def polling_thread
       return @polling_thread
     end
@@ -189,6 +198,14 @@ module Amigo
     end
 
     def check
+      self._check
+    rescue StandardError => e
+      self._log(:error, "async_autoscaler_unhandled_error", exception: e)
+      handled = self.on_unhandled_exception&.call(e)
+      raise e unless handled.eql?(true)
+    end
+
+    def _check
       now = Time.now
       skip_check = now < (@last_alerted + self.alert_interval)
       if skip_check

data/lib/amigo/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Amigo
-  VERSION = "1.8.0"
+  VERSION = "1.9.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sidekiq-amigo
 version: !ruby/object:Gem::Version
-  version: 1.8.0
+  version: 1.9.0
 platform: ruby
 authors:
 - Lithic Technology
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-27 00:00:00.000000000 Z
+date: 2025-02-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sidekiq