Sha256: a5f06b661cc5b969d666e94e6737a51c3a240e415a8281b64af3512b4ee5255c

Contents?: true

Size: 1.17 KB

Versions: 5

Compression:

Stored size: 1.17 KB

Contents

module Appsignal
  class Hooks
    class ShoryukenMiddleware
      def call(_worker_instance, queue, sqs_msg, body)
        metadata = {
          :queue => queue
        }
        exclude_keys = [:job_class, :queue_name, :arguments]
        metadata.merge!(body.reject { |key| exclude_keys.member?(key.to_sym) })
        metadata.merge!(sqs_msg.attributes)

        options = {
          :class => body["job_class"],
          :method => "perform",
          :metadata => metadata
        }
        options[:params] = body["arguments"] if body.key?("arguments")
        options[:queue_start] = Time.at(sqs_msg.attributes["SentTimestamp"].to_i / 1000) if sqs_msg.attributes.key?("SentTimestamp")

        Appsignal.monitor_transaction("perform_job.shoryuken", options) do
          yield
        end
      end
    end

    class ShoryukenHook < Appsignal::Hooks::Hook
      register :shoryuken

      def dependencies_present?
        defined?(::Shoryuken)
      end

      def install
        ::Shoryuken.configure_server do |config|
          config.server_middleware do |chain|
            chain.add Appsignal::Hooks::ShoryukenMiddleware
          end
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version Path
appsignal-2.1.2 lib/appsignal/hooks/shoryuken.rb
appsignal-2.1.1 lib/appsignal/hooks/shoryuken.rb
appsignal-2.1.1.beta.1 lib/appsignal/hooks/shoryuken.rb
appsignal-2.1.0 lib/appsignal/hooks/shoryuken.rb
appsignal-2.1.0.beta.1 lib/appsignal/hooks/shoryuken.rb