Contents

module CompositePrimaryKeys
  module CollectionAssociation
    def ids_writer(ids)
      pk_type = reflection.association_primary_key_type
      ids = Array(ids).reject(&:blank?)
      ids.map! { |i| pk_type.cast(i) }

      # CPK
      if reflection.association_primary_key.is_a?(Array)
        predicate = CompositePrimaryKeys::Predicates.cpk_in_predicate(klass.arel_table, reflection.association_primary_key, ids)
        records = klass.where(predicate).index_by do |r|
          reflection.association_primary_key.map{ |k| r.send(k) }
        end.values_at(*ids)
      else
        primary_key = reflection.association_primary_key
        records = klass.where(primary_key => ids).index_by do |r|
          r.public_send(primary_key)
        end.values_at(*ids).compact
      end

      if records.size != ids.size
        found_ids = records.map { |record| record.public_send(primary_key) }
        not_found_ids = ids - found_ids
        klass.all.raise_record_not_found_exception!(ids, records.size, ids.size, primary_key, not_found_ids)
      else
        replace(records)
      end
    end
  end
end

ActiveRecord::Associations::CollectionAssociation.prepend CompositePrimaryKeys::CollectionAssociation

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
composite_primary_keys-11.0.0.beta4	lib/composite_primary_keys/associations/collection_association.rb
composite_primary_keys-11.0.0.beta3	lib/composite_primary_keys/associations/collection_association.rb
composite_primary_keys-11.0.0.beta2	lib/composite_primary_keys/associations/collection_association.rb