Use separate workers to process imports, retry failures (#5207)

lolsob-rspec
Eugen Rochko 2017-10-04 00:39:32 +02:00 committed by GitHub
parent 7983da13cd
commit 1bf2afddcf
2 changed files with 38 additions and 45 deletions

View File

@ -0,0 +1,25 @@
# frozen_string_literal: true
class Import::RelationshipWorker
include Sidekiq::Worker
sidekiq_options queue: 'pull', retry: 8, dead: false
def perform(account_id, target_account_uri, relationship)
from_account = Account.find(account_id)
target_account = ResolveRemoteAccountService.new.call(target_account_uri)
return if target_account.nil?
case relationship
when 'follow'
FollowService.new.call(from_account, target_account.acct)
when 'block'
BlockService.new.call(from_account, target_account)
when 'mute'
MuteService.new.call(from_account, target_account)
end
rescue ActiveRecord::RecordNotFound
true
end
end

View File

@ -12,13 +12,8 @@ class ImportWorker
def perform(import_id) def perform(import_id)
@import = Import.find(import_id) @import = Import.find(import_id)
case @import.type Import::RelationshipWorker.push_bulk(import_rows) do |row|
when 'blocking' [@import.account_id, row.first, relationship_type]
process_blocks
when 'following'
process_follows
when 'muting'
process_mutes
end end
@import.destroy @import.destroy
@ -26,49 +21,22 @@ class ImportWorker
private private
def from_account
@import.account
end
def import_contents def import_contents
Paperclip.io_adapters.for(@import.data).read Paperclip.io_adapters.for(@import.data).read
end end
def relationship_type
case @import.type
when 'following'
'follow'
when 'blocking'
'block'
when 'muting'
'mute'
end
end
def import_rows def import_rows
CSV.new(import_contents).reject(&:blank?) CSV.new(import_contents).reject(&:blank?)
end end
def process_mutes
import_rows.each do |row|
begin
target_account = ResolveRemoteAccountService.new.call(row.first)
next if target_account.nil?
MuteService.new.call(from_account, target_account)
rescue Mastodon::UnexpectedResponseError, HTTP::Error, OpenSSL::SSL::SSLError
next
end
end
end
def process_blocks
import_rows.each do |row|
begin
target_account = ResolveRemoteAccountService.new.call(row.first)
next if target_account.nil?
BlockService.new.call(from_account, target_account)
rescue Mastodon::UnexpectedResponseError, HTTP::Error, OpenSSL::SSL::SSLError
next
end
end
end
def process_follows
import_rows.each do |row|
begin
FollowService.new.call(from_account, row.first)
rescue Mastodon::NotPermittedError, ActiveRecord::RecordNotFound, Mastodon::UnexpectedResponseError, HTTP::Error, OpenSSL::SSL::SSLError
next
end
end
end
end end