module Sequel::Plugins::PgArrayAssociations::DatasetMethods

Private Instance Methods

many_to_pg_array_association_filter_expression(op, ref, obj) click to toggle source

Support filtering by many_to_pg_array associations using a subquery.

# File lib/sequel/plugins/pg_array_associations.rb, line 504
def many_to_pg_array_association_filter_expression(op, ref, obj)
  pk = ref.qualify(model.table_name, ref.primary_key)
  key = ref[:key]
  expr = case obj
  when Sequel::Model
    if (assoc_pks = obj.get_column_value(key)) && !assoc_pks.empty?
      Sequel[pk=>assoc_pks.to_a]
    end
  when Array
    if (assoc_pks = obj.map{|o| o.get_column_value(key)}.flatten.compact.uniq) && !assoc_pks.empty?
      Sequel[pk=>assoc_pks]
    end
  when Sequel::Dataset
    Sequel[pk=>obj.select{Sequel.pg_array_op(ref.qualify(obj.model.table_name, ref[:key_column])).unnest}]
  end
  expr = Sequel::SQL::Constants::FALSE unless expr
  expr = add_association_filter_conditions(ref, obj, expr)
  association_filter_handle_inversion(op, expr, [pk])
end
pg_array_to_many_association_filter_expression(op, ref, obj) click to toggle source

Support filtering by pg_array_to_many associations using a subquery.

# File lib/sequel/plugins/pg_array_associations.rb, line 525
def pg_array_to_many_association_filter_expression(op, ref, obj)
  key = ref.qualify(model.table_name, ref[:key_column])
  expr = case obj
  when Sequel::Model
    if pkv = obj.get_column_value(ref.primary_key_method)
      Sequel.pg_array_op(key).contains(Sequel.pg_array([pkv], ref.array_type))
    end
  when Array
    if (pkvs = obj.map{|o| o.get_column_value(ref.primary_key_method)}.compact) && !pkvs.empty?
      Sequel.pg_array(key).overlaps(Sequel.pg_array(pkvs, ref.array_type))
    end
  when Sequel::Dataset
    Sequel.function(:coalesce, Sequel.pg_array_op(key).overlaps(obj.select{array_agg(ref.qualify(obj.model.table_name, ref.primary_key))}), Sequel::SQL::Constants::FALSE)
  end
  expr = Sequel::SQL::Constants::FALSE unless expr
  expr = add_association_filter_conditions(ref, obj, expr)
  association_filter_handle_inversion(op, expr, [key])
end