Rails 5: Use versioned migration

ledermann · May 26, 2016 · cfd9e6c · cfd9e6c
1 parent 57d95ad
commit cfd9e6c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/generators/rails_settings/migration/templates/migration.rb b/lib/generators/rails_settings/migration/templates/migration.rb
@@ -1,4 +1,10 @@
-class RailsSettingsMigration < ActiveRecord::Migration
+MIGRATION_BASE_CLASS = if ActiveRecord::VERSION::MAJOR >= 5
+  ActiveRecord::Migration[5.0]
+else
+  ActiveRecord::Migration
+end
+
+class RailsSettingsMigration < MIGRATION_BASE_CLASS
   def self.up
     create_table :settings do |t|
       t.string     :var,    :null => false