lib/sidekiq/throttled/middleware.rb in sidekiq-throttled-0.18.0 vs lib/sidekiq/throttled/middleware.rb in sidekiq-throttled-1.0.0.alpha

@@ -1,15 +1,17 @@
 # frozen_string_literal: true
 
 # internal
-require "sidekiq/throttled/registry"
+require_relative "./registry"
 
 module Sidekiq
   module Throttled
     # Server middleware that notifies strategy that job was finished.
     #
     # @private
     class Middleware
+      include Sidekiq::ServerMiddleware if Sidekiq::VERSION >= "6.5.0"
+
       # Called within Sidekiq job processing
       def call(_worker, msg, _queue)
         yield
       ensure
         Registry.get msg["class"] do |strategy|