fix wait_for_notify_defer crash on bad connection

royaltm · May 24, 2014 · 8752f83 · 8752f83
1 parent 01883ac
commit 8752f83
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/pg/em.rb b/lib/pg/em.rb
@@ -698,7 +698,12 @@ def #{defer_name}(*args, &blk)
       def wait_for_notify_defer(timeout = nil, &blk)
         df = FeaturedDeferrable.new
         begin
-          setup_emio_watcher.watch_notifies(df, timeout)
+          check_async_command_aborted!
+          if status == CONNECTION_OK
+            setup_emio_watcher.watch_notify(df, timeout)
+          else
+            raise_error ConnectionBad
+          end
         rescue Error => e
           ::EM.next_tick { async_autoreconnect!(df, e) }
         rescue Exception => e