module Rein::Constraint::Inclusion

This module contains methods for defining inclusion constraints.

Public Instance Methods

add_inclusion_constraint(*args) click to toggle source
# File lib/rein/constraint/inclusion.rb, line 9
def add_inclusion_constraint(*args)
  reversible do |dir|
    dir.up { _add_inclusion_constraint(*args) }
    dir.down { _remove_inclusion_constraint(*args) }
  end
end
remove_inclusion_constraint(*args) click to toggle source
# File lib/rein/constraint/inclusion.rb, line 16
def remove_inclusion_constraint(*args)
  reversible do |dir|
    dir.up { _remove_inclusion_constraint(*args) }
    dir.down { _add_inclusion_constraint(*args) }
  end
end

Private Instance Methods

_add_inclusion_constraint(table, attribute, options = {}) click to toggle source
# File lib/rein/constraint/inclusion.rb, line 25
def _add_inclusion_constraint(table, attribute, options = {})
  name = Util.constraint_name(table, attribute, 'inclusion', options)
  table = Util.wrap_identifier(table)
  values = options[:in].map { |value| quote(value) }.join(', ')
  attribute = Util.wrap_identifier(attribute)
  conditions = Util.conditions_with_if("#{attribute} IN (#{values})", options)
  sql = "ALTER TABLE #{table} ADD CONSTRAINT #{name} CHECK (#{conditions})"
  execute(Util.add_not_valid_suffix_if_required(sql, options))
end
_remove_inclusion_constraint(table, attribute, options = {}) click to toggle source
# File lib/rein/constraint/inclusion.rb, line 35
def _remove_inclusion_constraint(table, attribute, options = {})
  name = Util.constraint_name(table, attribute, 'inclusion', options)
  table = Util.wrap_identifier(table)
  execute("ALTER TABLE #{table} DROP CONSTRAINT #{name}")
end