Module: ActiveRecord::PostgresEnum::PostgreSQLAdapter

Defined in:
lib/active_record/postgres_enum/postgresql_adapter.rb

Constant Summary collapse

DEFINED_ENUMS_QUERY = 
<<~SQL
  SELECT
    t.OID,
    t.typname,
    t.typtype,
    array_to_string(array_agg(e.enumlabel ORDER BY e.enumsortorder), '\t\t', '') as enumlabels
  FROM pg_type t
  INNER JOIN pg_namespace n ON n.oid = t.typnamespace
  LEFT JOIN pg_enum e ON e.enumtypid = t.oid
  WHERE t.typtype = 'e'
  AND n.nspname = ANY(current_schemas(true))
  GROUP BY t.OID, t.typname, t.typtype
  ORDER BY t.typname
SQL

Instance Method Summary collapse

Instance Method Details

#add_enum_value(name, value, after: nil, before: nil, if_not_exists: nil) ⇒ Object 



48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 48

def add_enum_value(name, value, after: nil, before: nil, if_not_exists: nil)
  if_not_exists_statement = "IF NOT EXISTS" if if_not_exists

  sql = "ALTER TYPE #{name} ADD VALUE #{if_not_exists_statement} #{quote value}"
  if after
    sql += " AFTER #{quote after}"
  elsif before
    sql += " BEFORE #{quote before}"
  end
  execute sql
end

#create_enum(name, values, force: false, if_not_exists: nil) ⇒ Object 



27
28
29
30
31
32
33
34
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 27

def create_enum(name, values, force: false, if_not_exists: nil)
  return if if_not_exists && enum_types.include?(name.to_sym)

  drop_enum(name, cascade: force == :cascade, if_exists: true) if force

  values = values.map { |v| quote v }
  execute "CREATE TYPE #{name} AS ENUM (#{values.join(", ")})"
end

#drop_enum(name, cascade: nil, if_exists: nil) ⇒ Object 



36
37
38
39
40
41
42
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 36

def drop_enum(name, cascade: nil, if_exists: nil)
  if_exists_statement = "IF EXISTS" if if_exists
  cascade_statement = "CASCADE" if cascade

  sql = "DROP TYPE #{if_exists_statement} #{name} #{cascade_statement}"
  execute sql
end

#enum_types ⇒ Object 



21
22
23
24
25
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 21

def enum_types
  select_all(DEFINED_ENUMS_QUERY).each_with_object({}) do |row, memo|
    memo[row["typname"].to_sym] = row["enumlabels"].split("\t\t")
  end
end

#remove_enum_value(name, value) ⇒ Object 



60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 60

def remove_enum_value(name, value)
  sql = %{
    DELETE FROM pg_enum
    WHERE enumlabel = #{quote value}
    AND enumtypid IN (SELECT t.oid
                      FROM pg_type t
                      INNER JOIN pg_namespace n ON n.oid = t.typnamespace
                      WHERE t.typname = '#{name}'
                      AND n.nspname = ANY(current_schemas(true)))
  }
  execute sql
end

#rename_enum(name, new_name) ⇒ Object 



44
45
46
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 44

def rename_enum(name, new_name)
  execute "ALTER TYPE #{name} RENAME TO #{new_name}"
end

#rename_enum_value(name, existing_value, new_value) ⇒ Object 



73
74
75
76
77
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 73

def rename_enum_value(name, existing_value, new_value)
  raise "Renaming enum values is only supported in PostgreSQL 10.0+" unless rename_enum_value_supported?

  execute "ALTER TYPE #{name} RENAME VALUE #{quote existing_value} TO #{quote new_value}"
end

#rename_enum_value_supported? ⇒ Boolean

Returns:

  • (Boolean) 


79
80
81
 
# File 'lib/active_record/postgres_enum/postgresql_adapter.rb', line 79

def rename_enum_value_supported?
  ActiveRecord::Base.connection.send(:postgresql_version) >= 100_000
end