Contents

module CompositePrimaryKeys
  module CollectionAssociation
    def get_records
      cpk_applies = target.try(:composite?) || owner.try(:composite?)
      return scope.to_a if cpk_applies
      super
    end

    def ids_reader
      if loaded?
        load_target.map do |record|
          if reflection.association_primary_key.is_a?(Array)
            reflection.association_primary_key.map { |key| record.send(key) }
          else
            record.send(reflection.association_primary_key)
          end
        end
      else
        @association_ids ||= (
        column = "#{reflection.quoted_table_name}.#{reflection.association_primary_key}"
        scope.pluck(column)
        )
      end
    end

    def ids_writer(ids)
      pk_type = reflection.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 = Class.new.extend(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(reflection.association_primary_key => ids).index_by do |r|
          r.send(reflection.association_primary_key)
        end.values_at(*ids)
      end
      replace(records)
    end
  end
end

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

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
composite_primary_keys-9.0.10	lib/composite_primary_keys/associations/collection_association.rb
composite_primary_keys-9.0.9	lib/composite_primary_keys/associations/collection_association.rb
composite_primary_keys-9.0.8	lib/composite_primary_keys/associations/collection_association.rb
composite_primary_keys-9.0.7	lib/composite_primary_keys/associations/collection_association.rb