Release 0.4.0

Author: ngan

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    bigrails-redis (0.3.0)
+    bigrails-redis (0.4.0)
       rails (>= 6)
 
 GEM

README.md

@@ -95,8 +95,16 @@ This library also allows you to verify connections on demand. If you want, perfo
 # Verify all connections:
 Rails.application.redis.verify!
 
-# Verify a single connection:
-Rails.application.redis.verify!(:foobar)
+# Verify specific connections:
+Rails.application.redis.verify!(:foobar, :sidekiq)
+```
+
+### Disconnect Connections
+
+You can disconnect all connections with a single call. This is useful for "before fork" hooks.
+
+```ruby
+Rails.application.redis.disconnect
 ```
 
 ## Development

lib/big_rails/redis/registry.rb

@@ -6,6 +6,9 @@ class Registry
       class UnknownConnection < StandardError
       end
 
+      class VerificationError < StandardError
+      end
+
       attr_accessor :builder
 
       def initialize
@@ -36,11 +39,29 @@ def each(&block)
         configurations.keys.map { |name| self.for(name) }.each(&block)
       end
 
-      def verify!(name = nil)
-        if name
-          verify_connection(self.for(name))
-        else
-          each { |connection| verify_connection(connection) }
+      def disconnect
+        each do |connection|
+          if connection.is_a?(::ConnectionPool)
+            connection.reload { |conn| conn.quit }
+          else
+            connection.quit
+          end
+        end
+      end
+
+      def verify!(*names)
+        names.map! { |name| validate_name(name) }
+        names = configurations.keys if names.empty?
+        names.each do |name|
+          self.for(name).with do |connection|
+            next if connection.connected?
+
+            begin
+              connection.quit
+            rescue
+              raise VerificationError, "verification for '#{name}' failed"
+            end
+          end
         end
 
         true
@@ -66,14 +87,6 @@ def build_wrapped_connection(connection)
         end
       end
 
-      def verify_connection(connection)
-        connection.with do |conn|
-          connected = conn.connected?
-          conn.ping
-          conn.quit unless connected
-        end
-      end
-
       def validate_name(name)
         name = name.to_s
         unless configurations.key?(name)

lib/big_rails/redis/version.rb

@@ -2,6 +2,6 @@
 
 module BigRails
   module Redis
-    VERSION = "0.3.0"
+    VERSION = "0.4.0"
   end
 end

spec/big_rails/redis/registry_spec.rb

@@ -75,9 +75,9 @@
       it "verifies specified connection" do
         conn = instance.for("default")
         conn2 = instance.for("pooled")
-        expect(conn).to receive(:ping).and_call_original
+        expect(conn).to receive(:quit).and_call_original
         conn2.with do |redis|
-          expect(redis).to receive(:ping).and_call_original
+          expect(redis).to receive(:quit).and_call_original
         end
 
         instance.verify!("default")
@@ -89,9 +89,9 @@
       it "verifies all connections" do
         conn = instance.for("default")
         conn2 = instance.for("pooled")
-        expect(conn).to receive(:ping).and_call_original
+        expect(conn).to receive(:quit).and_call_original
         conn2.with do |redis|
-          expect(redis).to receive(:ping).and_call_original
+          expect(redis).to receive(:quit).and_call_original
         end
 
         instance.verify!