Sha256: 93f0193f4f2bce1604df63433e680cb81f6ad68146f790c3ee2cf6fbc00502e4
Contents?: true
Size: 1.19 KB
Versions: 29
Compression:
Stored size: 1.19 KB
Contents
module CompositePrimaryKeys
module CollectionAssociation
def ids_writer(ids)
primary_key = reflection.association_primary_key
pk_type = klass.type_for_attribute(primary_key)
ids = Array(ids).reject(&:blank?)
ids.map! { |i| pk_type.cast(i) }
# CPK-
if 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
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
29 entries across 29 versions & 1 rubygems