mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
177 lines
4.4 KiB
Ruby
177 lines
4.4 KiB
Ruby
require 'uri'
|
|
|
|
module Sidekiq
|
|
# This is not a public API
|
|
module WebHelpers
|
|
def strings
|
|
@@strings ||= begin
|
|
# Allow sidekiq-web extensions to add locale paths
|
|
# so extensions can be localized
|
|
settings.locales.each_with_object({}) do |path,global|
|
|
Dir["#{path}/*.yml"].each_with_object(global) do |file,hash|
|
|
strs = YAML.load(File.open(file))
|
|
hash.deep_merge!(strs)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def locale
|
|
lang = (request.env["HTTP_ACCEPT_LANGUAGE"] || 'en')[0,2]
|
|
strings[lang] ? lang : 'en'
|
|
end
|
|
|
|
def get_locale
|
|
strings[locale]
|
|
end
|
|
|
|
def t(msg, options={})
|
|
string = get_locale[msg] || msg
|
|
string % options
|
|
end
|
|
|
|
def reset_worker_list
|
|
Sidekiq.redis do |conn|
|
|
workers = conn.smembers('workers')
|
|
conn.srem('workers', workers) if !workers.empty?
|
|
end
|
|
end
|
|
|
|
def workers_size
|
|
@workers_size ||= Sidekiq.redis do |conn|
|
|
conn.scard('workers')
|
|
end
|
|
end
|
|
|
|
def workers
|
|
@workers ||= begin
|
|
to_rem = []
|
|
workers = Sidekiq.redis do |conn|
|
|
conn.smembers('workers').map do |w|
|
|
msg = conn.get("worker:#{w}")
|
|
msg ? [w, Sidekiq.load_json(msg)] : (to_rem << w; nil)
|
|
end.compact.sort { |x| x[1] ? -1 : 1 }
|
|
end
|
|
|
|
# Detect and clear out any orphaned worker records.
|
|
# These can be left in Redis if Sidekiq crashes hard
|
|
# while processing jobs.
|
|
if to_rem.size > 0
|
|
Sidekiq.redis { |conn| conn.srem('workers', to_rem) }
|
|
end
|
|
workers
|
|
end
|
|
end
|
|
|
|
def stats
|
|
@stats ||= Sidekiq::Stats.new
|
|
end
|
|
|
|
def retries_with_score(score)
|
|
Sidekiq.redis do |conn|
|
|
conn.zrangebyscore('retry', score, score)
|
|
end.map { |msg| Sidekiq.load_json(msg) }
|
|
end
|
|
|
|
def location
|
|
Sidekiq.redis { |conn| conn.client.location }
|
|
end
|
|
|
|
def redis_connection
|
|
Sidekiq.redis { |conn| conn.client.id }
|
|
end
|
|
|
|
def namespace
|
|
@@ns ||= Sidekiq.redis {|conn| conn.respond_to?(:namespace) ? conn.namespace : nil }
|
|
end
|
|
|
|
def root_path
|
|
"#{env['SCRIPT_NAME']}/"
|
|
end
|
|
|
|
def current_path
|
|
@current_path ||= request.path_info.gsub(/^\//,'')
|
|
end
|
|
|
|
def current_status
|
|
workers_size == 0 ? 'idle' : 'active'
|
|
end
|
|
|
|
def relative_time(time)
|
|
%{<time datetime="#{time.getutc.iso8601}">#{time}</time>}
|
|
end
|
|
|
|
def job_params(job, score)
|
|
"#{score}-#{job['jid']}"
|
|
end
|
|
|
|
def parse_params(params)
|
|
score, jid = params.split("-")
|
|
[score.to_f, jid]
|
|
end
|
|
|
|
SAFE_QPARAMS = %w(page poll)
|
|
|
|
# Merge options with current params, filter safe params, and stringify to query string
|
|
def qparams(options)
|
|
options = options.stringify_keys
|
|
params.merge(options).map { |key, value|
|
|
SAFE_QPARAMS.include?(key) ? "#{key}=#{value}" : next
|
|
}.join("&")
|
|
end
|
|
|
|
def truncate(text, truncate_after_chars = 2000)
|
|
truncate_after_chars && text.size > truncate_after_chars ? "#{text[0..truncate_after_chars]}..." : text
|
|
end
|
|
|
|
def display_args(args, truncate_after_chars = 2000)
|
|
args.map do |arg|
|
|
a = arg.inspect
|
|
h(truncate(a))
|
|
end.join(", ")
|
|
end
|
|
|
|
RETRY_JOB_KEYS = Set.new(%w(
|
|
queue class args retry_count retried_at failed_at
|
|
jid error_message error_class backtrace
|
|
error_backtrace enqueued_at retry
|
|
))
|
|
|
|
def retry_extra_items(retry_job)
|
|
@retry_extra_items ||= {}.tap do |extra|
|
|
retry_job.item.each do |key, value|
|
|
extra[key] = value unless RETRY_JOB_KEYS.include?(key)
|
|
end
|
|
end
|
|
end
|
|
|
|
def number_with_delimiter(number)
|
|
begin
|
|
Float(number)
|
|
rescue ArgumentError, TypeError
|
|
return number
|
|
end
|
|
|
|
options = {:delimiter => ',', :separator => '.'}
|
|
parts = number.to_s.to_str.split('.')
|
|
parts[0].gsub!(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1#{options[:delimiter]}")
|
|
parts.join(options[:separator])
|
|
end
|
|
|
|
def h(text)
|
|
Rack::Utils.escape_html(text)
|
|
end
|
|
|
|
# Any paginated list that performs an action needs to redirect
|
|
# back to the proper page after performing that action.
|
|
def redirect_with_query(url)
|
|
r = request.referer
|
|
if r && r =~ /\?/
|
|
ref = URI(r)
|
|
redirect("#{url}?#{ref.query}")
|
|
else
|
|
redirect url
|
|
end
|
|
end
|
|
end
|
|
end
|