From 3e5a0bc8250b3dc806e97e8370c319c40fc5ea28 Mon Sep 17 00:00:00 2001
From: Thibaut Girka <thib@sitedethib.com>
Date: Wed, 6 Mar 2019 23:06:09 +0100
Subject: Fix AddLockVersionToPolls migration

---
 .../20190306145741_add_lock_version_to_polls.rb    | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

(limited to 'db')

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
 
-- 
cgit