accept block when creating a session.

1 files changed, 7 insertions, 1 deletions

diff --git a/lib/humble/session_factory.rb b/lib/humble/session_factory.rb
index f76fb39..2d4b29d 100644
--- a/lib/humble/session_factory.rb
+++ b/lib/humble/session_factory.rb
@@ -5,7 +5,13 @@ module Humble
     end
 
     def create_session
-      Session.new(self, @configuration)
+      Session.new(self, @configuration).tap do |session|
+        begin
+          yield session if block_given?
+        ensure
+          session.dispose if block_given?
+        end
+      end
     end
 
     def create_connection