Class: Sidekiq::SortedSet
- Inherits:
-
Object
- Object
- Sidekiq::SortedSet
- Includes:
- Enumerable
- Defined in:
- lib/sidekiq/api.rb
Direct Known Subclasses
Instance Method Summary collapse
- #clear ⇒ Object
- #delete(score, jid = nil) ⇒ Object
- #each(&block) ⇒ Object
- #fetch(score, jid = nil) ⇒ Object
- #find_job(jid) ⇒ Object
-
#initialize(name) ⇒ SortedSet
constructor
A new instance of SortedSet.
- #schedule(timestamp, message) ⇒ Object
- #size ⇒ Object
Constructor Details
#initialize(name) ⇒ SortedSet
Returns a new instance of SortedSet.
260 261 262 263 |
# File 'lib/sidekiq/api.rb', line 260 def initialize(name) @zset = name @_size = size end |
Instance Method Details
#clear ⇒ Object
346 347 348 349 350 |
# File 'lib/sidekiq/api.rb', line 346 def clear Sidekiq.redis do |conn| conn.del(@zset) end end |
#delete(score, jid = nil) ⇒ Object
316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 |
# File 'lib/sidekiq/api.rb', line 316 def delete(score, jid = nil) if jid elements = Sidekiq.redis do |conn| conn.zrangebyscore(@zset, score, score) end elements_with_jid = elements.map do |element| = Sidekiq.load_json(element) if ["jid"] == jid _, @_size = Sidekiq.redis do |conn| conn.multi do conn.zrem(@zset, element) conn.zcard @zset end end end end elements_with_jid.count != 0 else count, @_size = Sidekiq.redis do |conn| conn.multi do conn.zremrangebyscore(@zset, score, score) conn.zcard @zset end end count != 0 end end |
#each(&block) ⇒ Object
275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 |
# File 'lib/sidekiq/api.rb', line 275 def each(&block) initial_size = @_size offset_size = 0 page = -1 page_size = 50 loop do range_start = page * page_size + offset_size range_end = page * page_size + offset_size + (page_size - 1) elements = Sidekiq.redis do |conn| conn.zrange @zset, range_start, range_end, :with_scores => true end break if elements.empty? page -= 1 elements.each do |element, score| block.call SortedEntry.new(self, score, element) end offset_size = initial_size - @_size end end |
#fetch(score, jid = nil) ⇒ Object
296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 |
# File 'lib/sidekiq/api.rb', line 296 def fetch(score, jid = nil) elements = Sidekiq.redis do |conn| conn.zrangebyscore(@zset, score, score) end elements.inject([]) do |result, element| entry = SortedEntry.new(self, score, element) if jid result << entry if entry.jid == jid else result << entry end result end end |
#find_job(jid) ⇒ Object
312 313 314 |
# File 'lib/sidekiq/api.rb', line 312 def find_job(jid) self.detect { |j| j.jid == jid } end |