Class: SweatShop::Worker

Inherits:
Object
  • Object
show all
Defined in:
lib/sweat_shop/worker.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.after_task(&block) ⇒ Object 



121
122
123
124
125
126
127
 
# File 'lib/sweat_shop/worker.rb', line 121

def self.after_task(&block)
  if block
    @after_task = block
  else
    @after_task
  end
end

.before_task(&block) ⇒ Object 



113
114
115
116
117
118
119
 
# File 'lib/sweat_shop/worker.rb', line 113

def self.before_task(&block)
  if block
    @before_task = block
  else
    @before_task
  end
end

.call_after_task(task) ⇒ Object 



92
93
94
95
 
# File 'lib/sweat_shop/worker.rb', line 92

def self.call_after_task(task)
  superclass.call_after_task(task) if superclass.respond_to?(:call_after_task)
  after_task.call(task) if after_task
end

.call_before_task(task) ⇒ Object 



87
88
89
90
 
# File 'lib/sweat_shop/worker.rb', line 87

def self.call_before_task(task)
  superclass.call_before_task(task) if superclass.respond_to?(:call_before_task)
  before_task.call(task) if before_task
end

.config ⇒ Object 



36
37
38
 
# File 'lib/sweat_shop/worker.rb', line 36

def self.config
  SweatShop.config
end

.confirm ⇒ Object 



60
61
62
 
# File 'lib/sweat_shop/worker.rb', line 60

def self.confirm
  queue.confirm(queue_name)
end

.delete_queue ⇒ Object 



44
45
46
 
# File 'lib/sweat_shop/worker.rb', line 44

def self.delete_queue
  queue.delete(queue_name)
end

.dequeue ⇒ Object 



56
57
58
 
# File 'lib/sweat_shop/worker.rb', line 56

def self.dequeue
  queue.dequeue(queue_name)
end

.do_task(task) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
 
# File 'lib/sweat_shop/worker.rb', line 76

def self.do_task(task)
  call_before_task(task)

  queued_at = task[:queued_at] ? "(queued #{Time.at(task[:queued_at]).strftime('%Y/%m/%d %H:%M:%S')})" : ''
  log("Dequeuing #{queue_name}::#{task[:method]} #{queued_at}")
  task[:result] = instance.send(task[:method], *task[:args])

  call_after_task(task)
  confirm
end

.do_tasks ⇒ Object 



70
71
72
73
74
 
# File 'lib/sweat_shop/worker.rb', line 70

def self.do_tasks
  while task = dequeue
    do_task(task)
  end
end

.enqueue(task) ⇒ Object 



52
53
54
 
# File 'lib/sweat_shop/worker.rb', line 52

def self.enqueue(task)
  queue.enqueue(queue_name, task)
end

.inherited(subclass) ⇒ Object 



5
6
7
 
# File 'lib/sweat_shop/worker.rb', line 5

def self.inherited(subclass)
  self.workers << subclass
end

.instance ⇒ Object 



32
33
34
 
# File 'lib/sweat_shop/worker.rb', line 32

def self.instance
  @instance ||= new
end

.log(msg) ⇒ Object 



109
110
111
 
# File 'lib/sweat_shop/worker.rb', line 109

def self.log(msg)
  SweatShop.log(msg)
end

.method_missing(method, *args, &block) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/sweat_shop/worker.rb', line 9

def self.method_missing(method, *args, &block)
  if method.to_s =~ /^async_(.*)/
    method        = $1
    expected_args = instance.method(method).arity
    if expected_args != args.size 
      raise ArgumentError.new("#{method} expects #{expected_args} arguments")
    end
    return instance.send(method, *args) unless config['enable']

    uid  = ::Digest::MD5.hexdigest("#{name}:#{method}:#{args}:#{Time.now.to_f}")
    task = {:args => args, :method => method, :uid => uid, :queued_at => Time.now.to_i}

    log("Putting #{uid} on #{queue_name}")
    enqueue(task)

    uid
  elsif instance.respond_to?(method)
    instance.send(method, *args)
  else
    super
  end
end

.queue ⇒ Object 



97
98
99
 
# File 'lib/sweat_shop/worker.rb', line 97

def self.queue
  SweatShop.queue(queue_group.to_s)
end

.queue_group(group = nil) ⇒ Object 



137
138
139
 
# File 'lib/sweat_shop/worker.rb', line 137

def self.queue_group(group=nil)
  group ? meta_def(:_queue_group){ group } : _queue_group
end

.queue_name ⇒ Object 



40
41
42
 
# File 'lib/sweat_shop/worker.rb', line 40

def self.queue_name
  @queue_name ||= self.to_s
end

.queue_size ⇒ Object 



48
49
50
 
# File 'lib/sweat_shop/worker.rb', line 48

def self.queue_size
  queue.queue_size(queue_name)
end

.stop ⇒ Object 



129
130
131
 
# File 'lib/sweat_shop/worker.rb', line 129

def self.stop
  instance.stop
end

.subscribe ⇒ Object 



64
65
66
67
68
 
# File 'lib/sweat_shop/worker.rb', line 64

def self.subscribe
  queue.subscribe(queue_name) do |task|
    do_task(task)
  end
end

.workers ⇒ Object 



101
102
103
 
# File 'lib/sweat_shop/worker.rb', line 101

def self.workers
  SweatShop.workers
end

Instance Method Details

#stop ⇒ Object

called before we exit – subclass can implement this method



134
 
# File 'lib/sweat_shop/worker.rb', line 134

def stop; end