From 22072262d7ff3335e438267fd0c8b00fac92ae89 Mon Sep 17 00:00:00 2001 From: Ismael Abreu Date: Fri, 16 Jan 2015 23:25:42 +0000 Subject: [PATCH] Protect private methods --- lib/sidekiq/api.rb | 88 ++++++++++++++++++++++++---------------------- 1 file changed, 45 insertions(+), 43 deletions(-) diff --git a/lib/sidekiq/api.rb b/lib/sidekiq/api.rb index afd00e0bf..9a6dc84e2 100644 --- a/lib/sidekiq/api.rb +++ b/lib/sidekiq/api.rb @@ -23,6 +23,51 @@ def dead_size stat 'dead_size' end + def fetch_stats! + stats + end + + def reset(*stats) + all = %w(failed processed) + stats = stats.empty? ? all : all & stats.flatten.compact.map(&:to_s) + + mset_args = [] + stats.each do |stat| + mset_args << "stat:#{stat}" + mset_args << 0 + end + Sidekiq.redis do |conn| + conn.mset(*mset_args) + end + end + + def queues + Sidekiq.redis do |conn| + queues = conn.smembers('queues'.freeze) + + lengths = conn.pipelined do + queues.each do |queue| + conn.llen("queue:#{queue}") + end + end + + i = 0 + array_of_arrays = queues.inject({}) do |memo, queue| + memo[queue] = lengths[i] + i += 1 + memo + end.sort_by { |_, size| size } + + Hash[array_of_arrays.reverse] + end + end + + def enqueued + queues.values.inject(&:+) || 0 + end + + private + def stats(*only) all = %w(processed failed queues enqueued scheduled_size retry_size dead_size) metrics = only.any? ? only : all @@ -38,10 +83,6 @@ def stats(*only) end end - def fetch_stats! - stats - end - def stat(s) stats(s)[s] end @@ -73,45 +114,6 @@ def load_stats(*metrics) loaded_stats end - def reset(*stats) - all = %w(failed processed) - stats = stats.empty? ? all : all & stats.flatten.compact.map(&:to_s) - - mset_args = [] - stats.each do |stat| - mset_args << "stat:#{stat}" - mset_args << 0 - end - Sidekiq.redis do |conn| - conn.mset(*mset_args) - end - end - - def queues - Sidekiq.redis do |conn| - queues = conn.smembers('queues'.freeze) - - lengths = conn.pipelined do - queues.each do |queue| - conn.llen("queue:#{queue}") - end - end - - i = 0 - array_of_arrays = queues.inject({}) do |memo, queue| - memo[queue] = lengths[i] - i += 1 - memo - end.sort_by { |_, size| size } - - Hash[array_of_arrays.reverse] - end - end - - def enqueued - queues.values.inject(&:+) || 0 - end - class History def initialize(days_previous, start_date = nil) @days_previous = days_previous