Skip to content

Commit

Permalink
Migrate all old direct messages to new conversations schema (#9085)
Browse files Browse the repository at this point in the history
  • Loading branch information
Gargron authored Oct 24, 2018
1 parent 9b53482 commit 4ea718e
Show file tree
Hide file tree
Showing 3 changed files with 47 additions and 1 deletion.
3 changes: 3 additions & 0 deletions app/models/account_conversation.rb
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,9 @@ def paginate_by_max_id(limit, max_id = nil, since_id = nil)

def add_status(recipient, status)
conversation = find_or_initialize_by(account: recipient, conversation_id: status.conversation_id, participant_account_ids: participants_from_status(recipient, status))

return conversation if conversation.status_ids.include?(status.id)

conversation.status_ids << status.id
conversation.unread = status.account_id != recipient.id
conversation.save
Expand Down
43 changes: 43 additions & 0 deletions db/migrate/20181024224956_migrate_account_conversations.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
class MigrateAccountConversations < ActiveRecord::Migration[5.2]
disable_ddl_transaction!

def up
say ''
say 'WARNING: This migration may take a *long* time for large instances'
say 'It will *not* lock tables for any significant time, but it may run'
say 'for a very long time. We will pause for 10 seconds to allow you to'
say 'interrupt this migration if you are not ready.'
say ''

10.downto(1) do |i|
say "Continuing in #{i} second#{i == 1 ? '' : 's'}...", true
sleep 1
end

local_direct_statuses.find_each do |status|
AccountConversation.add_status(status.account, status)
end

notifications_about_direct_statuses.find_each do |notification|
AccountConversation.add_status(notification.account, notification.target_status)
end
end

def down
end

private

def local_direct_statuses
Status.unscoped
.local
.where(visibility: :direct)
.includes(:account, mentions: :account)
end

def notifications_about_direct_statuses
Notification.joins(mention: :status)
.where(activity_type: 'Mention', statuses: { visibility: :direct })
.includes(:account, mention: { status: [:account, mentions: :account] })
end
end
2 changes: 1 addition & 1 deletion db/schema.rb
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
#
# It's strongly recommended that you check this file into your version control system.

ActiveRecord::Schema.define(version: 2018_10_18_205649) do
ActiveRecord::Schema.define(version: 2018_10_24_224956) do

# These are extensions that must be enabled in order to support this database
enable_extension "plpgsql"
Expand Down

0 comments on commit 4ea718e

Please sign in to comment.