attributes.rb 4.1 KB
Newer Older
1
module ActiveRecord
S
Sean Griffin 已提交
2
  module Attributes # :nodoc:
3 4
    extend ActiveSupport::Concern

5
    Type = ActiveRecord::Type
6

7
    included do
S
Sean Griffin 已提交
8
      class_attribute :user_provided_columns, instance_accessor: false # :internal:
9 10 11
      self.user_provided_columns = {}
    end

12
    module ClassMethods
S
Sean Griffin 已提交
13
      # Defines or overrides a attribute on this model. This allows customization of
14 15 16
      # Active Record's type casting behavior, as well as adding support for user defined
      # types.
      #
17 18 19 20 21 22 23 24 25 26 27
      # +name+ The name of the methods to define attribute methods for, and the column which
      # this will persist to.
      #
      # +cast_type+ A type object that contains information about how to type cast the value.
      # See the examples section for more information.
      #
      # ==== Options
      # The options hash accepts the following options:
      #
      # +default+ is the default value that the column should use on a new record.
      #
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
      # ==== Examples
      #
      # The type detected by Active Record can be overriden.
      #
      #   # db/schema.rb
      #   create_table :store_listings, force: true do |t|
      #     t.decimal :price_in_cents
      #   end
      #
      #   # app/models/store_listing.rb
      #   class StoreListing < ActiveRecord::Base
      #   end
      #
      #   store_listing = StoreListing.new(price_in_cents: '10.1')
      #
      #   # before
      #   store_listing.price_in_cents # => BigDecimal.new(10.1)
      #
      #   class StoreListing < ActiveRecord::Base
S
Sean Griffin 已提交
47
      #     attribute :price_in_cents, Type::Integer.new
48 49 50 51 52 53 54 55
      #   end
      #
      #   # after
      #   store_listing.price_in_cents # => 10
      #
      # Users may also define their own custom types, as long as they respond to the methods
      # defined on the value type. The `type_cast` method on your type object will be called
      # with values both from the database, and from your controllers. See
S
Sean Griffin 已提交
56
      # `ActiveRecord::Attributes::Type::Value` for the expected API. It is recommended that your
57 58 59 60 61 62 63 64 65 66 67 68 69 70
      # type objects inherit from an existing type, or the base value type.
      #
      #   class MoneyType < ActiveRecord::Type::Integer
      #     def type_cast(value)
      #       if value.include?('$')
      #         price_in_dollars = value.gsub(/\$/, '').to_f
      #         price_in_dollars * 100
      #       else
      #         value.to_i
      #       end
      #     end
      #   end
      #
      #   class StoreListing < ActiveRecord::Base
S
Sean Griffin 已提交
71
      #     attribute :price_in_cents, MoneyType.new
72 73 74 75
      #   end
      #
      #   store_listing = StoreListing.new(price_in_cents: '$10.00')
      #   store_listing.price_in_cents # => 1000
S
Sean Griffin 已提交
76
      def attribute(name, cast_type, options = {})
77
        name = name.to_s
78
        clear_caches_calculated_from_columns
79
        # Assign a new hash to ensure that subclasses do not share a hash
80
        self.user_provided_columns = user_provided_columns.merge(name => connection.new_column(name, options[:default], cast_type))
81 82 83 84
      end

      # Returns an array of column objects for the table associated with this class.
      def columns
85
        @columns ||= add_user_provided_columns(connection.schema_cache.columns(table_name))
86 87 88 89 90 91 92 93 94
      end

      # Returns a hash of column objects for the table associated with this class.
      def columns_hash
        @columns_hash ||= Hash[columns.map { |c| [c.name, c] }]
      end

      def reset_column_information # :nodoc:
        super
95
        clear_caches_calculated_from_columns
96 97 98 99 100
      end

      private

      def add_user_provided_columns(schema_columns)
101 102 103 104 105 106 107 108
        existing_columns = schema_columns.map do |column|
          user_provided_columns[column.name] || column
        end

        existing_column_names = existing_columns.map(&:name)
        new_columns = user_provided_columns.except(*existing_column_names).values

        existing_columns + new_columns
109
      end
110

111
      def clear_caches_calculated_from_columns
112
        @attributes_builder = nil
113 114
        @column_defaults = nil
        @column_names = nil
115 116 117
        @column_types = nil
        @columns = nil
        @columns_hash = nil
118
        @content_columns = nil
119
        @raw_column_defaults = nil
120
      end
121 122 123
    end
  end
end