Migrations iproved

1 files changed, 5 insertions, 18 deletions

diff --git a/db/migrate/20130218141327_convert_closed_to_state_in_merge_request.rb b/db/migrate/20130218141327_convert_closed_to_state_in_merge_request.rb
index 5aa5f48ce66..4d5c6ee569d 100644
--- a/db/migrate/20130218141327_convert_closed_to_state_in_merge_request.rb
+++ b/db/migrate/20130218141327_convert_closed_to_state_in_merge_request.rb
@@ -1,29 +1,16 @@
 class ConvertClosedToStateInMergeRequest < ActiveRecord::Migration
   def up
     MergeRequest.transaction do
-      MergeRequest.find_each do |mr|
-        if mr.closed? && mr.merged?
-          mr.state = :merged
-        else 
-          if mr.closed? 
-            mr.state = :closed
-          else 
-            mr.state = :opened
-          end
-        end
-
-        mr.save
-      end
+      MergeRequest.where("closed = 1 AND merged = 1").update_all("state = 'merged'")
+      MergeRequest.where("closed = 1 AND merged = 0").update_all("state = 'closed'")
+      MergeRequest.where("closed = 0").update_all("state = 'opened'")
     end
   end
 
   def down
     MergeRequest.transaction do
-      MergeRequest.find_each do |mr|
-        mr.closed = mr.closed? || mr.merged?
-        mr.closed = mr.merged?
-        mr.save
-      end
+      MergeRequest.where(state: :closed).update_all("closed = 1")
+      MergeRequest.where(state: :merged).update_all("closed = 1, merged = 1")
     end
   end
 end