Module: ActiveRecord::Persistence

Defined in:
lib/composite_primary_keys/persistence.rb

Instance Method Summary collapse

Instance Method Details

#destroy ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/composite_primary_keys/persistence.rb', line 3

def destroy
  destroy_associations

  if persisted?
    IdentityMap.remove(self) if IdentityMap.enabled?
    # CPK
    #pk         = self.class.primary_key
    #column     = self.class.columns_hash[pk]
    #substitute = connection.substitute_at(column, 0)

    primary_keys = Array(self.class.primary_key)
    bind_values = Array.new
    eq_predicates = Array.new
    primary_keys.each_with_index do |key, i|
      column = self.class.columns_hash[key.to_s]
      bind_values << [column, self[key]]
      substitute = connection.substitute_at(column, i)
      eq_predicates << self.class.arel_table[key].eq(substitute)
    end
    predicate = Arel::Nodes::And.new(eq_predicates)
    relation = self.class.unscoped.where(predicate)

    #relation = self.class.unscoped.where(
    #  self.class.arel_table[pk].eq(substitute))

    # CPK
    #relation.bind_values = [[column, id]]
    relation.bind_values = bind_values
    relation.delete_all
  end

  @destroyed = true
  freeze
end

#update(attribute_names = @attributes.keys) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/composite_primary_keys/persistence.rb', line 38

def update(attribute_names = @attributes.keys)
  klass = self.class
  if !self.composite?
    attributes_with_values = arel_attributes_values(false, false, attribute_names)
    return 0 if attributes_with_values.empty?
    stmt = klass.unscoped.where(klass.arel_table[klass.primary_key].eq(id)).arel.compile_update(attributes_with_values)
  else
    attributes_with_values = arel_attributes_values(can_change_primary_key?, false, attribute_names)
    return 0 if attributes_with_values.empty?

    if !can_change_primary_key? and primary_key_changed?
      raise ActiveRecord::CompositeKeyError, "Cannot update primary key values without ActiveModel::Dirty"
    elsif primary_key_changed?
      stmt = klass.unscoped.where(primary_key_was).arel.compile_update(attributes_with_values)
    else
      stmt = klass.unscoped.where(ids_hash).arel.compile_update(attributes_with_values)
    end
  end
  klass.connection.update stmt.to_sql
end