Changeset 60622

Timestamp:

08/10/2025 08:58:31 PM (2 months ago)

Author:

SergeyBiryukov

Message:

Database: Add support for SET STATEMENT in wpdb::get_table_from_query().

MariaDB supports setting various parameters for queries. This commit ensures that wpdb can properly retrieve the table name from such queries.

Reference: ​MariaDB Documentation: SET STATEMENT.

Follow-up to [30345].

Props dd32.
Fixes #63777.

Location:

trunk

Files:

• src/wp-includes/class-wpdb.php (modified) (1 diff)
• tests/phpunit/tests/db.php (modified) (1 diff)

trunk/src/wp-includes/class-wpdb.php

@@ -3804 +3804 @@
         $query = preg_replace( '/\((?!\s*select)[^(]*?\)/is', '()', $query );
 
+        // Strip any leading SET STATEMENT statements.
+        $query = preg_replace( '/^SET STATEMENT.+?\sFOR\s+/is', '', $query );
+
         // Quickly match most common queries.
         if ( preg_match(

trunk/tests/phpunit/tests/db.php

@@ -851 +851 @@
             // @ticket 32763
             'SELECT ' . str_repeat( 'a', 10000 ) . " FROM (SELECT * FROM $table) as subquery",
+
+            // @ticket 63777
+            "SET STATEMENT max_statement_time=1 FOR SELECT * FROM $table",
         );