db/migrate/20190807172051_mark_known_accounts.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

class MarkKnownAccounts < ActiveRecord::Migration[5.2]
  def up
    Rails.logger.info("Marking known accounts:")
    known_accounts = local_accounts | packmates | boosted_authors | faved_authors
    Rails.logger.info("  Updating account flags...")
    Account.where(id: known_accounts).in_batches.update_all(known: true)
  end

  private

  def boosted_authors
    Rails.logger.info("  Gathering boosted accounts...")
    Status.where(id: Status.local.reblogs.reorder(nil).select(:reblog_of_id)).reorder(nil).pluck(:account_id)
  end

  def faved_authors
    Rails.logger.info("  Gathering favourited accounts...")
    Status.where(id: Favourite.select(:status_id)).reorder(nil).pluck(:account_id)
  end

  def local_accounts
    Rails.logger.info("  Gathering local accounts...")
    Account.local.pluck(:id)
  end

  def packmates
    Rails.logger.info("  Gathering packmate accounts...")
    Account.local.flat_map { |account| account.following_ids | account.follower_ids }
  end
end