Module: PgEnumMigrations

Defined in:
lib/rails_string_enum/pg_enum_migration.rb

Instance Method Summary collapse

Instance Method Details

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

add_enum_value :color, ‘black’ add_enum_value :color, ‘purple’, after: ‘red’ add_enum_value :color, ‘pink’, before: ‘purple’ add_enum_value :color, ‘white’, schema: ‘public’



14
15
16
17
18
19
20
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 14

def add_enum_value(enum_name, value, before: nil, after: nil, schema: nil)
  opts = if    before then "BEFORE #{escape_enum_value(before)}"
         elsif after  then "AFTER #{escape_enum_value(after)}"
         else  ''
         end
  execute "ALTER TYPE #{enum_name(enum_name, schema)} ADD VALUE #{escape_enum_value(value)} #{opts}"
end

#create_enum(enum_name, values, schema: nil) ⇒ Object

create_enum :color, %w(red green blue) # default schema is ‘public’ create_enum :color, %w(red green blue), schema: ‘cmyk’



5
6
7
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 5

def create_enum(enum_name, values, schema: nil)
  execute "CREATE TYPE #{enum_name(enum_name, schema)} AS ENUM (#{escape_enum_values(values)})"
end

#delete_enum_value(enum_name, value_name, scheme: nil) ⇒ Object

you should delete record with deleting value Order.back_picking.delete_all or Order.back_picking.update_all(state: nil)

if exists index with condition - add_index :orders, :state, where: “state NOT IN (‘closed’, ‘canceled’, ‘returned’)” this method show exeption ERROR: operator does not exist: order_type_enum <> order_type_enum_new you must first remove and then create an index

delete_enum_value :order_state_enum, ‘back_picking’



37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 37

def delete_enum_value(enum_name, value_name, scheme: nil)
  old_values = select_values("SELECT enumlabel FROM pg_catalog.pg_enum WHERE enumtypid = #{enum_name_as_srt(enum_name, scheme)}::regtype::oid")
  new_values = old_values - Array(value_name)

  execute <<-SQL
    ALTER TYPE #{enum_name} rename to #{enum_name}_old;
    CREATE TYPE #{enum_name} AS enum (#{escape_enum_values(new_values)});
  SQL

  cols_using_enum = select_rows("SELECT table_name, column_name FROM information_schema.columns WHERE udt_name = '#{enum_name}_old'")
  cols_using_enum.each do |table_name, column_name|
    execute <<-SQL
      ALTER TABLE #{table_name}
      ALTER COLUMN #{column_name} TYPE #{enum_name} USING #{column_name}::text::#{enum_name};
    SQL
  end

  execute <<-SQL
    DROP TYPE #{enum_name}_old
  SQL
end

#drop_enum(enum_name, schema: nil) ⇒ Object

drop_enum :color drop_enum :color, schema: ‘cmyk’



24
25
26
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 24

def drop_enum(enum_name, schema: nil)
  execute "DROP TYPE #{enum_name(enum_name, schema)}"
end

#int_to_enums(table, col_name, enum_name:, definitions: nil, default: nil, use_exist_enum: false) ⇒ Object

int_to_enums :users, :partner_type, enum_name: ‘user_partner_type_enum’, definitions: { retail: 0, affiliate: 1, wholesale: 2 }



96
97
98
99
100
101
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 96

def int_to_enums(table, col_name, enum_name:, definitions: nil, default: nil, use_exist_enum: false)
  convert_sql = definitions.map {|str, int| "WHEN #{int} THEN '#{str}'" }.join(' ')
  convert_sql = "CASE #{col_name} #{convert_sql} END"

  convert_to_enum table, col_name, enum_name, definitions.keys, convert_sql, default, use_exist_enum
end

#rename_enum_value(enum_name, old_value_name, new_value_name, scheme: 'public') ⇒ Object

rename_enum_value :order_state_enum, ‘accept’, ‘init’



62
63
64
65
66
67
68
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 62

def rename_enum_value(enum_name, old_value_name, new_value_name, scheme: 'public')
  execute <<-SQL
    UPDATE pg_catalog.pg_enum
    SET enumlabel = '#{new_value_name}'
    WHERE enumtypid = '#{scheme}.#{enum_name}'::regtype::oid AND enumlabel = '#{old_value_name}'
  SQL
end

#reorder_enum_values(enum_name, ordered_values, scheme: nil) ⇒ Object

reorder_enum_values :order_state_enum, %w(lost returning obtain accept)



73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 73

def reorder_enum_values(enum_name, ordered_values, scheme: nil)
  all_values = select_values("SELECT enumlabel FROM pg_catalog.pg_enum WHERE enumtypid = '#{scheme}.#{enum_name}'::regtype::oid")
  max_order =  select_value("SELECT max(enumsortorder) FROM pg_catalog.pg_enum WHERE enumtypid = '#{scheme}.#{enum_name}'::regtype::oid").to_i + 1

  ordered_sql = (ordered_values | all_values).map.with_index{|v, i| "WHEN '#{v}' THEN #{i + max_order}"}.join(' ')

  execute <<-SQL
    UPDATE pg_catalog.pg_enum
    SET enumsortorder = CASE enumlabel #{ordered_sql} END
    WHERE enumtypid = #{enum_name_as_srt(enum_name, scheme)}::regtype::oid
  SQL
end

#string_to_enums(table, col_name, enum_name:, definitions: nil, default: nil, use_exist_enum: false) ⇒ Object

string_to_enums :order, :state, enum_name: ‘order_state_enum’, definitions: %w(accept confirmed)



89
90
91
 
# File 'lib/rails_string_enum/pg_enum_migration.rb', line 89

def string_to_enums(table, col_name, enum_name:, definitions: nil, default: nil, use_exist_enum: false)
  convert_to_enum table, col_name, enum_name, definitions, col_name, default, use_exist_enum
end