Database#transaction returns block result

Author: alexcwatt

CHANGELOG.md

@@ -33,6 +33,7 @@ This release drops support for Ruby 2.7. [#453] @flavorjones
 - `Database#columns` returns a list of internal frozen strings. [#155, #474, #486] @tenderlove
 - Freeze results that come from the database. [#480] @tenderlove
 - The encoding of a Database is no longer cached. [#485] @tenderlove
+- `Database#transaction` returns the result of the block when used with a block. @alexcwatt
 
 
 ### Removed

lib/sqlite3/database.rb

@@ -644,16 +644,17 @@ def transaction(mode = nil)
       if block_given?
         abort = false
         begin
-          yield self
+          result = yield self
         rescue
           abort = true
           raise
         ensure
           abort and rollback or commit
         end
+        result
+      else
+        true
       end
-
-      true
     end
 
     # Commits the current transaction. If there is no current transaction,

test/test_database.rb

@@ -686,5 +686,15 @@ def test_default_transaction_mode
     ensure
       tf&.unlink
     end
+
+    def test_transaction_returns_true_without_block
+      transaction = @db.transaction
+      assert_equal true, transaction
+    end
+
+    def test_transaction_returns_block_result
+      result = @db.transaction { :foo }
+      assert_equal :foo, result
+    end
   end
 end