2012-03-04 15:58:16 -05:00
|
|
|
require 'sinatra/base'
|
|
|
|
require 'slim'
|
2012-07-18 01:14:15 -04:00
|
|
|
require 'sidekiq/paginator'
|
|
|
|
|
2012-03-04 15:58:16 -05:00
|
|
|
module Sidekiq
|
|
|
|
class Web < Sinatra::Base
|
2012-07-18 01:14:15 -04:00
|
|
|
include Sidekiq::Paginator
|
|
|
|
|
2012-03-05 22:27:27 -05:00
|
|
|
dir = File.expand_path(File.dirname(__FILE__) + "/../../web")
|
2012-11-02 14:11:34 -04:00
|
|
|
set :public_folder, "#{dir}/assets"
|
2012-03-04 15:58:16 -05:00
|
|
|
set :views, "#{dir}/views"
|
|
|
|
set :root, "#{dir}/public"
|
2012-03-05 23:53:14 -05:00
|
|
|
set :slim, :pretty => true
|
2012-09-30 08:57:51 -04:00
|
|
|
|
2012-03-05 23:53:14 -05:00
|
|
|
helpers do
|
2012-03-28 15:36:16 -04:00
|
|
|
|
2012-06-05 10:51:18 -04:00
|
|
|
def reset_worker_list
|
|
|
|
Sidekiq.redis do |conn|
|
|
|
|
workers = conn.smembers('workers')
|
|
|
|
workers.each do |name|
|
|
|
|
conn.srem('workers', name)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-03-05 23:53:14 -05:00
|
|
|
def workers
|
2012-03-06 23:17:42 -05:00
|
|
|
@workers ||= begin
|
2012-03-17 12:41:24 -04:00
|
|
|
Sidekiq.redis do |conn|
|
2012-03-07 23:22:21 -05:00
|
|
|
conn.smembers('workers').map do |w|
|
|
|
|
msg = conn.get("worker:#{w}")
|
2012-04-24 10:15:29 -04:00
|
|
|
msg ? [w, Sidekiq.load_json(msg)] : nil
|
|
|
|
end.compact.sort { |x| x[1] ? -1 : 1 }
|
2012-03-07 23:22:21 -05:00
|
|
|
end
|
2012-03-06 23:17:42 -05:00
|
|
|
end
|
2012-03-05 23:53:14 -05:00
|
|
|
end
|
2012-03-06 23:17:42 -05:00
|
|
|
|
2012-12-05 11:46:55 -05:00
|
|
|
def stats
|
|
|
|
@stats ||= Sidekiq::Stats.new
|
2012-03-07 23:22:21 -05:00
|
|
|
end
|
|
|
|
|
2012-12-05 11:46:55 -05:00
|
|
|
def scheduled_job_count
|
|
|
|
Sidekiq::ScheduledSet.new.size
|
2012-03-07 23:22:21 -05:00
|
|
|
end
|
|
|
|
|
2012-12-05 11:46:55 -05:00
|
|
|
def retry_job_count
|
|
|
|
Sidekiq::RetrySet.new.size
|
2012-05-22 10:16:08 -04:00
|
|
|
end
|
|
|
|
|
2012-03-29 13:48:06 -04:00
|
|
|
def retries_with_score(score)
|
|
|
|
Sidekiq.redis do |conn|
|
2012-03-29 23:55:16 -04:00
|
|
|
results = conn.zrangebyscore('retry', score, score)
|
2012-04-22 17:02:35 -04:00
|
|
|
results.map { |msg| Sidekiq.load_json(msg) }
|
2012-03-29 13:48:06 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-03-05 23:53:14 -05:00
|
|
|
def location
|
2012-03-17 12:41:24 -04:00
|
|
|
Sidekiq.redis { |conn| conn.client.location }
|
2012-03-05 23:53:14 -05:00
|
|
|
end
|
2012-03-06 23:17:42 -05:00
|
|
|
|
2012-03-05 23:53:14 -05:00
|
|
|
def root_path
|
|
|
|
"#{env['SCRIPT_NAME']}/"
|
|
|
|
end
|
2012-03-06 23:17:42 -05:00
|
|
|
|
2012-09-30 08:57:51 -04:00
|
|
|
def current_path
|
|
|
|
@current_path ||= request.path_info.gsub(/^\//,'')
|
|
|
|
end
|
|
|
|
|
2012-03-17 12:41:24 -04:00
|
|
|
def current_status
|
2012-04-24 10:15:29 -04:00
|
|
|
return 'idle' if workers.size == 0
|
2012-03-06 23:17:42 -05:00
|
|
|
return 'active'
|
|
|
|
end
|
2012-03-29 16:57:32 -04:00
|
|
|
|
|
|
|
def relative_time(time)
|
2012-03-30 01:24:22 -04:00
|
|
|
%{<time datetime="#{time.getutc.iso8601}">#{time}</time>}
|
2012-03-29 16:57:32 -04:00
|
|
|
end
|
2012-05-11 19:48:03 -04:00
|
|
|
|
2012-12-02 23:18:31 -05:00
|
|
|
def job_params(job, score)
|
|
|
|
"#{score}-#{job['jid']}"
|
|
|
|
end
|
|
|
|
|
2012-11-26 14:53:22 -05:00
|
|
|
def parse_params(params)
|
|
|
|
score, jid = params.split("-")
|
|
|
|
[score.to_f, jid]
|
|
|
|
end
|
|
|
|
|
2012-05-11 19:48:03 -04:00
|
|
|
def display_args(args, count=100)
|
|
|
|
args.map { |arg| a = arg.inspect; a.size > count ? "#{a[0..count]}..." : a }.join(", ")
|
|
|
|
end
|
2012-08-20 18:28:14 -04:00
|
|
|
|
|
|
|
def tabs
|
2012-08-21 11:06:20 -04:00
|
|
|
self.class.tabs
|
2012-08-20 18:28:14 -04:00
|
|
|
end
|
2012-08-31 10:53:08 -04:00
|
|
|
|
2012-09-01 00:50:19 -04:00
|
|
|
def number_with_delimiter(number)
|
2012-08-31 10:53:08 -04:00
|
|
|
begin
|
|
|
|
Float(number)
|
|
|
|
rescue ArgumentError, TypeError
|
2012-09-01 00:50:19 -04:00
|
|
|
return number
|
2012-08-31 10:53:08 -04:00
|
|
|
end
|
|
|
|
|
2012-09-01 00:50:19 -04:00
|
|
|
options = {:delimiter => ',', :separator => '.'}
|
2012-08-31 10:53:08 -04:00
|
|
|
parts = number.to_s.to_str.split('.')
|
|
|
|
parts[0].gsub!(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1#{options[:delimiter]}")
|
2012-08-31 13:14:00 -04:00
|
|
|
parts.join(options[:separator])
|
2012-08-31 10:53:08 -04:00
|
|
|
end
|
2012-03-05 23:53:14 -05:00
|
|
|
end
|
|
|
|
|
2012-03-04 15:58:16 -05:00
|
|
|
get "/" do
|
|
|
|
slim :index
|
|
|
|
end
|
2012-03-05 23:53:14 -05:00
|
|
|
|
2012-06-08 20:20:11 -04:00
|
|
|
get "/queues" do
|
2012-12-04 16:57:48 -05:00
|
|
|
@queues = Sidekiq::Stats.new.queues
|
2012-06-08 20:20:11 -04:00
|
|
|
slim :queues
|
|
|
|
end
|
|
|
|
|
2012-03-05 23:53:14 -05:00
|
|
|
get "/queues/:name" do
|
2012-03-17 12:41:24 -04:00
|
|
|
halt 404 unless params[:name]
|
2012-07-18 01:14:15 -04:00
|
|
|
@count = (params[:count] || 25).to_i
|
2012-03-05 23:53:14 -05:00
|
|
|
@name = params[:name]
|
2012-07-18 01:14:15 -04:00
|
|
|
(@current_page, @total_size, @messages) = page("queue:#{@name}", params[:page], @count)
|
|
|
|
@messages = @messages.map {|msg| Sidekiq.load_json(msg) }
|
2012-03-05 23:53:14 -05:00
|
|
|
slim :queue
|
|
|
|
end
|
2012-03-29 13:48:06 -04:00
|
|
|
|
2012-06-05 10:51:18 -04:00
|
|
|
post "/reset" do
|
|
|
|
reset_worker_list
|
|
|
|
redirect root_path
|
|
|
|
end
|
|
|
|
|
2012-05-02 23:19:59 -04:00
|
|
|
post "/queues/:name" do
|
2012-11-25 21:43:48 -05:00
|
|
|
Sidekiq::Queue.new(params[:name]).clear
|
2012-06-08 20:20:11 -04:00
|
|
|
redirect "#{root_path}queues"
|
2012-05-02 23:19:59 -04:00
|
|
|
end
|
|
|
|
|
2012-10-23 19:11:14 -04:00
|
|
|
post "/queues/:name/delete" do
|
2012-12-02 16:23:05 -05:00
|
|
|
Sidekiq::Job.new(params[:key_val], params[:name]).delete
|
2012-10-23 19:11:14 -04:00
|
|
|
redirect "#{root_path}queues/#{params[:name]}"
|
|
|
|
end
|
|
|
|
|
2012-05-11 19:48:03 -04:00
|
|
|
get '/retries' do
|
2012-07-18 01:14:15 -04:00
|
|
|
@count = (params[:count] || 25).to_i
|
|
|
|
(@current_page, @total_size, @retries) = page("retry", params[:page], @count)
|
|
|
|
@retries = @retries.map {|msg, score| [Sidekiq.load_json(msg), score] }
|
2012-05-11 19:48:03 -04:00
|
|
|
slim :retries
|
|
|
|
end
|
|
|
|
|
2012-11-26 14:53:22 -05:00
|
|
|
get "/retries/:key" do
|
|
|
|
halt 404 unless params['key']
|
|
|
|
@retry = Sidekiq::RetrySet.new.fetch(*parse_params(params['key'])).first
|
2012-11-26 11:22:48 -05:00
|
|
|
redirect "#{root_path}retries" if @retry.nil?
|
|
|
|
slim :retry
|
2012-05-25 23:21:42 -04:00
|
|
|
end
|
|
|
|
|
2012-05-11 19:48:03 -04:00
|
|
|
post '/retries' do
|
2012-11-26 14:53:22 -05:00
|
|
|
halt 404 unless params['key']
|
|
|
|
|
|
|
|
params['key'].each do |key|
|
|
|
|
job = Sidekiq::RetrySet.new.fetch(*parse_params(key)).first
|
|
|
|
if params['retry']
|
|
|
|
job.retry
|
|
|
|
elsif params['delete']
|
|
|
|
job.delete
|
|
|
|
end
|
2012-05-11 19:48:03 -04:00
|
|
|
end
|
2012-06-18 07:59:25 -04:00
|
|
|
redirect "#{root_path}retries"
|
2012-05-11 19:48:03 -04:00
|
|
|
end
|
|
|
|
|
2012-11-25 21:43:48 -05:00
|
|
|
post "/retries/all/delete" do
|
|
|
|
Sidekiq::RetrySet.new.clear
|
|
|
|
redirect "#{root_path}retries"
|
|
|
|
end
|
|
|
|
|
|
|
|
post "/retries/all/retry" do
|
2012-12-02 23:14:12 -05:00
|
|
|
Sidekiq::RetrySet.new.each { |job| job.retry }
|
2012-11-25 21:43:48 -05:00
|
|
|
redirect "#{root_path}retries"
|
2012-05-11 19:48:03 -04:00
|
|
|
end
|
|
|
|
|
2012-11-26 14:53:22 -05:00
|
|
|
post "/retries/:key" do
|
|
|
|
halt 404 unless params['key']
|
|
|
|
job = Sidekiq::RetrySet.new.fetch(*parse_params(params['key'])).first
|
2012-03-29 23:55:16 -04:00
|
|
|
if params['retry']
|
2012-11-26 14:53:22 -05:00
|
|
|
job.retry
|
2012-05-11 19:48:03 -04:00
|
|
|
elsif params['delete']
|
2012-11-26 14:53:22 -05:00
|
|
|
job.delete
|
2012-05-11 19:48:03 -04:00
|
|
|
end
|
2012-06-18 07:59:25 -04:00
|
|
|
redirect "#{root_path}retries"
|
2012-05-11 19:48:03 -04:00
|
|
|
end
|
|
|
|
|
2012-11-26 11:22:48 -05:00
|
|
|
get '/scheduled' do
|
|
|
|
@count = (params[:count] || 25).to_i
|
|
|
|
(@current_page, @total_size, @scheduled) = page("schedule", params[:page], @count)
|
|
|
|
@scheduled = @scheduled.map {|msg, score| [Sidekiq.load_json(msg), score] }
|
|
|
|
slim :scheduled
|
|
|
|
end
|
|
|
|
|
|
|
|
post '/scheduled' do
|
2012-11-26 14:53:22 -05:00
|
|
|
halt 404 unless params['key']
|
2012-11-26 11:22:48 -05:00
|
|
|
halt 404 unless params['delete']
|
2012-11-26 14:53:22 -05:00
|
|
|
params['key'].each do |key|
|
|
|
|
Sidekiq::ScheduledSet.new.fetch(*parse_params(key)).first.delete
|
|
|
|
end
|
2012-11-26 11:22:48 -05:00
|
|
|
redirect "#{root_path}scheduled"
|
2012-03-29 23:55:16 -04:00
|
|
|
end
|
2012-05-11 19:48:03 -04:00
|
|
|
|
2012-08-20 18:28:14 -04:00
|
|
|
def self.tabs
|
2012-09-30 08:57:51 -04:00
|
|
|
@tabs ||= {
|
|
|
|
"Workers" =>'',
|
|
|
|
"Queues" =>'queues',
|
|
|
|
"Retries" =>'retries',
|
|
|
|
"Scheduled" =>'scheduled'
|
|
|
|
}
|
2012-08-20 18:28:14 -04:00
|
|
|
end
|
|
|
|
|
2012-03-04 15:58:16 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
end
|
2012-11-02 14:11:34 -04:00
|
|
|
|