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.



15
16
17
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 15

def initialize(store)
  @store = store
end

Instance Method Details

#query_conditions_for_initial_load ⇒ Object 



36
37
38
39
40
41
42
43
44
45
46
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 36

def query_conditions_for_initial_load
  known_type_names = @store.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 



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 19

def run(records)
  nodes = records.reject { |row| @store.key? row["oid"].to_i }
  mapped = nodes.extract! { |row| @store.key? row["typname"] }
  ranges = nodes.extract! { |row| row["typtype"] == "r" }
  enums = nodes.extract! { |row| row["typtype"] == "e" }
  domains = nodes.extract! { |row| row["typtype"] == "d" }
  arrays = nodes.extract! { |row| row["typinput"] == "array_in" }
  composites = nodes.extract! { |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