Improve migration

db/migrate/20181129203927_initial_migration.rb

@@ -2,29 +2,11 @@
 
 class InitialMigration < ActiveRecord::Migration[6.0]
   def change
-    reversible do |dir|
+    enum :sex, %i[male female]
-      dir.up do
-        execute <<~SQL
-          CREATE TYPE sex AS ENUM ('male', 'female');
 
-          CREATE TYPE relationship_status AS ENUM (
+    enum :relationship_status, %i[supporter excluded member]
-            'supporter',
-            'excluded',
-            'member'
-          );
 
-          CREATE TYPE relationship_role AS ENUM ('manager', 'supervisor');
+    enum :relationship_role, %i[manager supervisor]
-        SQL
-      end
-
-      dir.down do
-        execute <<~SQL
-          DROP TYPE sex;
-          DROP TYPE relationship_status;
-          DROP TYPE relationship_role;
-        SQL
-      end
-    end
 
     create_table :contacts_lists do |t|
       t.timestamps null: false
@@ -251,6 +233,21 @@ class InitialMigration < ActiveRecord::Migration[6.0]
 
 private
 
+  def enum(name, values)
+    reversible do |dir|
+      dir.up do
+        execute <<~SQL
+          CREATE TYPE #{name}
+            AS ENUM (#{values.map { |s| "'#{s}'" }.join(', ')})
+        SQL
+      end
+
+      dir.down do
+        execute "DROP TYPE #{name}"
+      end
+    end
+  end
+
   def constraint(table, name, check)
     reversible do |dir|
       dir.up do