blob: 50fcb8e07e58db8b25e14c2a4213df1d8ea76fa1 (
plain) (
blame)
1
2
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
module Arel
class Deletion < Compound
def to_sql
build_query \
"DELETE",
"FROM #{table_sql}",
("WHERE #{wheres.collect(&:to_sql).join('\n\tAND ')}" unless wheres.blank? ),
("LIMIT #{taken}" unless taken.blank? )
end
end
class Insert < Compound
def to_sql
insertion_attributes_values_sql = if record.is_a?(Value)
record.value
else
attributes = record.keys.sort_by do |attribute|
attribute.name.to_s
end
first = attributes.collect do |key|
engine.quote_column_name(key.name)
end.join(', ')
second = attributes.collect do |key|
key.format(record[key])
end.join(', ')
build_query "(#{first})", "VALUES (#{second})"
end
build_query \
"INSERT",
"INTO #{table_sql}",
insertion_attributes_values_sql
end
end
class Update < Compound
def to_sql
build_query \
"UPDATE #{table_sql} SET",
assignment_sql,
build_update_conditions_sql
end
protected
def assignment_sql
if assignments.respond_to?(:collect)
attributes = assignments.keys.sort_by do |attribute|
attribute.name.to_s
end
attributes.map do |attribute|
value = assignments[attribute]
"#{engine.quote_column_name(attribute.name)} = #{attribute.format(value)}"
end.join(",\n")
else
assignments.value
end
end
def build_update_conditions_sql
conditions = ""
conditions << " WHERE #{wheres.collect(&:to_sql).join('\n\tAND ')}" unless wheres.blank?
conditions << " ORDER BY #{order_clauses.join(', ')}" unless orders.blank?
unless taken.blank?
conditions << " LIMIT #{taken}"
if engine.adapter_name != "MySQL"
begin
quote_primary_key = engine.quote_column_name(table.name.classify.constantize.primary_key)
rescue NameError
quote_primary_key = engine.quote_column_name("id")
end
conditions = "WHERE #{quote_primary_key} IN (SELECT #{quote_primary_key} FROM #{engine.connection.quote_table_name table.name} #{conditions})"
end
end
conditions
end
end
end
|