Connection, Context: added transaction()

Author: dg

src/Database/Connection.php

@@ -152,6 +152,23 @@ public function rollBack(): void
 	}
 
 
+	/**
+	 * @return mixed
+	 */
+	public function transaction(callable $callback)
+	{
+		$this->beginTransaction();
+		try {
+			$res = $callback();
+		} catch (\Throwable $e) {
+			$this->rollBack();
+			throw $e;
+		}
+		$this->commit();
+		return $res;
+	}
+
+
 	/**
 	 * Generates and executes SQL query.
 	 */

src/Database/Context.php

@@ -60,6 +60,15 @@ public function rollBack(): void
 	}
 
 
+	/**
+	 * @return mixed
+	 */
+	public function transaction(callable $callback)
+	{
+		return $this->connection->transaction($callback);
+	}
+
+
 	public function getInsertId(string $sequence = null): string
 	{
 		return $this->connection->getInsertId($sequence);

tests/Database/Context.transaction.phpt

@@ -23,6 +23,18 @@ test(function () use ($context) {
 });
 
 
+test(function () use ($context) {
+	Assert::exception(function () use ($context) {
+		$context->transaction(function () use ($context) {
+			$context->query('DELETE FROM book');
+			throw new Exception('my exception');
+		});
+	}, Exception::class, 'my exception');
+
+	Assert::same(3, $context->fetchField('SELECT id FROM book WHERE id = ', 3));
+});
+
+
 test(function () use ($context) {
 	$context->beginTransaction();
 	$context->query('DELETE FROM book');