From dfb45374d220287012ffaae68ceee7159993bbac Mon Sep 17 00:00:00 2001
From: ThibG <thib@sitedethib.com>
Date: Thu, 7 Mar 2019 01:50:06 +0100
Subject: [PATCH] Fix AddLockVersionToPolls migration (#10200)

---
 ...0190306145741_add_lock_version_to_polls.rb | 22 +++++++++++++++++--
 1 file 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 cf506404da..5bb8cd3b48 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