Beaneater::NotConnected error initializes with a connection instance

lib/beaneater/connection.rb

@@ -139,7 +139,7 @@ def config
     # Raises an error to be triggered when the connection has failed
     # @raise [Beaneater::NotConnected] Beanstalkd is no longer connected
     def raise_not_connected!
-      raise NotConnected, "Connection to beanstalk '#{@host}:#{@port}' is closed!"
+      raise NotConnected.new(self), "Connection to beanstalk '#{@host}:#{@port}' is closed!"
     end
   end # Connection
 end # Beaneater

lib/beaneater/errors.rb

@@ -1,6 +1,24 @@
 module Beaneater
   # Raises when a beanstalkd instance is no longer accessible.
-  class NotConnected < RuntimeError; end
+  class NotConnected < RuntimeError
+    # @!attribute connection
+    #   @return [Connection] returns the connection that errored
+    #   @example @ex.connection # => #<Beaneater::Connection host="localhost" port=11300>
+    attr_reader :connection
+
+    # Initialize connection error
+    #
+    # @param [Beaneater::Connection] the connection that errored
+    #
+    # @example
+    #   Beaneater::UnexpectedResponse.new(NotConnected, conn)
+    #
+    def initialize(connection)
+      @connection = connection
+      super
+    end
+  end
+
   # Raises when the tube name specified is invalid.
   class InvalidTubeName < RuntimeError; end
   # Raises when a job has not been reserved properly.