about summary refs log tree commit diff
path: root/lib
diff options
context:
space:
mode:
authorEugen Rochko <eugen@zeonfederated.com>2021-12-27 00:47:20 +0100
committerGitHub <noreply@github.com>2021-12-27 00:47:20 +0100
commitfe71548844715d796b9a6d2dd5234f70cf080f13 (patch)
tree754d3e3fda806b3ec042e3a5bc5fe4d1dda08995 /lib
parente65080181af82c14d3441a0890f2ba0a6fb9cd7e (diff)
Fix warnings on Rails boot (#16946)
Diffstat (limited to 'lib')
-rw-r--r--lib/sidekiq_error_handler.rb24
1 files changed, 24 insertions, 0 deletions
diff --git a/lib/sidekiq_error_handler.rb b/lib/sidekiq_error_handler.rb
new file mode 100644
index 000000000..358afd540
--- /dev/null
+++ b/lib/sidekiq_error_handler.rb
@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+class SidekiqErrorHandler
+  BACKTRACE_LIMIT = 3
+
+  def call(*)
+    yield
+  rescue Mastodon::HostValidationError
+    # Do not retry
+  rescue => e
+    limit_backtrace_and_raise(e)
+  ensure
+    socket = Thread.current[:statsd_socket]
+    socket&.close
+    Thread.current[:statsd_socket] = nil
+  end
+
+  private
+
+  def limit_backtrace_and_raise(exception)
+    exception.set_backtrace(exception.backtrace.first(BACKTRACE_LIMIT))
+    raise exception
+  end
+end