Class: Sidekiq::JobSet

Inherits:

SortedSet

• Object
• SortedSet
• Sidekiq::JobSet

Defined in:

lib/sidekiq/api.rb

Direct Known Subclasses

DeadSet, RetrySet, ScheduledSet

Instance Attribute Summary

Attributes inherited from SortedSet

#name

Instance Method Summary

• #delete(score, jid = nil) ⇒ Object
• #each ⇒ Object
• #fetch(score, jid = nil) ⇒ Object
• #find_job(jid) ⇒ Object
• #schedule(timestamp, message) ⇒ Object

Methods inherited from SortedSet

#clear, #initialize, #size

Constructor Details

This class inherits a constructor from Sidekiq::SortedSet

Instance Method Details

#delete(score, jid = nil) ⇒ Object

# File 'lib/sidekiq/api.rb', line 522

def delete(score, jid = nil)
  if jid
    elements = Sidekiq.redis do |conn|
      conn.zrangebyscore(name, score, score)
    end

    elements_with_jid = elements.map do |element|
      message = Sidekiq.load_json(element)

      if message["jid"] == jid
        _, @_size = Sidekiq.redis do |conn|
          conn.multi do
            conn.zrem(name, element)
            conn.zcard name
          end
        end
      end
    end
    elements_with_jid.count != 0
  else
    count, @_size = Sidekiq.redis do |conn|
      conn.multi do
        conn.zremrangebyscore(name, score, score)
        conn.zcard name
      end
    end
    count != 0
  end
end

#each ⇒ Object

# File 'lib/sidekiq/api.rb', line 481

def each
  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 name, range_start, range_end, with_scores: true
    end
    break if elements.empty?
    page -= 1
    elements.each do |element, score|
      yield SortedEntry.new(self, score, element)
    end
    offset_size = initial_size - @_size
  end
end

#fetch(score, jid = nil) ⇒ Object

# File 'lib/sidekiq/api.rb', line 502

def fetch(score, jid = nil)
  elements = Sidekiq.redis do |conn|
    conn.zrangebyscore(name, 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

# File 'lib/sidekiq/api.rb', line 518

def find_job(jid)
  self.detect { |j| j.jid == jid }
end

#schedule(timestamp, message) ⇒ Object

# File 'lib/sidekiq/api.rb', line 475

def schedule(timestamp, message)
  Sidekiq.redis do |conn|
    conn.zadd(name, timestamp.to_f.to_s, Sidekiq.dump_json(message))
  end
end