Module: ResqueSpec

Extended by:
ResqueSpec
Includes:
Resque::Helpers
Included in:
ResqueSpec
Defined in:
lib/tresque/resque_spec/helpers.rb,
lib/tresque/resque_spec/scheduler.rb,
lib/tresque/resque_spec/resque_spec.rb

Defined Under Namespace

Modules: BaseMethods, Helpers, SchedulerExt, SchedulerExtMethods

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#disable_ext ⇒ Object

Returns the value of attribute disable_ext.



25
26
27
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 25

def disable_ext
  @disable_ext
end

#inline ⇒ Object

Returns the value of attribute inline.



24
25
26
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 24

def inline
  @inline
end

Instance Method Details

#clear_all! ⇒ Object 



145
146
147
148
149
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 145

def clear_all!
  [*Resque.redis.keys].each do |key|
    Resque.redis.del(key)
  end
end

#clear_locked! ⇒ Object 



139
140
141
142
143
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 139

def clear_locked!
  [*Resque.redis.keys("lock:*")].each do |key|
    Resque.redis.del(key)
  end
end

#delayed?(klass, *args) ⇒ Boolean

Check if we have queued a delayed worker for ‘klass` with `*args` Very slow, checks every `delayed` key

Returns:

  • (Boolean) 


106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 106

def delayed?(klass, *args)
  klass = klass.to_s unless klass.is_a? String
  [*Resque.redis.keys("delayed:*")].each do |key|
    [*Resque.redis.lrange(key, 0, -1)].each do |item|
      parsed_item = JSON.parse(item)
      return true if parsed_item['class'] == klass && parsed_item['args'] == [*args]
    end
  end

  false
end

#delayed_key(klass, *args) ⇒ Object

Get back the key of the delayed worker if it exists



119
120
121
122
123
124
125
126
127
128
129
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 119

def delayed_key(klass, *args)
  klass = klass.to_s unless klass.is_a? String
  [*Resque.redis.keys("delayed:*")].each do |key|
    [*Resque.redis.lrange(key, 0, -1)].each do |item|
      parsed_item = JSON.parse(item)
      return key if parsed_item['class'] == klass && parsed_item['args'] == [*args]
    end
  end

  nil
end

#delete_all(queue_name) ⇒ Object 



92
93
94
95
96
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 92

def delete_all(queue_name)
  queue = "queue:#{queue_name}"
  Resque.redis.del(queue)
  reset!
end

#dequeue(queue_name, klass, *args) ⇒ Object 



27
28
29
30
31
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 27

def dequeue(queue_name, klass, *args)
  queue_by_name(queue_name).delete_if do |job|
    job[:class] == klass.to_s && args.empty? || job[:args] == args
  end
end

#enqueue(queue_name, klass, *args) ⇒ Object 



33
34
35
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 33

def enqueue(queue_name, klass, *args)
  perform_or_store(queue_name, :class => klass.to_s, :args => args)
end

#enqueue_at(time, klass, *args) ⇒ Object 



49
50
51
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 49

def enqueue_at(time, klass, *args)
  enqueue_at_with_queue(schedule_queue_name(klass), time, klass, *args)
end

#enqueue_at_with_queue(queue, time, klass, *args) ⇒ Object 



53
54
55
56
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 53

def enqueue_at_with_queue(queue, time, klass, *args)
  is_time?(time)
  perform_or_store(queue, :class => klass.to_s, :time  => time, :stored_at => Time.now, :args => args)
end

#enqueue_in(time, klass, *args) ⇒ Object 



58
59
60
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 58

def enqueue_in(time, klass, *args)
  enqueue_at(Time.now + time, klass, *args)
end

#enqueue_in_with_queue(queue, time, klass, *args) ⇒ Object 



62
63
64
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 62

def enqueue_in_with_queue(queue, time, klass, *args)
  enqueue_at_with_queue(queue, Time.now + time, klass, *args)
end

#locked?(klass, *args) ⇒ Boolean

check if the worker has a lock with the provided args

Returns:

  • (Boolean) 


132
133
134
135
136
137
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 132

def locked?(klass, *args)
  klass = klass.to_s unless klass.is_a? String

  key = "lock:#{klass}-#{[*args].join('-')}"
  Resque.redis.keys(key).present?
end

#peek(queue_name, start = 0, count = 1) ⇒ Object 



74
75
76
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 74

def peek(queue_name, start = 0, count = 1)
  queue_by_name(queue_name).slice(start, count)
end

#perform_all(queue_name) ⇒ Object 



41
42
43
44
45
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 41

def perform_all(queue_name)
  while job = shift_queue(queue_name)
    perform(queue_name, job)
  end
end

#perform_all! ⇒ Object 



98
99
100
101
102
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 98

def perform_all!
  ::TResque::Registry.queues.each do |queue_name|
    ResqueSpec.perform_all(queue_name)
  end
end

#perform_next(queue_name) ⇒ Object 



37
38
39
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 37

def perform_next(queue_name)
  perform(queue_name, shift_queue(queue_name))
end

#pop(queue_name) ⇒ Object 



61
62
63
64
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 61

def pop(queue_name)
  return unless payload = shift_queue(queue_name)
  new_job(queue_name, payload)
end

#queue_by_name(name) ⇒ Object 



66
67
68
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 66

def queue_by_name(name)
  queues[name.to_s]
end

#queue_for(klass) ⇒ Object 



70
71
72
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 70

def queue_for(klass)
  queue_by_name(queue_name(klass))
end

#queue_name(klass) ⇒ Object 



78
79
80
81
82
83
84
85
86
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 78

def queue_name(klass)
  if klass.is_a?(String)
    klass = Kernel.const_get(klass) rescue nil
  end

  name_from_instance_var(klass) or
    name_from_queue_accessor(klass) or
      raise ::Resque::NoQueueError.new("Jobs must be placed onto a queue.")
end

#queues ⇒ Object 



88
89
90
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 88

def queues
  @queues ||= Hash.new {|h,k| h[k] = []}
end

#remove_delayed(klass, *args) ⇒ Object 



66
67
68
69
70
71
72
73
74
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 66

def remove_delayed(klass, *args)
  sched_queue = queue_by_name(schedule_queue_name(klass))
  count_before_remove = sched_queue.length
  sched_queue.delete_if do |job|
    job[:class] == klass.to_s && job[:args] == args
  end
  # Return number of removed items to match Resque Scheduler behaviour
  count_before_remove - sched_queue.length
end

#reset! ⇒ Object 



151
152
153
154
155
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 151

def reset!
  clear_all!
  queues.clear
  self.inline = false
end

#schedule_for(klass) ⇒ Object 



76
77
78
 
# File 'lib/tresque/resque_spec/scheduler.rb', line 76

def schedule_for(klass)
  queue_by_name(schedule_queue_name(klass))
end

#shift_queue(queue_name) ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/tresque/resque_spec/resque_spec.rb', line 47

def shift_queue(queue_name)
  # pass scheduled ones
  array = queue_by_name(queue_name) || []
  index = nil
  array.each_with_index do |hash, i|
    if hash[:time].to_i <= Time.now.to_i
      index = i
      break
    end
  end
  return nil unless index
  array.delete_at(index)
end