Module: ActiveRecord::AttributeMethods::PrimaryKey::ClassMethods

Defined in:
lib/active_record/attribute_methods/primary_key.rb

Constant Summary collapse

ID_ATTRIBUTE_METHODS = 
%w(id id= id? id_before_type_cast id_was id_in_database id_for_database).to_set
PRIMARY_KEY_NOT_SET = 
BasicObject.new

Instance Method Summary collapse

Instance Method Details

#composite_primary_key? ⇒ Boolean

:nodoc:

Returns:

  • (Boolean) 


87
88
89
90
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 87

def composite_primary_key? # :nodoc:
  reset_primary_key if PRIMARY_KEY_NOT_SET.equal?(@primary_key)
  @composite_primary_key
end

#dangerous_attribute_method?(method_name) ⇒ Boolean

Returns:

  • (Boolean) 


75
76
77
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 75

def dangerous_attribute_method?(method_name)
  super && !ID_ATTRIBUTE_METHODS.include?(method_name)
end

#get_primary_key(base_name) ⇒ Object

:nodoc:



106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 106

def get_primary_key(base_name) # :nodoc:
  if base_name && primary_key_prefix_type == :table_name
    base_name.foreign_key(false)
  elsif base_name && primary_key_prefix_type == :table_name_with_underscore
    base_name.foreign_key
  elsif ActiveRecord::Base != self && table_exists?
    schema_cache.primary_keys(table_name)
  else
    "id"
  end
end

#instance_method_already_implemented?(method_name) ⇒ Boolean

Returns:

  • (Boolean) 


71
72
73
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 71

def instance_method_already_implemented?(method_name)
  super || primary_key && ID_ATTRIBUTE_METHODS.include?(method_name)
end

#primary_key ⇒ Object

Defines the primary key field – can be overridden in subclasses. Overwriting will negate any effect of the primary_key_prefix_type setting, though.



82
83
84
85
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 82

def primary_key
  reset_primary_key if PRIMARY_KEY_NOT_SET.equal?(@primary_key)
  @primary_key
end

#primary_key=(value) ⇒ Object

Sets the name of the primary key column.

class Project < ActiveRecord::Base
  self.primary_key = 'sysid'
end

You can also define the #primary_key method yourself:

class Project < ActiveRecord::Base
  def self.primary_key
    'foo_' + super
  end
end

Project.primary_key # => "foo_id"



133
134
135
136
137
138
139
140
141
142
143
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 133

def primary_key=(value)
  @primary_key = if value.is_a?(Array)
    @composite_primary_key = true
    include CompositePrimaryKey
    @primary_key = value.map { |v| -v.to_s }.freeze
  elsif value
    -value.to_s
  end
  @quoted_primary_key = nil
  @attributes_builder = nil
end

#quoted_primary_key ⇒ Object

Returns a quoted version of the primary key name, used to construct SQL statements.



94
95
96
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 94

def quoted_primary_key
  @quoted_primary_key ||= adapter_class.quote_column_name(primary_key)
end

#reset_primary_key ⇒ Object

:nodoc:



98
99
100
101
102
103
104
 
# File 'lib/active_record/attribute_methods/primary_key.rb', line 98

def reset_primary_key # :nodoc:
  if base_class?
    self.primary_key = get_primary_key(base_class.name)
  else
    self.primary_key = base_class.primary_key
  end
end