Fix AddLockVersionToPolls migration

1 files changed, 20 insertions, 2 deletions

diff --git a/db/migrate/20190306145741_add_lock_version_to_polls.rb b/db/migrate/20190306145741_add_lock_version_to_polls.rb
index cf506404d..5bb8cd3b4 100644
--- a/db/migrate/20190306145741_add_lock_version_to_polls.rb
+++ b/db/migrate/20190306145741_add_lock_version_to_polls.rb
@@ -1,6 +1,24 @@
+require Rails.root.join('lib', 'mastodon', 'migration_helpers')
+
 class AddLockVersionToPolls < ActiveRecord::Migration[5.2]
-  def change
-    add_column :polls, :lock_version, :integer, null: false, default: 0
+  include Mastodon::MigrationHelpers
+
+  disable_ddl_transaction!
+
+  def up
+    safety_assured do
+      add_column_with_default(
+        :polls,
+        :lock_version,
+        :integer,
+        allow_null: false,
+        default: 0
+      )
+    end
+  end
+
+  def down
+    remove_column :polls, :lock_version
   end
 end