Refs #37678 - Update evr migration to use DSL

Katello · Aug 5, 2024 · a815009 · a815009
1 parent 98555b6
commit a815009
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/db/migrate/20240624121212_katello_recreate_evr_constructs.rb b/db/migrate/20240624121212_katello_recreate_evr_constructs.rb
@@ -1,14 +1,7 @@
 class KatelloRecreateEvrConstructs < ActiveRecord::Migration[6.1]
-  def change
-    count = select_value <<~SQL
-      SELECT count(*) FROM pg_extension WHERE extname = 'evr';
-    SQL
-    if count.to_i == 0
-      return
-    else
-      execute <<~SQL
-        DROP EXTENSION evr CASCADE;
-      SQL
+  def up
+    if extension_enabled?('evr')
+      disable_extension('evr')
 
       execute <<~SQL
               create type evr_array_item as (
@@ -156,4 +149,8 @@ def change
       create_trigger :evr_update_trigger_katello_installed_packages, on: :katello_installed_packages
     end
   end
+
+  def down
+    fail ActiveRecord::IrreversibleMigration
+  end
 end