blob: f865f0b2c8dcac7dab04042338e2a70fb18f9b4f (
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
|
module ActiveRecord
module AttributeMethods
module Write
WriterMethodCache = Class.new(AttributeMethodCache) {
private
def method_body(method_name, const_name)
<<-EOMETHOD
def #{method_name}(value)
name = ::ActiveRecord::AttributeMethods::AttrNames::ATTR_#{const_name}
write_attribute(name, value)
end
EOMETHOD
end
}.new
extend ActiveSupport::Concern
included do
attribute_method_suffix "="
end
module ClassMethods
protected
# See define_method_attribute in read.rb for an explanation of
# this code.
def define_method_attribute=(name)
method = WriterMethodCache[name]
generated_attribute_methods.module_eval {
define_method "#{name}=", method
}
end
end
# Updates the attribute identified by <tt>attr_name</tt> with the
# specified +value+. Empty strings for fixnum and float columns are
# turned into +nil+.
def write_attribute(attr_name, value)
attr_name = attr_name.to_s
attr_name = self.class.primary_key if attr_name == 'id' && self.class.primary_key
@attributes_cache.delete(attr_name)
column = column_for_attribute(attr_name)
# If we're dealing with a binary column, write the data to the cache
# so we don't attempt to typecast multiple times.
if column && column.binary?
@attributes_cache[attr_name] = value
end
if column || @attributes.has_key?(attr_name)
@attributes[attr_name] = type_cast_attribute_for_write(column, value)
else
raise ActiveModel::MissingAttributeError, "can't write unknown attribute `#{attr_name}'"
end
end
alias_method :raw_write_attribute, :write_attribute
private
# Handle *= for method_missing.
def attribute=(attribute_name, value)
write_attribute(attribute_name, value)
end
def type_cast_attribute_for_write(column, value)
return value unless column
column.type_cast_for_write value
end
end
end
end
|
