Module: EnumKit::ActiveRecordExtensions::ConnectionAdapters::PostgreSQLAdapter

Defined in:
lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb

Overview

:nodoc:

Constant Summary collapse

ENUM_QUERY =

Returns An SQL query that returns all available enum types in the database.

Returns:

  • An SQL query that returns all available enum types in the database.

"SELECT\npg_type.OID,\npg_type.typname,\npg_type.typtype,\narray_to_string(array_agg(pg_enum.enumlabel ORDER BY pg_enum.enumsortorder), '\\t\\t', '') as values\nFROM pg_type\nLEFT JOIN pg_enum ON pg_enum.enumtypid = pg_type.oid\nWHERE pg_type.typtype = 'e'\nGROUP BY pg_type.OID, pg_type.typname, pg_type.typtype\nORDER BY pg_type.typname\n"

Instance Method Summary collapse

Instance Method Details

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

Add a new value to an enum type in the database.

Note that you can’t specify both :before and :after.

Raises:

Parameters:

  • The enum’s name.

  • The value to add.

  • (defaults to: nil)

    An existing value after which the new value should be inserted.

  • (defaults to: nil)

    An existing value before which the new value should be inserted.



81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 81

def add_enum_value(name, value, after: nil, before: nil)
  name  = EnumKit.sanitize_name!(name)
  value = EnumKit.sanitize_value!(value)

  statement = "ALTER TYPE #{name} ADD VALUE #{EnumKit.sqlize(value)}"

  raise ArgumentError, "You can't specify both :before and :after" if before && after

  statement += " AFTER #{EnumKit.sqlize(EnumKit.sanitize_value!(after))}"   if after
  statement += " BEFORE #{EnumKit.sqlize(EnumKit.sanitize_value!(before))}" if before

  execute(statement)
end

#create_enum(name, values) ⇒ Object

Create a new enum type in the database.

Parameters:

  • The enum’s name.

  • The enum’s acceptable values.



43
44
45
46
47
48
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 43

def create_enum(name, values)
  name   = EnumKit.sanitize_name!(name)
  values = EnumKit.sanitize_values!(values)

  execute "CREATE TYPE #{name} AS ENUM #{EnumKit.sqlize(values)}"
end

#drop_enum(name) ⇒ Object

Drop an existing enum type from the database.

Parameters:

  • The enum’s name.



66
67
68
69
70
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 66

def drop_enum(name)
  name = EnumKit.sanitize_name!(name)

  execute "DROP TYPE #{name}"
end

#ensure_renaming_enum_values_is_supported! ⇒ Object

Raise an exception if the active PostgreSQL version doesn’t support renaming enum values.

Raises:



127
128
129
130
131
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 127

def ensure_renaming_enum_values_is_supported!
  return if ActiveRecord::Base.connection.postgresql_version >= 100_000

  raise NotImplementedError, 'PostgreSQL 10.0+ is required to enable renaming of enum values.'
end

#enums ⇒ Hash

Returns The enum types available in the database.

Returns:

  • The enum types available in the database.



32
33
34
35
36
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 32

def enums
  select_all(ENUM_QUERY.tr("\n", ' ').strip).each_with_object({}) do |row, enums|
    enums[row['typname'].to_sym] = row['values'].split("\t\t")
  end
end

#migration_keys ⇒ Object

:nodoc:



113
114
115
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 113

def migration_keys
  super + [:enum_type]
end

#prepare_column_options(column) ⇒ Object

:nodoc:



119
120
121
122
123
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 119

def prepare_column_options(column)
  spec = super
  spec[:enum_type] = column.sql_type.inspect if column.type == :enum
  spec
end

#rename_enum(current_name, new_name) ⇒ Object

Rename an existing enum type.

Parameters:

  • The enum’s current name.

  • The enum’s new name.



55
56
57
58
59
60
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 55

def rename_enum(current_name, new_name)
  current_name = EnumKit.sanitize_name!(current_name)
  new_name     = EnumKit.sanitize_name!(new_name)

  execute "ALTER TYPE #{current_name} RENAME TO #{new_name}"
end

#rename_enum_value(name, current_name, new_name) ⇒ Object

Rename a value within an enum type in the database.

Parameters:

  • The enum’s name.

  • The enum value’s current name.

  • The enum value’s new name.



101
102
103
104
105
106
107
108
109
 
# File 'lib/enum_kit/active_record_extensions/connection_adapters/postgresql_adapter.rb', line 101

def rename_enum_value(name, current_name, new_name)
  ensure_renaming_enum_values_is_supported!

  name         = EnumKit.sanitize_name!(name)
  current_name = EnumKit.sanitize_value!(current_name)
  new_name     = EnumKit.sanitize_value!(new_name)

  execute "ALTER TYPE #{name} RENAME VALUE #{EnumKit.sqlize(current_name)} TO #{EnumKit.sqlize(new_name)}"
end