From 6bd8e351361a895e7ada7d7702bb099034103d62 Mon Sep 17 00:00:00 2001
From: Jeremy Kemper <jeremy@bitsweat.net>
Date: Sun, 26 Feb 2006 23:12:01 +0000
Subject: ActiveRecord::Base.remove_connection explicitly closes database
 connections and doesn't corrupt the connection cache. Introducing the
 disconnect! instance method for the PostgreSQL, MySQL, and SQL Server
 adapters; implementations for the others are welcome. References #3591.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@3674 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 .../abstract/connection_specification.rb                    | 13 +++++++------
 .../active_record/connection_adapters/abstract_adapter.rb   |  8 +++++++-
 .../lib/active_record/connection_adapters/mysql_adapter.rb  |  6 +++++-
 .../active_record/connection_adapters/postgresql_adapter.rb |  5 +++++
 .../active_record/connection_adapters/sqlserver_adapter.rb  |  8 +++++++-
 5 files changed, 31 insertions(+), 9 deletions(-)

(limited to 'activerecord/lib')

diff --git a/activerecord/lib/active_record/connection_adapters/abstract/connection_specification.rb b/activerecord/lib/active_record/connection_adapters/abstract/connection_specification.rb
index 84dde1b14f..a73ce02104 100644
--- a/activerecord/lib/active_record/connection_adapters/abstract/connection_specification.rb
+++ b/activerecord/lib/active_record/connection_adapters/abstract/connection_specification.rb
@@ -130,12 +130,13 @@ module ActiveRecord
     # can be used as argument for establish_connection, for easy
     # re-establishing of the connection.
     def self.remove_connection(klass=self)
-      conn = @@defined_connections[klass.name]
-      @@defined_connections.delete(klass.name)
-      @@connection_cache[Thread.current.object_id].delete(klass.name)
-      active_connections.delete(klass.name)
-      @connection = nil
-      conn.config if conn
+      spec = @@defined_connections[klass.name]
+      konn = active_connections[klass.name]
+      @@defined_connections.delete_if { |key, value| value == spec }
+      @@connection_cache[Thread.current.object_id].delete_if { |key, value| value == konn }
+      active_connections.delete_if { |key, value| value == konn }
+      konn.disconnect! if konn
+      spec.config if spec
     end
 
     # Set the connection for the class.
diff --git a/activerecord/lib/active_record/connection_adapters/abstract_adapter.rb b/activerecord/lib/active_record/connection_adapters/abstract_adapter.rb
index b01ef89dfc..8187112b64 100755
--- a/activerecord/lib/active_record/connection_adapters/abstract_adapter.rb
+++ b/activerecord/lib/active_record/connection_adapters/abstract_adapter.rb
@@ -63,11 +63,17 @@ module ActiveRecord
 
       # Is this connection active and ready to perform queries?
       def active?
-        true
+        @active != false
       end
 
       # Close this connection and open a new one in its place.
       def reconnect!
+        @active = true
+      end
+
+      # Close this connection
+      def disconnect!
+        @active = false
       end
 
 
diff --git a/activerecord/lib/active_record/connection_adapters/mysql_adapter.rb b/activerecord/lib/active_record/connection_adapters/mysql_adapter.rb
index 6c768cb9f6..dd7331c57d 100755
--- a/activerecord/lib/active_record/connection_adapters/mysql_adapter.rb
+++ b/activerecord/lib/active_record/connection_adapters/mysql_adapter.rb
@@ -160,9 +160,13 @@ module ActiveRecord
       end
 
       def reconnect!
-        @connection.close rescue nil
+        disconnect!
         connect
       end
+      
+      def disconnect!
+        @connection.close rescue nil
+      end
 
 
       # DATABASE STATEMENTS ======================================
diff --git a/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb b/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb
index a5bef4138d..86b89c6c1c 100644
--- a/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb
+++ b/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb
@@ -75,6 +75,11 @@ module ActiveRecord
           configure_connection
         end
       end
+      
+      def disconnect!
+        # Both postgres and postgres-pr respond to :close
+        @connection.close rescue nil
+      end
 
       def native_database_types
         {
diff --git a/activerecord/lib/active_record/connection_adapters/sqlserver_adapter.rb b/activerecord/lib/active_record/connection_adapters/sqlserver_adapter.rb
index 708ec086fd..be06887fa0 100644
--- a/activerecord/lib/active_record/connection_adapters/sqlserver_adapter.rb
+++ b/activerecord/lib/active_record/connection_adapters/sqlserver_adapter.rb
@@ -216,12 +216,18 @@ module ActiveRecord
 
       # Reconnects to the database, returns false if no connection could be made.
       def reconnect!
-        @connection.disconnect rescue nil
+        disconnect!
         @connection = DBI.connect(*@connection_options)
       rescue DBI::DatabaseError => e
         @logger.warn "#{adapter_name} reconnection failed: #{e.message}" if @logger
         false
       end
+      
+      # Disconnects from the database
+      
+      def disconnect!
+        @connection.disconnect rescue nil
+      end
 
       def select_all(sql, name = nil)
         select(sql, name)
-- 
cgit v1.2.3