Module: Disco::Model

Defined in:
lib/disco/model.rb

Instance Method Summary collapse

Instance Method Details



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
# File 'lib/disco/model.rb', line 3

def has_recommended(name, class_name: nil)
  if ActiveRecord::VERSION::MAJOR < 6
    raise Disco::Error, "Requires Active Record 6+"
  end

  class_name ||= name.to_s.singularize.camelize
  subject_type = model_name.name

  class_eval do
    unless reflect_on_association(:recommendations)
      has_many :recommendations, class_name: "Disco::Recommendation", as: :subject, dependent: :destroy
    end

    has_many :"recommended_#{name}", -> { where("disco_recommendations.context = ?", name).order("disco_recommendations.score DESC") }, through: :recommendations, source: :item, source_type: class_name

    define_method("update_recommended_#{name}") do |items|
      now = Time.now
      items = items.map { |item| {subject_type: subject_type, subject_id: id, item_type: class_name, item_id: item.fetch(:item_id), context: name, score: item.fetch(:score), created_at: now, updated_at: now} }

      self.class.transaction do
        recommendations.where(context: name).delete_all

        if items.any?
          recommendations.insert_all!(items)
        end
      end
    end
  end
end