提交 2c30c9fe 编写于 作者: J Joshua Peek

Undefine id and let it automatically be generated

上级 f8d2c77c
......@@ -77,6 +77,9 @@ def define_attribute_methods
end
end
end
unless generated_methods.include?("id")
define_read_method(:id, primary_key, columns_hash[primary_key.to_s])
end
end
def undefine_attribute_methods
......@@ -89,7 +92,6 @@ def undefine_attribute_methods
# method is defined by Active Record though.
def instance_method_already_implemented?(method_name)
method_name = method_name.to_s
return true if method_name == "id"
@_defined_class_methods ||= ancestors.first(ancestors.index(ActiveRecord::Base)).sum([]) { |m| m.public_instance_methods(false) | m.private_instance_methods(false) | m.protected_instance_methods(false) }.map {|m| m.to_s }.to_set
@@_defined_activerecord_methods ||= (ActiveRecord::Base.public_instance_methods(false) | ActiveRecord::Base.private_instance_methods(false) | ActiveRecord::Base.protected_instance_methods(false)).map{|m| m.to_s }.to_set
raise DangerousAttributeError, "#{method_name} is defined by ActiveRecord" if @@_defined_activerecord_methods.include?(method_name)
......@@ -109,9 +111,7 @@ def attribute_method_suffixes
# Evaluate the definition for an attribute related method
def evaluate_attribute_method(attr_name, method_definition, method_name)
unless method_name.to_s == primary_key.to_s
generated_methods << method_name
end
generated_methods << method_name
begin
class_eval(method_definition, __FILE__, __LINE__)
......
......@@ -5,6 +5,7 @@ module Read
included do
attribute_method_suffix ""
undef_method :id
end
module ClassMethods
......@@ -54,7 +55,7 @@ def define_read_method(symbol, attr_name, column)
if cache_attribute?(attr_name)
access_code = "@attributes_cache['#{attr_name}'] ||= (#{access_code})"
end
evaluate_attribute_method attr_name, "def #{symbol}; #{access_code}; end", attr_name
evaluate_attribute_method attr_name, "def #{symbol}; #{access_code}; end", symbol
end
end
......@@ -78,16 +79,6 @@ def read_attribute(attr_name)
end
end
# A model instance's primary key is always available as model.id
# whether you name it the default 'id' or set it to something else.
def id
attr_name = self.class.primary_key
column = column_for_attribute(attr_name)
self.class.send(:define_read_method, :id, attr_name, column)
id
end
# Returns true if the attribute is of a text column and marked for serialization.
def unserializable_attribute?(attr_name, column)
column.text? && self.class.serialized_attributes[attr_name]
......
......@@ -74,10 +74,6 @@ def test_kernel_methods_not_implemented_in_activerecord
end
end
def test_primary_key_implemented
assert Class.new(ActiveRecord::Base).instance_method_already_implemented?('id')
end
def test_defined_kernel_methods_implemented_in_model
%w(test name display y).each do |method|
klass = Class.new ActiveRecord::Base
......
......@@ -422,11 +422,6 @@ def test_query_attribute_with_custom_fields
end
def test_reader_for_invalid_column_names
Topic.send(:define_read_method, "mumub-jumbo".to_sym, "mumub-jumbo", nil)
assert !Topic.generated_methods.include?("mumub-jumbo")
end
def test_non_attribute_access_and_assignment
topic = Topic.new
assert !topic.respond_to?("mumbo")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册