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

Inherits:
Object
  • Object
show all
Defined in:
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

:nodoc:



11
12
13
 
# File '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 '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 '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