mirror of
https://github.com/codeninjasllc/discourse.git
synced 2024-12-23 05:52:37 -05:00
d6d9a7fa09
These are jobs that will run on every host that is running discourse. If you have multiple hosts running the same site you get independent schedules
134 lines
2.9 KiB
Ruby
134 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.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
|