aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/attribute_set/builder.rb
blob: d4a787f2feb4eaa67ac84975f478252f707a2043 (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
module ActiveRecord
  class AttributeSet # :nodoc:
    class Builder # :nodoc:
      attr_reader :types

      def initialize(types)
        @types = types
      end

      def build_from_database(values = {}, additional_types = {})
        attributes = build_attributes_from_values(values, additional_types)
        add_uninitialized_attributes(attributes)
        AttributeSet.new(attributes)
      end

      private

      def build_attributes_from_values(values, additional_types)
        values.each_with_object({}) do |(name, value), hash|
          type = additional_types.fetch(name, types[name])
          hash[name] = Attribute.from_database(name, value, type)
        end
      end

      def add_uninitialized_attributes(attributes)
        types.each_key do |name|
          next if attributes.key? name
          type = types[name]
          attributes[name] =
            Attribute.uninitialized(name, type)
        end
      end
    end
  end
end