summaryrefslogtreecommitdiff
path: root/lib/rbot/timer.rb
blob: 8adb14d8d42f955480c22402ec48f478e1b4468f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
# changes:
#  1. Timer::Timer ---> Timer
#  2. timer id is now the object_id of the action
#  3. Timer resolution removed, we're always arbitrary precision now
#  4. I don't see any obvious races [not that i did see any in old impl, though]
#  5. We're tickless now, so no need to jerk start/stop
#  6. We should be pretty fast now, wrt old impl
#  7. reschedule/remove/block now accept nil as an action id (meaning "current")
#  8. repeatability is ignored for 0-period repeatable timers
#  9. configure() method superceeds reschedule() [the latter stays as compat]

require 'thread'
require 'monitor'

# Timer handler, manage multiple Action objects, calling them when required.
# When the Timer is constructed, a new Thread is created to manage timed
# delays and run Actions.
#
# XXX: there is no way to stop the timer currently. I'm keeping it this way
# to weed out old Timer implementation legacy in rbot code. -jsn.
class Timer

  # class representing individual timed action
  class Action

    # Time when the Action should be called next
    attr_accessor :next

    # Options are:
    # start::    Time when the Action should be run for the first time.
    #            Repeatable Actions will be repeated after that, see
    #            :period. One-time Actions will not (obviously)
    #            Default: Time.now + :period
    # period::   How often repeatable Action should be run, in seconds.
    #            Default: 1
    # blocked::  if true, Action starts as blocked (i.e. will stay dormant
    #            until unblocked)
    # args::     Arguments to pass to the Action callback. Default: []
    # repeat::   Should the Action be called repeatedly? Default: false
    # code::     You can specify the Action body using &block, *or* using
    #            this option.

    def initialize(options = {}, &block)
      opts = {
        :period => 1,
        :blocked => false,
        :args => [],
        :repeat => false
      }.merge(options)

      @block = nil
      debug("adding timer #{self} :period => #{opts[:period]}, :repeat => #{opts[:repeat].inspect}")
      self.configure(opts, &block)
      debug("added #{self}")
    end

    # Provides for on-the-fly reconfiguration of the Actions
    # Accept the same arguments as the constructor
    def configure(opts = {}, &block)
      @period = opts[:period] if opts.include? :period
      @blocked = opts[:blocked] if opts.include? :blocked
      @repeat = opts[:repeat] if opts.include? :repeat

      if block_given?
        @block = block 
      elsif opts[:code]
        @block = opts[:code]
      end

      raise 'huh?? blockless action?' unless @block
      if opts.include? :args
        @args = Array === opts[:args] ? opts[:args] : [opts[:args]]
      end

      if opts[:start] and (Time === opts[:start])
        self.next = opts[:start]
      else
        self.next = Time.now + (opts[:start] || @period)
      end
    end

    # modify the Action period
    def reschedule(period, &block)
      self.configure(:period => period, &block)
    end

    # blocks an Action, so it won't be run
    def block
      @blocked = true
    end

    # unblocks a blocked Action
    def unblock
      @blocked = false
    end

    def blocked?
      @blocked
    end

    # calls the Action callback, resets .next to the Time of the next call,
    # if the Action is repeatable.
    def run(now = Time.now)
      raise 'inappropriate time to run()' unless self.next && self.next <= now
      self.next = nil
      begin
        @block.call(*@args)
      rescue Exception => e
        error "Timer action #{self.inspect}: block #{@block.inspect} failed!"
        error e.pretty_inspect
        debug e.backtrace.join("\n")
      end

      if @repeat && @period > 0
        self.next = now + @period
      end

      return self.next
    end
  end

  # creates a new Timer and starts it.
  def initialize
    self.extend(MonitorMixin)
    @tick = self.new_cond
    @thread = nil
    @actions = Hash.new
    @current = nil
    self.start
  end

  # Creates and installs a new Action, repeatable by default.
  # _period_:: Action period
  # _opts_::   options for Action#new, see there
  # _block_::  Action callback code
  #
  # Returns the id of the created Action
  def add(period, opts = {}, &block)
    a = Action.new({:repeat => true, :period => period}.merge(opts), &block)
    self.synchronize do
      @actions[a.object_id] = a
      @tick.signal
    end
    return a.object_id
  end

  # Creates and installs a new Action, one-time by default.
  # _period_:: Action delay
  # _opts_::   options for Action#new, see there
  # _block_::  Action callback code
  #
  # Returns the id of the created Action
  def add_once(period, opts = {}, &block)
    self.add(period, {:repeat => false}.merge(opts), &block)
  end

  # blocks an existing Action
  # _aid_:: Action id, obtained previously from add() or add_once()
  def block(aid)
    debug "blocking #{aid}"
    self.synchronize { self[aid].block }
  end

  # unblocks an existing blocked Action
  # _aid_:: Action id, obtained previously from add() or add_once()
  def unblock(aid)
    debug "unblocking #{aid}"
    self.synchronize do
      self[aid].unblock
      @tick.signal
    end
  end

  # removes an existing blocked Action
  # _aid_:: Action id, obtained previously from add() or add_once()
  def remove(aid)
    self.synchronize do
      @actions.delete(aid) # or raise "nonexistent action #{aid}"
    end
  end

  alias :delete :remove

  # Provides for on-the-fly reconfiguration of Actions
  # _aid_::   Action id, obtained previously from add() or add_once()
  # _opts_::  see Action#new
  # _block_:: (optional) new Action callback code
  def configure(aid, opts = {}, &block)
    self.synchronize do
      self[aid].configure(opts, &block)
      @tick.signal
    end
  end

  # changes Action period
  # _aid_:: Action id
  # _period_:: new period
  # _block_:: (optional) new Action callback code
  def reschedule(aid, period, &block)
    self.configure(aid, :period => period, &block)
  end

  def start
    raise 'already started' if @thread
    @stopping = false
    debug "starting timer #{self}"
    @thread = Thread.new do
      loop do
        tmout = self.run_actions
        break if tmout and tmout < 0
        self.synchronize { @tick.wait(tmout) }
      end
    end
  end

  def stop
    raise 'already stopped' unless @thread
    debug "stopping timer #{self}..."
    @stopping = true
    self.synchronize { @tick.signal }
    @thread.join(60) or @thread.kill
    debug "timer #{self} stopped"
    @thread = nil
  end

  protected

  def [](aid)
    aid ||= @current
    raise "no current action" unless aid
    raise "nonexistent action #{aid}" unless @actions.include? aid
    @actions[aid]
  end

  def run_actions(now = Time.now)
    nxt = nil
    @actions.keys.each do |k|
      return -1 if @stopping
      a = @actions[k]
      next if (!a) or a.blocked?

      if a.next <= now
        begin
          @current = k
          v = a.run(now)
        ensure
          @current = nil
        end

        unless v
          @actions.delete k
          next
        end
      else
        v = a.next
      end

      nxt = v if v and ((!nxt) or (v < nxt))
    end

    if nxt
      delta = nxt - now
      delta = 0 if delta < 0
      return delta
    else
      return nil
    end
  end

end