rails_2_preload 0.2.1 → 0.2.2

data/lib/rails_2_preload/spin.rb

@@ -0,0 +1,33 @@
+require 'rails_2_preload'
+
+module Rails2Preload
+  module Spin
+
+    # Called from .spin.rb to add all hooks necessary to integrate
+    # Rails2Preload.
+    def self.add_spin_hooks
+      # Inside the context of Spin, we'll preload until the application classes
+      # (i.e. models) are loaded. This is optimal for unit testing.
+      Rails2Preload.preload_until(:load_application_classes)
+      ::Spin.hook(:before_preload) { Rails2Preload.prepare_rails }
+
+      ::Spin.hook(:after_preload) do
+        if Rails2Preload.preload_methods.include? :load_application_classes
+          # If classes are preloaded, empty the connection pool so forked
+          # processes are forced to establish new connections. Otherwise, the
+          # second time a test is run, an exception like this is raised:
+          #   ActiveRecord::StatementInvalid PGError: no connection to the server
+          ActiveRecord::Base.connection_pool.disconnect!
+        end
+      end
+
+      ::Spin.hook(:after_fork) do
+        # Create a new initializer instance, but reuse the configuration already
+        # established by the preload phase.
+        initializer = ::Rails::Initializer.new(::Rails.configuration)
+        initializer.postload
+      end
+    end
+
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_2_preload
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
   prerelease: 
 platform: ruby
 authors:
@@ -19,6 +19,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/rails_2_preload/spin.rb
 - lib/rails_2_preload.rb
 homepage: https://github.com/paperlesspost/rails-2-preload
 licenses: []