Support old rake tasks and more

* User config class
* Add global --redis, --namespace
* Add  --config option which loads config from YAML
* Preload config in every method (unfortunately couldn't find `before_filter` or something like that functionality in thor)

Author: lukaszx0

Rakefile

@@ -1,6 +1,7 @@
 require 'rubygems'
 require 'bundler/setup'
 require 'bundler/gem_tasks'
+require 'resque/tasks'
 require 'rake/testtask'
 require 'yard'
 

bin/resque

@@ -1,6 +1,4 @@
 #!/usr/bin/env ruby
-
-require 'thor'
 require 'resque/cli'
 
 Resque::CLI.start(ARGV)

examples/sinatra/Gemfile

@@ -1,5 +1,6 @@
 # A sample Gemfile
 source "https://rubygems.org"
 
+gem "rake"
 gem "sinatra"
-gem "resque", path: "../../"
+gem "resque", path: "../../"

examples/sinatra/Rakefile

@@ -0,0 +1 @@
+require "resque/tasks"

lib/resque/cli.rb

@@ -1,7 +1,11 @@
+require 'thor'
 require "resque"
 
 module Resque
   class CLI < Thor
+    class_option :config,    :aliases => ["-c"], :type => :string
+    class_option :redis,     :aliases => ["-R"], :type => :string
+    class_option :namespace, :aliases => ["-N"], :type => :string
 
     desc "work", "Start processing jobs."
     option :queue,     :aliases => ["-q"], :type => :string,  :default => "*"
@@ -13,30 +17,33 @@ class CLI < Thor
     #option :verbose,   :aliases => ["-v"], :type => :boolean, :default => false
     #option :vverbose,  :aliases => ["-vv"], :type => :boolean, :default => false
     def work
-      queues = options[:queue].to_s.split(',')
+      load_config
 
-      load_enviroment(options[:require])
-      worker = Resque::Worker.new(*queues)
+      load_enviroment(Resque.config.require)
+      worker = Resque::Worker.new(*Resque.config.queues)
 
-      worker.term_timeout = options[:timeout]
+      worker.term_timeout = Resque.config.timeout
       #worker.verbose = options[:verbose]
       #worker.very_verbose = options[:vverbose]
 
-      if options[:deamon]
+      if Resque.config.deamon
         Process.daemon(true)
       end
 
-      if options.has_key?(:pid)
-        File.open(options[:pid], 'w') { |f| f << worker.pid }
+      if Resque.config.pid
+        File.open(Resque.config.pid, 'w') { |f| f << worker.pid }
       end
 
       Resque.logger.info "Starting worker #{worker}"
 
-      worker.work(options[:interval]) # interval, will block
+      worker.work(Resque.config.interval) # interval, will block
     end
 
     desc "kill WORKER", "Kills a worker"
     def kill(worker)
+      load_config
+
+      before_action
       pid = worker.split(':')[1].to_i
 
       begin
@@ -51,12 +58,16 @@ def kill(worker)
 
     desc "remove WORKER", "Removes a worker"
     def remove(worker)
+      load_config
+
       Resque.remove_worker(worker)
       puts "Removed #{worker}"
     end
 
     desc "list", "Lists known workers"
     def list
+      load_config
+
       if Resque.workers.any?
         Resque.workers.each do |worker|
           puts "#{worker} (#{worker.state})"
@@ -68,6 +79,12 @@ def list
 
     protected
 
+      def load_config
+        Resque.config = Resque::Config.new(YAML.load_file(File.expand_path(options[:config]))) if options[:config]
+        Resque.config.redis = options[:redis] if options[:redis]
+        Resque.config.namespace = options[:namespace] if options[:namespace]
+      end
+
       def load_enviroment(file = nil)
         return if file.nil?
 

lib/resque/tasks.rb

@@ -0,0 +1,44 @@
+# require 'resque/tasks'
+# will give you the resque tasks
+require 'resque/cli'
+
+namespace :resque do
+  task :setup
+
+  desc "Start a Resque worker"
+  task :work => [ :preload, :setup ] do
+    Resque::CLI.new.work
+  end
+
+  desc "Start multiple Resque workers. Should only be used in dev mode."
+  task :workers do
+    Resque::CLI.new.workers
+  end
+
+  # Preload app files if this is Rails
+  task :preload => :setup do
+    if defined?(Rails) && Rails.respond_to?(:application)
+      # Rails 3
+      Rails.application.eager_load!
+    elsif defined?(Rails::Initializer)
+      # Rails 2.3
+      $rails_rake_task = false
+      Rails::Initializer.run :load_application_classes
+    end
+  end
+
+  namespace :failures do
+    desc "Sort the 'failed' queue for the redis_multi_queue failure backend"
+    task :sort do
+      require 'resque'
+      require 'resque/failure/redis'
+
+      warn "Sorting #{Resque::Failure.count} failures..."
+      Resque::Failure.each(0, Resque::Failure.count) do |_, failure|
+        data = Resque.encode(failure)
+        Resque.redis.rpush(Resque::Failure.failure_queue_name(failure['queue']), data)
+      end
+      warn "done!"
+    end
+  end
+end