Implement PDO::exec() API

Author: JanJakes

tests/WP_PDO_MySQL_On_SQLite_PDO_API_Tests.php

@@ -20,6 +20,38 @@ public function test_query(): void {
 		$this->assertInstanceOf( PDOStatement::class, $result );
 	}
 
+	public function test_exec(): void {
+		$result = $this->driver->exec( 'SELECT 1' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'CREATE TABLE t (id INT)' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'INSERT INTO t (id) VALUES (1)' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'INSERT INTO t (id) VALUES (2), (3)' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id = 0' );
+		$this->assertEquals( 0, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id = 1' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'UPDATE t SET id = 10 + id WHERE id < 10' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'DELETE FROM t WHERE id = 11' );
+		$this->assertEquals( 1, $result );
+
+		$result = $this->driver->exec( 'DELETE FROM t' );
+		$this->assertEquals( 2, $result );
+
+		$result = $this->driver->exec( 'DROP TABLE t' );
+		$this->assertEquals( 0, $result );
+	}
+
 	public function test_begin_transaction(): void {
 		$result = $this->driver->beginTransaction();
 		$this->assertTrue( $result );

wp-includes/sqlite-ast/class-wp-pdo-mysql-on-sqlite.php

@@ -772,7 +772,8 @@ public function query( string $query, ?int $fetch_mode = PDO::FETCH_COLUMN, ...$
 				$this->commit_wrapper_transaction();
 			}
 
-			return new WP_PDO_Synthetic_Statement();
+			$affected_rows = is_int( $this->last_return_value ) ? $this->last_return_value : 0;
+			return new WP_PDO_Synthetic_Statement( $affected_rows );
 		} catch ( Throwable $e ) {
 			try {
 				$this->rollback_user_transaction();
@@ -789,6 +790,17 @@ public function query( string $query, ?int $fetch_mode = PDO::FETCH_COLUMN, ...$
 		}
 	}
 
+	/**
+	 * PDO API: Execute a MySQL statement and return the number of affected rows.
+	 *
+	 * @return int|false The number of affected rows or false on failure.
+	 */
+	#[ReturnTypeWillChange]
+	public function exec( $query ) {
+		$stmt = $this->query( $query );
+		return $stmt->rowCount();
+	}
+
 	/**
 	 * PDO API: Begin a transaction.
 	 *

wp-includes/sqlite-ast/class-wp-pdo-synthetic-statement.php

@@ -87,6 +87,24 @@ public function fetchAll( $mode = PDO::FETCH_DEFAULT, ...$args ): array {
 class WP_PDO_Synthetic_Statement extends PDOStatement {
 	use WP_PDO_Synthetic_Statement_PHP_Compat;
 
+	/**
+	 * The number of affected rows.
+	 *
+	 * @var int
+	 */
+	private $affected_rows;
+
+	/**
+	 * Constructor.
+	 *
+	 * @param int $affected_rows The number of affected rows.
+	 */
+	public function __construct(
+		int $affected_rows = 0
+	) {
+		$this->affected_rows = $affected_rows;
+	}
+
 	/**
 	 * Execute a prepared statement.
 	 *
@@ -112,7 +130,7 @@ public function columnCount(): int {
 	 * @return int The number of rows affected by the statement.
 	 */
 	public function rowCount(): int {
-		throw new RuntimeException( 'Not implemented' );
+		return $this->affected_rows;
 	}
 
 	/**