Try this way to provide database connection

521a7caf · Lin Jen-Shin · 887aeefb · 521a7caf
Commit 521a7caf authored Feb 07, 2017 by Lin Jen-Shin
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 6 deletions

db/post_migrate/20170206040400_remove_inactive_default_email_services.rb .../20170206040400_remove_inactive_default_email_services.rb +3 -6

No files found.
--- a/db/post_migrate/20170206040400_remove_inactive_default_email_services.rb
+++ b/db/post_migrate/20170206040400_remove_inactive_default_email_services.rb
@@ -31,13 +31,10 @@ class RemoveInactiveDefaultEmailServices < ActiveRecord::Migration
    end
  end

-  def with_connection
+  def with_connection(&block)
    pool = ActiveRecord::Base.establish_connection
-    connection = pool.connection
-
-    yield(connection)
-
+    pool.with_connection(&block)
  ensure
-    connection.close
+    pool.disconnect!
  end
 end