Class: ActiveRecord::ConnectionAdapters::PostgreSQL::OID::TypeMapInitializer

Inherits:
Object
  • Object
show all
Defined in:
activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb

Overview

This class uses the data from PostgreSQL pg_type table to build the OID -> Type mapping.

- OID is an integer representing the type.
- Type is an OID::Type object.

This class has side effects on the store passed during initialization.

Instance Method Summary collapse

Constructor Details

#initialize(store) ⇒ TypeMapInitializer

Returns a new instance of TypeMapInitializer.


11
12
13
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 11

def initialize(store)
  @store = store
end

Instance Method Details

#query_conditions_for_initial_load(type_map) ⇒ Object


32
33
34
35
36
37
38
39
40
41
42
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 32

def query_conditions_for_initial_load(type_map)
  known_type_names = type_map.keys.map { |n| "'#{n}'" }
  known_type_types = %w('r' 'e' 'd')
  <<-SQL % [known_type_names.join(", "), known_type_types.join(", ")]
    WHERE
      t.typname IN (%s)
      OR t.typtype IN (%s)
      OR t.typinput = 'array_in(cstring,oid,integer)'::regprocedure
      OR t.typelem != 0
  SQL
end

#run(records) ⇒ Object


15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 15

def run(records)
  nodes = records.reject { |row| @store.key? row['oid'].to_i }
  mapped, nodes = nodes.partition { |row| @store.key? row['typname'] }
  ranges, nodes = nodes.partition { |row| row['typtype'] == 'r'.freeze }
  enums, nodes = nodes.partition { |row| row['typtype'] == 'e'.freeze }
  domains, nodes = nodes.partition { |row| row['typtype'] == 'd'.freeze }
  arrays, nodes = nodes.partition { |row| row['typinput'] == 'array_in'.freeze }
  composites, nodes = nodes.partition { |row| row['typelem'].to_i != 0 }

  mapped.each     { |row| register_mapped_type(row)    }
  enums.each      { |row| register_enum_type(row)      }
  domains.each    { |row| register_domain_type(row)    }
  arrays.each     { |row| register_array_type(row)     }
  ranges.each     { |row| register_range_type(row)     }
  composites.each { |row| register_composite_type(row) }
end