1
0
Fork 0
mirror of https://github.com/mperham/sidekiq.git synced 2022-11-09 13:52:34 -05:00
mperham--sidekiq/lib/sidekiq/web_helpers.rb
2013-10-30 23:10:23 +01:00

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