Skip to content

Commit a3b5644

Browse files
JanJakesJanJakes
committed
wip
1 parent 7a71ad9 commit a3b5644

3 files changed

Lines changed: 95 additions & 21 deletions

File tree

‎tests/WP_PDO_MySQL_On_SQLite_PDO_API_Tests.php‎

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ class WP_PDO_MySQL_On_SQLite_PDO_API_Tests extends TestCase {
77
private $driver;
88

99
public function setUp(): void {
10-
$this->driver = new WP_PDO_MySQL_On_SQLite( 'mysql-on-sqlite:path=:memory:;dbname=WordPress;' );
10+
$this->driver = new WP_PDO_MySQL_On_SQLite( 'mysql-on-sqlite:path=:memory:;dbname=wp;' );
11+
$this->driver->setAttribute( PDO::ATTR_STRINGIFY_FETCHES, false );
1112
}
1213

1314
public function test_connection(): void {
@@ -246,8 +247,8 @@ public function data_pdo_fetch_methods(): Generator {
246247
"SELECT 1, 'abc', 2, 'two' as `2`",
247248
PDO::FETCH_BOTH,
248249
array(
249-
1 => 1, // int
250-
0 => 1, // int
250+
1 => 1,
251+
0 => 1,
251252
'abc' => 'abc',
252253
'2' => 'two',
253254
'3' => 'two',
@@ -266,7 +267,7 @@ public function data_pdo_fetch_methods(): Generator {
266267
"SELECT 1, 'abc', 2, 'two' as `2`",
267268
PDO::FETCH_ASSOC,
268269
array(
269-
'1' => 1, // int
270+
'1' => 1,
270271
'abc' => 'abc',
271272
'2' => 'two',
272273
),
@@ -277,7 +278,7 @@ public function data_pdo_fetch_methods(): Generator {
277278
"SELECT 1, 'abc', 2, 'two' as `2`",
278279
PDO::FETCH_NAMED,
279280
array(
280-
'1' => 1, // int
281+
'1' => 1,
281282
'abc' => 'abc',
282283
'2' => array( 2, 'two' ),
283284
),
@@ -288,7 +289,7 @@ public function data_pdo_fetch_methods(): Generator {
288289
"SELECT 1, 'abc', 2, 'two' as `2`",
289290
PDO::FETCH_OBJ,
290291
(object) array(
291-
'1' => 1, // int
292+
'1' => 1,
292293
'abc' => 'abc',
293294
'2' => 'two',
294295
),

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

Lines changed: 33 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -443,6 +443,13 @@ class WP_PDO_MySQL_On_SQLite extends PDO {
443443
*/
444444
private $information_schema_builder;
445445

446+
/**
447+
* PDO API: The PDO attributes of the connection.
448+
*
449+
* @var array<int, mixed>
450+
*/
451+
private $pdo_attributes = array();
452+
446453
/**
447454
* Last executed MySQL query.
448455
*
@@ -853,9 +860,9 @@ public function query( string $query, ?int $fetch_mode = null, ...$fetch_mode_ar
853860

854861
$affected_rows = is_int( $this->last_return_value ) ? $this->last_return_value : 0;
855862
$rows = is_array( $this->last_result ) ? $this->last_result : array();
856-
$columns = is_array( $this->last_column_meta ) ? array_column( $this->last_column_meta, 'name' ) : array();
863+
$columns = is_array( $this->last_column_meta ) ? $this->last_column_meta : array();
857864

858-
$stmt = new WP_PDO_Synthetic_Statement( $rows, $columns, $affected_rows );
865+
$stmt = new WP_PDO_Synthetic_Statement( $this, $columns, $rows, $affected_rows );
859866
$stmt->setFetchMode( $fetch_mode, ...$fetch_mode_args );
860867
return $stmt;
861868
} catch ( Throwable $e ) {
@@ -945,6 +952,30 @@ public function inTransaction(): bool {
945952
return $this->connection->get_pdo()->inTransaction();
946953
}
947954

955+
/**
956+
* PDO API: Set a PDO attribute.
957+
*
958+
* @param int $attribute The attribute to set.
959+
* @param mixed $value The value of the attribute.
960+
* @return bool True on success, false on failure.
961+
*/
962+
public function setAttribute( int $attribute, $value ): bool {
963+
parent::setAttribute( $attribute, $value );
964+
$this->pdo_attributes[ $attribute ] = $value;
965+
return true;
966+
}
967+
968+
/**
969+
* PDO API: Get a PDO attribute.
970+
*
971+
* @param int $attribute The attribute to get.
972+
* @return mixed The value of the attribute, null when not set.
973+
*/
974+
#[ReturnTypeWillChange]
975+
public function getAttribute( int $attribute ) {
976+
return $this->pdo_attributes[ $attribute ] ?? null;
977+
}
978+
948979
/**
949980
* Get the SQLite connection instance.
950981
*

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

Lines changed: 55 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,18 @@
1414

1515
class WP_PDO_Synthetic_Statement extends PDOStatement {
1616
/**
17-
* Column names of the columns in the result set.
17+
* The PDO connection.
18+
*
19+
* @var PDO
20+
*/
21+
private $pdo;
22+
23+
/**
24+
* Column metadata as reported by SQLite.
1825
*
1926
* @var array<string>
2027
*/
21-
private $columns = array();
28+
private $sqlite_column_meta = array();
2229

2330
/**
2431
* Rows of the result set.
@@ -44,9 +51,18 @@ class WP_PDO_Synthetic_Statement extends PDOStatement {
4451
/**
4552
* The current fetch mode.
4653
*
54+
* The PDO::FETCH_DEFAULT constant is available from PHP 8.0.
55+
*
4756
* @var int
4857
*/
49-
private $fetch_mode = PDO::FETCH_DEFAULT;
58+
private $fetch_mode = 0; // PDO::FETCH_DEFAULT
59+
60+
/**
61+
* The PDO attributes of the statement.
62+
*
63+
* @var array<int, mixed>
64+
*/
65+
private $attributes = array();
5066

5167
/**
5268
* Additional arguments for the current fetch mode.
@@ -56,21 +72,23 @@ class WP_PDO_Synthetic_Statement extends PDOStatement {
5672
private $fetch_mode_args = array();
5773

5874
public function __construct(
75+
PDO $pdo,
76+
array $sqlite_column_metadata,
5977
array $rows,
60-
array $columns,
6178
int $affected_rows = 0
6279
) {
63-
$this->rows = $rows;
64-
$this->columns = $columns;
65-
$this->affected_rows = $affected_rows;
80+
$this->pdo = $pdo;
81+
$this->sqlite_column_meta = $sqlite_column_metadata;
82+
$this->rows = $rows;
83+
$this->affected_rows = $affected_rows;
6684
}
6785

6886
public function execute( ?array $params = null ): bool {
6987
return true;
7088
}
7189

7290
public function columnCount(): int {
73-
return count( $this->columns );
91+
return count( $this->sqlite_column_meta );
7492
}
7593

7694
public function rowCount(): int {
@@ -88,18 +106,32 @@ public function fetch(
88106
}
89107
// TODO: $cursorOffset
90108

91-
if ( PDO::FETCH_DEFAULT === $mode ) {
109+
// 0 is PDO::FETCH_DEFAULT (the constant is available from PHP 8.0).
110+
if ( 0 === $mode ) {
92111
$mode = $this->fetch_mode;
93112
}
94113

95114
$row = $this->rows[ $this->row_offset ];
96115
$this->row_offset += 1;
97116

117+
$column_names = array_column( $this->sqlite_column_meta, 'name' );
118+
119+
if ( PHP_VERSION_ID < 80100 && ! $this->getAttribute( PDO::ATTR_STRINGIFY_FETCHES ) ) {
120+
foreach ( $row as $i => $value ) {
121+
$type = $this->sqlite_column_meta[ $i ]['native_type'];
122+
if ( 'integer' === $type ) {
123+
$row[ $i ] = (int) $value;
124+
} elseif ( 'float' === $type ) {
125+
$row[ $i ] = (float) $value;
126+
}
127+
}
128+
}
129+
98130
switch ( $mode ) {
99131
case PDO::FETCH_BOTH:
100132
$values = array();
101133
foreach ( $row as $i => $value ) {
102-
$name = $this->columns[ $i ];
134+
$name = $column_names[ $i ];
103135
$values[ $name ] = $value;
104136
if ( ! array_key_exists( $i, $values ) ) {
105137
$values[ $i ] = $value;
@@ -109,11 +141,11 @@ public function fetch(
109141
case PDO::FETCH_NUM:
110142
return $row;
111143
case PDO::FETCH_ASSOC:
112-
return array_combine( $this->columns, $row );
144+
return array_combine( $column_names, $row );
113145
case PDO::FETCH_NAMED:
114146
$values = array();
115147
foreach ( $row as $i => $value ) {
116-
$name = $this->columns[ $i ];
148+
$name = $column_names[ $i ];
117149
if ( is_array( $values[ $name ] ?? null ) ) {
118150
$values[ $name ][] = $value;
119151
} elseif ( array_key_exists( $name, $values ) ) {
@@ -124,7 +156,7 @@ public function fetch(
124156
}
125157
return $values;
126158
case PDO::FETCH_OBJ:
127-
$assoc = array_combine( $this->columns, $row );
159+
$assoc = array_combine( $column_names, $row );
128160
return (object) $assoc;
129161
default:
130162
throw new ValueError( sprintf( 'Unsupported fetch mode: %d', $mode ) );
@@ -167,6 +199,16 @@ public function setFetchMode( int $mode, ...$args ): bool {
167199
return true;
168200
}
169201

202+
#[ReturnTypeWillChange]
203+
public function getAttribute( int $attribute ) {
204+
return $this->attributes[ $attribute ] ?? $this->pdo->getAttribute( $attribute );
205+
}
206+
207+
public function setAttribute( int $attribute, $value ): bool {
208+
$this->attributes[ $attribute ] = $value;
209+
return true;
210+
}
211+
170212
// TODO:
171213
// public function bindColumn()
172214
// public function bindParam()

0 commit comments

Comments
 (0)