Module: ActiveRecord::PGEnum::PostgreSQLAdapter

Defined in:
lib/active_record/pg_enum/6.1/postgresql_adapter.rb

Instance Method Summary collapse

Instance Method Details

#enum_types ⇒ Object

Helper method used by the monkeypatch internals. Provides a hash of ENUM types as they exist currently.

Example:

{ "foo_type" => ["foo", "bar", "baz"] }



15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/active_record/pg_enum/6.1/postgresql_adapter.rb', line 15

def enum_types
  res = exec_query(<<-SQL.strip_heredoc, "SCHEMA")
    SELECT t.typname AS enum_name, array_agg(e.enumlabel ORDER BY e.enumsortorder) AS enum_value
    FROM pg_type t
    JOIN pg_enum e ON t.oid = e.enumtypid
    JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
    WHERE n.nspname = ANY (current_schemas(false))
    GROUP BY enum_name;
  SQL

  coltype = res.column_types["enum_value"]
  deserialize = if coltype.respond_to?(:deserialize)
    proc { |values| coltype.deserialize(values) }
  elsif coltype.respond_to?(:type_cast_from_database)
    proc { |values| coltype.type_cast_from_database(values) }
  else
    proc { |values| coltype.type_cast(values) }
  end

  res.rows
    .map { |name, values| [name, values] }
    .sort { |a, b| a.first <=> b.first }
    .each_with_object({}) { |(name, values), memo| memo[name] = deserialize.call(values) }
end