Using custom Statement class, to support this type of flow: $stmt = $…

…db->prepare(); $db->multiAtomic(function() use ($stmt) { $stmt->execute(); });

lib/Db.php

@@ -68,6 +68,7 @@ function __construct($dsn, $user = null, $password = null, array $options = [])
         $this->dsn = $dsn;
         $this->user = $user;
         $this->password = $password;
+        $options[\PDO::ATTR_STATEMENT_CLASS] = [Statement::class, [$this]];
         $this->options = $options;
         $this->id = self::$nextId++;
         $this->queryCount = 0;
@@ -194,6 +195,7 @@ public function prepare($statement, array $driver_options = array())
     {
         $timer = $this->debugLog("prepare");
         $this->beginTransactionIfNeeded();
+        /** @var Statement $ret */
         $ret = $this->getPdo()->prepare($statement, $driver_options);
         $this->queryCount++;
         $this->debugTimerEnd($timer, "prepare(): " . $statement);
@@ -329,7 +331,7 @@ public function emulateRollback()
         $this->shouldStartTransaction = false;
     }
 
-    protected function beginTransactionIfNeeded()
+    public function beginTransactionIfNeeded()
     {
         if ($this->shouldStartTransaction) {
             $this->debugLog("Actually starting a transaction");

lib/Statement.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace tinyorm;
+
+class Statement extends \PDOStatement
+{
+    /** @var Db */
+    private $db;
+
+    protected function __construct(Db $db)
+    {
+        $this->db = $db;
+    }
+
+    function execute($input_parameters = null)
+    {
+        $this->db->beginTransactionIfNeeded();
+        return parent::execute($input_parameters);
+    }
+}