Skip to content
This repository has been archived by the owner on May 16, 2019. It is now read-only.

Commit

Permalink
Remove progress estimate from MigrateAccountConversations (mastodon#9168
Browse files Browse the repository at this point in the history
)
  • Loading branch information
Gargron authored Oct 30, 2018
1 parent 7800e1a commit f2290e3
Showing 1 changed file with 6 additions and 24 deletions.
30 changes: 6 additions & 24 deletions db/migrate/20181024224956_migrate_account_conversations.rb
Original file line number Diff line number Diff line change
Expand Up @@ -14,17 +14,15 @@ def up
sleep 1
end

total = estimate_rows(local_direct_statuses) + estimate_rows(notifications_about_direct_statuses)
migrated = 0
started_time = Time.zone.now
last_time = Time.zone.now
migrated = 0
last_time = Time.zone.now

local_direct_statuses.includes(:account, mentions: :account).find_each do |status|
AccountConversation.add_status(status.account, status)
migrated += 1

if Time.zone.now - last_time > 1
say_progress(migrated, total, started_time)
say_progress(migrated)
last_time = Time.zone.now
end
end
Expand All @@ -34,7 +32,7 @@ def up
migrated += 1

if Time.zone.now - last_time > 1
say_progress(migrated, total, started_time)
say_progress(migrated)
last_time = Time.zone.now
end
end
Expand All @@ -45,24 +43,8 @@ def down

private

def estimate_rows(query)
result = exec_query("EXPLAIN #{query.to_sql}").first
result['QUERY PLAN'].scan(/ rows=([\d]+)/).first&.first&.to_i || 0
end

def say_progress(migrated, total, started_time)
status = "Migrated #{migrated} rows"

percentage = 100.0 * migrated / total
status += " (~#{sprintf('%.2f', percentage)}%, "

remaining_time = (100.0 - percentage) * (Time.zone.now - started_time) / percentage

status += "#{(remaining_time / 60).to_i}:"
status += sprintf('%02d', remaining_time.to_i % 60)
status += ' remaining)'

say status, true
def say_progress(migrated)
say "Migrated #{migrated} rows", true
end

def local_direct_statuses
Expand Down

0 comments on commit f2290e3

Please sign in to comment.