Implement PDO and PDOStatement getAttribute() and setAttribute()

Author: JanJakes

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

@@ -443,6 +443,16 @@ class WP_PDO_MySQL_On_SQLite extends PDO {
 	 */
 	private $information_schema_builder;
 
+	/**
+	 * PDO API: The PDO attributes of the connection.
+	 *
+	 * TODO: Add PDO default attribute values.
+	 *
+	 * @var array<int, mixed>
+	 */
+	private $pdo_attributes = array(
+	);
+
 	/**
 	 * Last executed MySQL query.
 	 *
@@ -775,7 +785,7 @@ public function query( string $query, ?int $fetch_mode = PDO::FETCH_COLUMN, ...$
 			$columns       = is_array( $this->last_column_meta ) ? $this->last_column_meta : array();
 			$rows          = is_array( $this->last_result ) ? $this->last_result : array();
 			$affected_rows = is_int( $this->last_return_value ) ? $this->last_return_value : 0;
-			return new WP_PDO_Synthetic_Statement( $columns, $rows, $affected_rows );
+			return new WP_PDO_Synthetic_Statement( $this, $columns, $rows, $affected_rows );
 		} catch ( Throwable $e ) {
 			try {
 				$this->rollback_user_transaction();
@@ -872,6 +882,29 @@ public function inTransaction(): bool {
 		return $this->connection->get_pdo()->inTransaction();
 	}
 
+	/**
+	 * PDO API: Set a PDO attribute.
+	 *
+	 * @param int   $attribute The attribute to set.
+	 * @param mixed $value     The value of the attribute.
+	 * @return bool            True on success, false on failure.
+	 */
+	public function setAttribute( $attribute, $value ): bool {
+		$this->pdo_attributes[ $attribute ] = $value;
+		return true;
+	}
+
+	/**
+	 * PDO API: Get a PDO attribute.
+	 *
+	 * @param  int   $attribute The attribute to get.
+	 * @return mixed            The value of the attribute.
+	 */
+	#[ReturnTypeWillChange]
+	public function getAttribute( $attribute ) {
+		return $this->pdo_attributes[ $attribute ] ?? null;
+	}
+
 	/**
 	 * Get the SQLite connection instance.
 	 *

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

@@ -95,6 +95,13 @@ class ValueError extends Error {
 class WP_PDO_Synthetic_Statement extends PDOStatement {
 	use WP_PDO_Synthetic_Statement_PHP_Compat;
 
+	/**
+	 * The PDO connection.
+	 *
+	 * @var PDO
+	 */
+	private $pdo;
+
 	/**
 	 * Basic column metadata (containing at least name, table name, and native type).
 	 *
@@ -139,18 +146,28 @@ class WP_PDO_Synthetic_Statement extends PDOStatement {
 	 */
 	private $fetch_mode_args = array();
 
+	/**
+	 * The PDO attributes set for this statement.
+	 *
+	 * @var array<int, mixed>
+	 */
+	private $attributes = array();
+
 	/**
 	 * Constructor.
 	 *
+	 * @param PDO   $pdo           The PDO connection.
 	 * @param array $columns       Basic column metadata (containing at least name, table name, and native type).
 	 * @param array $rows          Rows of the result set.
 	 * @param int   $affected_rows The number of affected rows.
 	 */
 	public function __construct(
+		PDO $pdo,
 		array $columns,
 		array $rows,
 		int $affected_rows
 	) {
+		$this->pdo           = $pdo;
 		$this->columns       = $columns;
 		$this->rows          = $rows;
 		$this->affected_rows = $affected_rows;
@@ -337,7 +354,7 @@ public function errorInfo(): array {
 	 */
 	#[ReturnTypeWillChange]
 	public function getAttribute( $attribute ) {
-		throw new RuntimeException( 'Not implemented' );
+		return $this->attributes[ $attribute ] ?? $this->pdo->getAttribute( $attribute );
 	}
 
 	/**
@@ -348,7 +365,8 @@ public function getAttribute( $attribute ) {
 	 * @return bool             True on success, false on failure.
 	 */
 	public function setAttribute( $attribute, $value ): bool {
-		throw new RuntimeException( 'Not implemented' );
+		$this->attributes[ $attribute ] = $value;
+		return true;
 	}
 
 	/**