treehouse
/
mastodon
22
8
Fork 5
Code Issues 32 Pull Requests Projects 1 Releases Wiki Activity

Fix race condition when processing incoming OStatus messages (#5013) 

* Avoid races in incoming OStatus toots processing

* oops

* oops again
remotes/1727458204337373841/tmp_refs/heads/signup-info-prompt
ThibG 2017-09-19 21:44:18 +02:00 committed by Eugen Rochko
parent bb4d005a83
commit 34fa305a00
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app/lib/ostatus/activity/creation.rb
View File

@ -14,14 +14,22 @@ class OStatus::Activity::Creation < OStatus::Activity::Base
return result if result.first.present?
end
RedisLock.acquire(lock_options) do |lock|
if lock.acquired?
# Return early if status already exists in db
@status = find_status(id)
return [@status, false] unless @status.nil?
@status = process_status
end
end
[@status, true]
end
def process_status
Rails.logger.debug "Creating remote status #{id}"
# Return early if status already exists in db
status = find_status(id)
return [status, false] unless status.nil?
cached_reblog = reblog
status = nil
ApplicationRecord.transaction do
status = Status.create!(
@ -55,7 +63,7 @@ class OStatus::Activity::Creation < OStatus::Activity::Base
LinkCrawlWorker.perform_async(status.id) unless status.spoiler_text?
DistributionWorker.perform_async(status.id)
[status, true]
status
end
def perform_via_activitypub
@ -179,4 +187,8 @@ class OStatus::Activity::Creation < OStatus::Activity::Base
Account.where(uri: href).or(Account.where(url: href)).first || FetchRemoteAccountService.new.call(href)
end
end
def lock_options
{ redis: Redis.current, key: "create:#{id}" }
end
end