Class: DataMapper::Adapters::Sql::Commands::SaveCommand

Inherits:
Object
  • Object
show all
Defined in:
lib/data_mapper/adapters/sql/commands/save_command.rb

Instance Method Summary collapse

Constructor Details

#initialize(adapter, session, instance) ⇒ SaveCommand

Returns a new instance of SaveCommand.



8
9
10
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 8

def initialize(adapter, session, instance)
  @adapter, @session, @instance = adapter, session, instance
end

Instance Method Details

#call ⇒ Object 



99
100
101
102
103
104
105
106
107
108
109
110
111
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 99

def call
  if @instance.kind_of?(Class)
    return false if @adapter.table_exists?(@instance)
    execute_create_table(to_create_table_sql)
  else
    return false unless @instance.dirty?
    callback(:before_save)
    result = @instance.new_record? ? insert! : update!
    @instance.session = @session
    callback(:after_save)
    result
  end
end

#callback(name) ⇒ Object 



63
64
65
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 63

def callback(name)
  @instance.class.callbacks.execute(name, @instance)
end

#column_long_form(column) ⇒ Object 



52
53
54
55
56
57
58
59
60
61
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 52

def column_long_form(column)
  long_form = "#{column.to_sql} #{@adapter.class::TYPES[column.type] || column.type}"
        
  long_form << "(#{column.size})" unless column.size.nil?
  long_form << " NOT NULL" unless column.nullable?
  long_form << " auto_increment" if column.key?
  long_form << " default #{column.options[:default]}" if column.options.has_key?(:default)
        
  return long_form
end

#insert! ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 67

def insert!
  callback(:before_create)
  
  result = execute_insert(to_insert_sql)
  
  if result
    @instance.instance_variable_set(:@new_record, false)
    @instance.instance_variable_set(:@id, result)
    calculate_original_hashes(@instance)
    @session.identity_map.set(@instance)
    callback(:after_create)
  end

  return result
rescue => error
  @adapter.log.error(error)
  raise error
end

#to_create_table_sql ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 38

def to_create_table_sql
  table = @adapter[@instance]
        
  sql = "CREATE TABLE " << table.to_sql << " ("
        
  sql << table.columns.map do |column|
    column_long_form(column)
  end.join(', ')
        
  sql << ", PRIMARY KEY (#{table.key.to_sql}))"
        
  return sql
end

#to_insert_sql ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 24

def to_insert_sql
        
  table = @adapter[@instance.class]
        
  keys = []
  values = []
        
  @instance.dirty_attributes.each_pair { |k,v| keys << table[k].to_sql; values << v }
        
  # Formatting is a bit off here, but it looks nicer in the log this way.
  sql = "INSERT INTO #{table.to_sql} (#{keys.join(', ')}) \
    VALUES (#{values.map { |v| @adapter.quote_value(v) }.join(', ')})"
end

#to_update_sql ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 12

def to_update_sql
  table = @adapter[@instance.class]
        
  sql = "UPDATE " << table.to_sql << " SET "
        
  @instance.dirty_attributes.map do |k, v|
    sql << table[k].to_sql << " = " << @adapter.quote_value(v) << ", "
  end
        
  sql[0, sql.size - 2] << " WHERE #{table.key.to_sql} = " << @adapter.quote_value(@instance.key)
end

#update! ⇒ Object 



86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/data_mapper/adapters/sql/commands/save_command.rb', line 86

def update!
  callback(:before_update)
  
  result = execute_update(to_update_sql)
  
  calculate_original_hashes(@instance)
  callback(:after_update)
  return result
rescue => error
  @adapter.log.error(error)
  raise error
end