135 lines
2.9 KiB
Ruby
135 lines
2.9 KiB
Ruby
module Scheduler
|
|
class ScheduleInfo
|
|
attr_accessor :next_run,
|
|
:prev_run,
|
|
:prev_duration,
|
|
:prev_result,
|
|
:current_owner
|
|
|
|
def initialize(klass, manager)
|
|
@klass = klass
|
|
@manager = manager
|
|
|
|
data = nil
|
|
|
|
if data = $redis.get(key)
|
|
data = JSON.parse(data)
|
|
end
|
|
|
|
if data
|
|
@next_run = data["next_run"]
|
|
@prev_run = data["prev_run"]
|
|
@prev_result = data["prev_result"]
|
|
@prev_duration = data["prev_duration"]
|
|
@current_owner = data["current_owner"]
|
|
end
|
|
rescue
|
|
# corrupt redis
|
|
@next_run = @prev_run = @prev_result = @prev_duration = @current_owner = nil
|
|
end
|
|
|
|
def valid?
|
|
return false unless @next_run
|
|
(!@prev_run && @next_run < Time.now.to_i + 5.minutes) || valid_every? || valid_daily?
|
|
end
|
|
|
|
def valid_every?
|
|
return false unless @klass.every
|
|
!!@prev_run &&
|
|
@prev_run <= Time.now.to_i &&
|
|
@next_run < @prev_run + @klass.every * (1 + @manager.random_ratio)
|
|
end
|
|
|
|
def valid_daily?
|
|
return false unless @klass.daily
|
|
!!@prev_run &&
|
|
@prev_run <= Time.now.to_i &&
|
|
@next_run < @prev_run + 1.day
|
|
end
|
|
|
|
def schedule_every!
|
|
if !valid? && @prev_run
|
|
mixup = @klass.every * @manager.random_ratio
|
|
mixup = (mixup * Random.rand - mixup / 2).to_i
|
|
@next_run = @prev_run + mixup + @klass.every
|
|
end
|
|
|
|
if !valid?
|
|
@next_run = Time.now.to_i + 5.minutes * Random.rand
|
|
end
|
|
end
|
|
|
|
def schedule_daily!
|
|
return if valid?
|
|
|
|
at = @klass.daily[:at] || 0
|
|
today_begin = Time.zone.now.midnight.to_i
|
|
today_offset = DateTime.now.seconds_since_midnight
|
|
|
|
# If it's later today
|
|
if at > today_offset
|
|
@next_run = today_begin + at
|
|
else
|
|
# Otherwise do it tomorrow
|
|
@next_run = today_begin + 1.day + at
|
|
end
|
|
end
|
|
|
|
def schedule!
|
|
if @klass.every
|
|
schedule_every!
|
|
elsif @klass.daily
|
|
schedule_daily!
|
|
end
|
|
|
|
write!
|
|
end
|
|
|
|
def write!
|
|
|
|
clear!
|
|
redis.set key, {
|
|
next_run: @next_run,
|
|
prev_run: @prev_run,
|
|
prev_duration: @prev_duration,
|
|
prev_result: @prev_result,
|
|
current_owner: @current_owner
|
|
}.to_json
|
|
|
|
redis.zadd queue_key, @next_run , @klass
|
|
end
|
|
|
|
def del!
|
|
clear!
|
|
@next_run = @prev_run = @prev_result = @prev_duration = @current_owner = nil
|
|
end
|
|
|
|
def key
|
|
if @klass.is_per_host
|
|
Manager.schedule_key(@klass, @manager.hostname)
|
|
else
|
|
Manager.schedule_key(@klass)
|
|
end
|
|
end
|
|
|
|
def queue_key
|
|
if @klass.is_per_host
|
|
Manager.queue_key(@manager.hostname)
|
|
else
|
|
Manager.queue_key
|
|
end
|
|
end
|
|
|
|
def redis
|
|
@manager.redis
|
|
end
|
|
|
|
private
|
|
def clear!
|
|
redis.del key
|
|
redis.zrem queue_key, @klass
|
|
end
|
|
|
|
end
|
|
end
|