mirror of
https://github.com/lunaisnotaboy/mastodon.git
synced 2024-09-10 12:18:10 +00:00
52 lines
1.2 KiB
Ruby
52 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class PollExpirationNotifyWorker
|
|
include Sidekiq::Worker
|
|
|
|
sidekiq_options lock: :until_executing
|
|
|
|
def perform(poll_id)
|
|
@poll = Poll.find(poll_id)
|
|
|
|
return if does_not_expire?
|
|
requeue! && return if not_due_yet?
|
|
|
|
notify_remote_voters_and_owner! if @poll.local?
|
|
notify_local_voters!
|
|
rescue ActiveRecord::RecordNotFound
|
|
true
|
|
end
|
|
|
|
def self.remove_from_scheduled(poll_id)
|
|
queue = Sidekiq::ScheduledSet.new
|
|
queue.select { |scheduled| scheduled.klass == name && scheduled.args[0] == poll_id }.map(&:delete)
|
|
end
|
|
|
|
private
|
|
|
|
def does_not_expire?
|
|
@poll.expires_at.nil?
|
|
end
|
|
|
|
def not_due_yet?
|
|
@poll.expires_at.present? && !@poll.expired?
|
|
end
|
|
|
|
def requeue!
|
|
PollExpirationNotifyWorker.perform_at(@poll.expires_at + 5.minutes, @poll.id)
|
|
end
|
|
|
|
def notify_remote_voters_and_owner!
|
|
ActivityPub::DistributePollUpdateWorker.perform_async(@poll.status.id)
|
|
LocalNotificationWorker.perform_async(@poll.account_id, @poll.id, 'Poll', 'poll')
|
|
end
|
|
|
|
def notify_local_voters!
|
|
@poll.voters.merge(Account.local).select(:id).find_in_batches do |accounts|
|
|
LocalNotificationWorker.push_bulk(accounts) do |account|
|
|
[account.id, @poll.id, 'Poll', 'poll']
|
|
end
|
|
end
|
|
end
|
|
end
|