server.rb

require 'cgi'
require 'resque/server'
require 'resque/scheduler/server'

# Extend Resque::Server to add tabs.
module GMQ
  module Server

    # Adds `resque-retry` web interface elements to `resque-web`
    #
    # @api private
    def self.included(base)
      base.class_eval {

        # get '/gmq' do
        #   erb local_template('retry.erb')
        # end

        get '/gmq/:timestamp' do
          erb local_template('retry_timestamp.erb')
        end

        get '/gmq' do
          erb local_template('retryAll.erb')
        end

        get '/gmq-slow-retry' do
          erb local_template('retryAllSlowly.erb')
        end

        post '/gmq/:timestamp/remove' do
          Resque.delayed_timestamp_peek(params[:timestamp], 0, 0).each do |job|
            cancel_retry(job)
          end
          redirect u('retry')
        end

        post '/gmq/:timestamp/jobs/:id/remove' do
          job = Resque.decode(params[:id])
          cancel_retry(job)
          redirect u("retry/#{params[:timestamp]}")
        end
      }
    end

    # Helper methods used by retry tab.
    module Helpers
      # builds a retry key for the specified job.
      def retry_key_for_job(job)
        klass = get_class(job)
        if klass.respond_to?(:redis_retry_key)
          klass.redis_retry_key(job['args'])
        else
          nil
        end
      end

      # gets the number of retry attempts for a job.
      def retry_attempts_for_job(job)
        Resque.redis.get(retry_key_for_job(job))
      end

      # gets the failure details hash for a job.
      def retry_failure_details(retry_key)
        Resque.decode(Resque.redis.get("failure-#{retry_key}"))
      end

      # reads a 'local' template file.
      def local_template(path)
        # Is there a better way to specify alternate template locations with sinatra?
        File.read(File.join(File.dirname(__FILE__), "server/views/#{path}"))
      end

      # cancels job retry
      def cancel_retry(job)
        klass = get_class(job)
        retry_key = retry_key_for_job(job)
        Resque.remove_delayed(klass, *job['args'])
        Resque.redis.del("failure-#{retry_key}")
        Resque.redis.del(retry_key)
      end

      private
      def get_class(job)
        Resque::Job.new(nil, nil).constantize(job['class'])
      end
    end

  end
end

Resque::Server.tabs << 'GMQ'
Resque::Server.tabs << 'GMQ-Slow-Retry'
Resque::Server.class_eval do
  include GMQ::Server
  helpers GMQ::Server::Helpers
end