Some worker updates.

This commit is contained in:
Fosco Marotto 2021-02-05 13:25:46 -05:00
parent c45d8adbf2
commit a43a7af771
4 changed files with 7 additions and 57 deletions

View File

@ -7,21 +7,12 @@ class RemoveStatusService < BaseService
@payload = Oj.dump(event: :delete, payload: status.id.to_s) @payload = Oj.dump(event: :delete, payload: status.id.to_s)
@status = status @status = status
@account = status.account @account = status.account
@tags = status.tags.pluck(:name).to_a
@mentions = status.active_mentions.includes(:account).to_a
@reblogs = status.reblogs.includes(:account).to_a @reblogs = status.reblogs.includes(:account).to_a
@options = options @options = options
RedisLock.acquire(lock_options) do |lock| RedisLock.acquire(lock_options) do |lock|
if lock.acquired? if lock.acquired?
remove_from_self if status.account.local?
remove_from_followers
remove_from_lists
remove_from_affected
remove_reblogs remove_reblogs
remove_from_hashtags
remove_from_pro
@status.destroy! @status.destroy!
else else
raise GabSocial::RaceConditionError raise GabSocial::RaceConditionError
@ -38,30 +29,6 @@ class RemoveStatusService < BaseService
private private
def remove_from_self
FeedManager.instance.unpush_from_home(@account, @status)
end
def remove_from_followers
@account.followers_for_local_distribution.reorder(nil).find_each do |follower|
FeedManager.instance.unpush_from_home(follower, @status)
end
end
def remove_from_lists
@account.lists_for_local_distribution.select(:id, :account_id).reorder(nil).find_each do |list|
FeedManager.instance.unpush_from_list(list, @status)
end
end
def remove_from_affected
redis.with do |conn|
@mentions.map(&:account).select(&:local?).each do |account|
conn.publish("timeline:#{account.id}", @payload)
end
end
end
def remove_reblogs def remove_reblogs
# We delete reblogs of the status before the original status, # We delete reblogs of the status before the original status,
# because once original status is gone, reblogs will disappear # because once original status is gone, reblogs will disappear
@ -72,25 +39,6 @@ class RemoveStatusService < BaseService
end end
end end
def remove_from_hashtags
return unless @status.public_visibility?
redis.with do |conn|
@tags.each do |hashtag|
conn.publish("timeline:hashtag:#{hashtag}", @payload)
conn.publish("timeline:hashtag:#{hashtag}:local", @payload) if @status.local?
end
end
end
def remove_from_pro
redis.with do |conn|
if @account.is_pro || @account.is_donor || @account.is_investor || @account.is_verified
conn.publish('timeline:pro', @payload)
end
end
end
def lock_options def lock_options
{ redis: Redis.current, key: "distribute:#{@status.id}" } { redis: Redis.current, key: "distribute:#{@status.id}" }
end end

View File

@ -3,6 +3,8 @@
class LocalNotificationWorker class LocalNotificationWorker
include Sidekiq::Worker include Sidekiq::Worker
sidekiq_options queue: 'default', retry: 3
def perform(receiver_account_id, activity_id = nil, activity_class_name = nil) def perform(receiver_account_id, activity_id = nil, activity_class_name = nil)
return true if activity_id.nil? or activity_class_name.nil? return true if activity_id.nil? or activity_class_name.nil?

View File

@ -4,9 +4,9 @@ class MuteWorker
include Sidekiq::Worker include Sidekiq::Worker
def perform(account_id, target_account_id) def perform(account_id, target_account_id)
FeedManager.instance.clear_from_timeline( #FeedManager.instance.clear_from_timeline(
Account.find(account_id), # Account.find(account_id),
Account.find(target_account_id) # Account.find(target_account_id)
) #)
end end
end end

View File

@ -6,6 +6,6 @@ class UnmergeWorker
sidekiq_options queue: 'pull' sidekiq_options queue: 'pull'
def perform(from_account_id, into_account_id) def perform(from_account_id, into_account_id)
FeedManager.instance.unmerge_from_timeline(Account.find(from_account_id), Account.find(into_account_id)) # FeedManager.instance.unmerge_from_timeline(Account.find(from_account_id), Account.find(into_account_id))
end end
end end