app/models/follow_recommendation.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
31
32
33
34
35
36
37
38
39

# frozen_string_literal: true
# == Schema Information
#
# Table name: follow_recommendations
#
#  account_id :bigint(8)        primary key
#  rank       :decimal(, )
#  reason     :text             is an Array
#

class FollowRecommendation < ApplicationRecord
  self.primary_key = :account_id

  belongs_to :account_summary, foreign_key: :account_id
  belongs_to :account, foreign_key: :account_id

  scope :safe, -> { joins(:account_summary).merge(AccountSummary.safe) }
  scope :localized, ->(locale) { joins(:account_summary).merge(AccountSummary.localized(locale)) }
  scope :filtered, -> { joins(:account_summary).merge(AccountSummary.filtered) }

  def readonly?
    true
  end

  def self.get(account, limit, exclude_account_ids = [])
    account_ids = Redis.current.zrevrange("follow_recommendations:#{account.user_locale}", 0, -1).map(&:to_i) - exclude_account_ids - [account.id]

    return [] if account_ids.empty? || limit < 1

    accounts = Account.followable_by(account)
                      .not_excluded_by_account(account)
                      .not_domain_blocked_by_account(account)
                      .where(id: account_ids)
                      .limit(limit)
                      .index_by(&:id)

    account_ids.map { |id| accounts[id] }.compact
  end
end