files.rb in sidekiq-belt-0.3.2

- old
+ new

@@ -2,23 +2,27 @@
 
 require "sidekiq"
 
 require_relative "periodic_pause"
 require_relative "periodic_run"
+require_relative "periodic_sort"
 
 module Sidekiq
   module Belt
     module Ent
       module Files
         def self.use!(options = [:all])
           return unless Sidekiq.ent?
 
-          all = options.include?(:all)
+          Sidekiq::Belt::Ent::PeriodicPause.use! if should_use?(:periodic_pause, options)
+          Sidekiq::Belt::Ent::PeriodicRun.use! if should_use?(:periodic_run, options)
+          Sidekiq::Belt::Ent::PeriodicSort.use! if should_use?(:periodic_sort, options)
 
-          Sidekiq::Belt::Ent::PeriodicPause.use! if all || options.include?(:periodic_pause)
-          Sidekiq::Belt::Ent::PeriodicRun.use! if all || options.include?(:periodic_run)
-
           true
+        end
+
+        def self.should_use?(key, options)
+          options.include?(:all) || options.include?(key)
         end
       end
     end
   end
 end