Properly disconnect databases when using Spring, fixes #97

lib/sequel_rails/railtie.rb

@@ -15,6 +15,7 @@
 require 'sequel_rails/migrations'
 require 'sequel_rails/railties/log_subscriber'
 require 'sequel_rails/railties/i18n_support'
+require 'sequel_rails/railties/spring_support'
 require 'sequel_rails/railties/controller_runtime'
 require 'sequel_rails/sequel/database/active_support_notification'
 require 'action_dispatch/middleware/session/sequel_store'
@@ -70,6 +71,15 @@ class Railtie < Rails::Railtie
       ::SequelRails.setup ::Rails.env unless app.config.sequel[:skip_connect]
     end
 
+    initializer 'sequel.spring' do |app|
+      if defined?(::Spring)
+        class ::Spring::Application
+          include ::SequelRails::SpringSupport
+          alias_method_chain :disconnect_database, :sequel
+        end
+      end
+    end
+
     # Support overwriting crucial steps in subclasses
     def configure_sequel(app)
       rails_db_config = begin

lib/sequel_rails/railties/spring_support.rb

@@ -0,0 +1,13 @@
+module SequelRails
+  module SpringSupport
+    def disconnect_database_with_sequel
+      Sequel::DATABASES.each { |db| db.disconnect } if sequel_configured?
+      disconnect_database_without_sequel
+    end
+
+  private
+    def sequel_configured?
+      defined?(Sequel::DATABASES)
+    end
+  end
+end