lib/disco/model.rb in disco-0.3.2 vs lib/disco/model.rb in disco-0.4.0

- old
+ new

@@ -1,33 +1,30 @@ module Disco module Model 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: model_name.name, 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} } + 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? - if recommendations.respond_to?(:insert_all!) - # Rails 6 - recommendations.insert_all!(items) - elsif recommendations.respond_to?(:bulk_import!) - # activerecord-import - recommendations.bulk_import!(items, validate: false) - else - recommendations.create!([items]) - end + recommendations.insert_all!(items) end end end end end