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

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 117

def assert_valid_registration(oid, oid_type)
  raise ArgumentError, "can't register nil type for OID #{oid}" if oid_type.nil?
  oid.to_i
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 52

def query_conditions_for_array_types
  known_type_oids = @store.keys.reject { |k| k.is_a?(String) }
  <<~SQL % [known_type_oids.join(", ")]
    WHERE
      t.typelem IN (%s)
  SQL
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 36

def query_conditions_for_known_type_names
  known_type_names = @store.keys.map { |n| "'#{n}'" }
  <<~SQL % known_type_names.join(", ")
    WHERE
      t.typname IN (%s)
  SQL
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 44

def query_conditions_for_known_type_types
  known_type_types = %w('r' 'e' 'd')
  <<~SQL % known_type_types.join(", ")
    WHERE
      t.typtype IN (%s)
  SQL
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 95

def register(oid, oid_type = nil, &block)
  oid = assert_valid_registration(oid, oid_type || block)
  if block_given?
    @store.register_type(oid, &block)
  else
    @store.register_type(oid, oid_type)
  end
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 69

def register_array_type(row)
  register_with_subtype(row["oid"], row["typelem"].to_i) do |subtype|
    OID::Array.new(subtype, row["typdelim"])
  end
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 89

def register_composite_type(row)
  if subtype = @store.lookup(row["typelem"].to_i)
    register row["oid"], OID::Vector.new(row["typdelim"], subtype)
  end
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 81

def register_domain_type(row)
  if base_type = @store.lookup(row["typbasetype"].to_i)
    register row["oid"], base_type
  else
    warn "unknown base type (OID: #{row["typbasetype"]}) for domain #{row["typname"]}."
  end
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 65

def register_enum_type(row)
  register row["oid"], OID::Enum.new
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 61

def register_mapped_type(row)
  alias_type row["oid"], row["typname"]
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 75

def register_range_type(row)
  register_with_subtype(row["oid"], row["rngsubtype"].to_i) do |subtype|
    OID::Range.new(subtype, row["typname"].to_sym)
  end
end

# File 'activerecord/lib/active_record/connection_adapters/postgresql/oid/type_map_initializer.rb', line 109

def register_with_subtype(oid, target_oid)
  if @store.key?(target_oid)
    register(oid) do |_, *args|
      yield @store.lookup(target_oid, *args)
    end
  end
end

# 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