Class: Sweatshop::Worker

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

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.after_task(&block) ⇒ Object 



127
128
129
130
131
132
133
 
# File 'lib/sweatshop/worker.rb', line 127

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

.before_task(&block) ⇒ Object 



119
120
121
122
123
124
125
 
# File 'lib/sweatshop/worker.rb', line 119

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

.call_after_task(task) ⇒ Object 



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

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 



104
105
106
107
 
# File 'lib/sweatshop/worker.rb', line 104

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

.call_exception_handler(exception) ⇒ Object 



114
115
116
117
 
# File 'lib/sweatshop/worker.rb', line 114

def self.call_exception_handler(exception)
  superclass.call_exception_handler(exception) if superclass.respond_to?(:call_exception_handler)
  on_exception.call(exception) if on_exception
end

.config ⇒ Object 



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

def self.config
  Sweatshop.config
end

.confirm ⇒ Object 



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

def self.confirm
  queue.confirm(queue_name)
end

.delete_queue ⇒ Object 



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

def self.delete_queue
  queue.delete(queue_name)
end

.dequeue ⇒ Object 



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

def self.dequeue
  queue.dequeue(queue_name)
end

.do_task(task) ⇒ Object 



70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/sweatshop/worker.rb', line 70

def self.do_task(task)
  begin
    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
  rescue SystemExit
    exit
  rescue Exception => e
    log("Caught Exception: #{e.message}, \n#{e.backtrace.join("\n")}")
    call_exception_handler(e)
  end
end

.do_tasks ⇒ Object 



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

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

.enqueue(task) ⇒ Object 



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

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

.inherited(subclass) ⇒ Object 



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

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

.instance ⇒ Object 



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

def self.instance
  @instance ||= new
end

.log(msg) ⇒ Object 



100
101
102
 
# File 'lib/sweatshop/worker.rb', line 100

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/sweatshop/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

.on_exception(&block) ⇒ Object 



135
136
137
138
139
140
141
 
# File 'lib/sweatshop/worker.rb', line 135

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

.queue ⇒ Object 



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

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

.queue_group(group = nil) ⇒ Object 



151
152
153
 
# File 'lib/sweatshop/worker.rb', line 151

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

.queue_name ⇒ Object 



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

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

.queue_size ⇒ Object 



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

def self.queue_size
  queue.queue_size(queue_name)
end

.stop ⇒ Object 



143
144
145
 
# File 'lib/sweatshop/worker.rb', line 143

def self.stop
  instance.stop
end

.workers ⇒ Object 



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

def self.workers
  Sweatshop.workers
end

Instance Method Details

#stop ⇒ Object

called before we exit – subclass can implement this method



148
 
# File 'lib/sweatshop/worker.rb', line 148

def stop; end