Module: SidekiqUniqueJobs::Util
- Defined in:
- lib/sidekiq_unique_jobs/util.rb
Class Method Summary collapse
- .batch_delete(keys) ⇒ Object
- .connection(&block) ⇒ Object
- .del(pattern = SCAN_PATTERN, count = 0, dry_run = true) ⇒ Object
- .dry_run(keys, pattern = nil) ⇒ Object
- .keys(pattern = SCAN_PATTERN, count = DEFAULT_COUNT) ⇒ Object
- .keys_by_keys(pattern, _count) ⇒ Object
- .keys_by_scan(pattern, count) ⇒ Object
- .logger ⇒ Object
- .prefix(key) ⇒ Object
- .prefix_keys(keys) ⇒ Object
- .redis_keys_method ⇒ Object
- .redis_version ⇒ Object
- .timed(&block) ⇒ Object
- .unique_prefix ⇒ Object
Class Method Details
.batch_delete(keys) ⇒ Object
36 37 38 39 40 41 42 43 44 45 46 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 36 def batch_delete(keys) connection do |conn| keys.each_slice(500) do |chunk| conn.pipelined do chunk.each do |key| conn.del key end end end end end |
.connection(&block) ⇒ Object
75 76 77 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 75 def connection(&block) SidekiqUniqueJobs.connection(&block) end |
.del(pattern = SCAN_PATTERN, count = 0, dry_run = true) ⇒ Object
14 15 16 17 18 19 20 21 22 23 24 25 26 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 14 def del(pattern = SCAN_PATTERN, count = 0, dry_run = true) logger.debug { "Deleting keys by: #{pattern}" } keys, time = timed { keys(pattern, count) } logger.debug { "#{keys.size} matching keys found in #{time} sec." } keys = dry_run(keys) logger.debug { "#{keys.size} matching keys after postprocessing" } unless dry_run logger.debug { "deleting #{keys}..." } _, time = timed { batch_delete(keys) } logger.debug { "Deleted in #{time} sec." } end keys.size end |
.dry_run(keys, pattern = nil) ⇒ Object
48 49 50 51 52 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 48 def dry_run(keys, pattern = nil) return keys if pattern.nil? regex = Regexp.new(pattern) keys.select { |k| regex.match k } end |
.keys(pattern = SCAN_PATTERN, count = DEFAULT_COUNT) ⇒ Object
10 11 12 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 10 def keys(pattern = SCAN_PATTERN, count = DEFAULT_COUNT) send("keys_by_#{redis_keys_method}", pattern, count) end |
.keys_by_keys(pattern, _count) ⇒ Object
32 33 34 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 32 def keys_by_keys(pattern, _count) connection { |conn| conn.keys(prefix(pattern)).to_a } end |
.keys_by_scan(pattern, count) ⇒ Object
28 29 30 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 28 def keys_by_scan(pattern, count) connection { |conn| conn.scan_each(match: prefix(pattern), count: count).to_a } end |
.logger ⇒ Object
87 88 89 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 87 def logger Sidekiq.logger end |
.prefix(key) ⇒ Object
66 67 68 69 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 66 def prefix(key) return key if unique_prefix.nil? "#{unique_prefix}:#{key}" end |
.prefix_keys(keys) ⇒ Object
61 62 63 64 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 61 def prefix_keys(keys) keys = Array(keys).flatten.compact keys.map { |key| prefix(key) } end |
.redis_keys_method ⇒ Object
83 84 85 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 83 def redis_keys_method redis_version >= '2.8' ? SCAN_METHOD : KEYS_METHOD end |
.redis_version ⇒ Object
79 80 81 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 79 def redis_version SidekiqUniqueJobs.redis_version end |
.timed(&block) ⇒ Object
54 55 56 57 58 59 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 54 def timed(&block) start = Time.now result = block.call elapsed = (Time.now - start).round(2) [result, elapsed] end |
.unique_prefix ⇒ Object
71 72 73 |
# File 'lib/sidekiq_unique_jobs/util.rb', line 71 def unique_prefix SidekiqUniqueJobs.config.unique_prefix end |