深入理解 WordPress `wpdb` 类的 `query()` 方法源码：它如何处理不同类型的 SQL 查询。

各位观众老爷们，晚上好！ 今天咱们来聊聊 WordPress 里一个重量级人物——wpdb 类的 query() 方法。 这家伙可是 WordPress 操作数据库的核心大脑，几乎所有的数据库交互都得经过它。 咱们今天就扒一扒它的底裤，看看它到底是怎么处理各种各样的 SQL 查询的。

一、wpdb 类：你的数据库好伙伴

在开始深入 query() 之前，咱们先简单认识一下 wpdb 类。 简单来说，wpdb 就是 WordPress 封装的一个用于和数据库打交道的类。 你可以把它想象成一个数据库翻译官，你告诉它你想做什么（用 SQL 语句），它负责把你的意思传达给数据库，然后把数据库返回的结果翻译成 PHP 容易理解的形式给你。

你可以通过全局变量 $wpdb 访问这个类的实例。 比如，你想查询 wp_posts 表里所有文章的标题，就可以这么写：

global $wpdb;
$results = $wpdb->get_results( "SELECT post_title FROM {$wpdb->posts}" );

foreach ( $results as $row ) {
    echo $row->post_title . '<br>';
}

这里，$wpdb->posts 是一个预定义的变量，它存储了 wp_posts 表的实际名称（带前缀）。 这样做的目的是为了方便你在不同的 WordPress 安装中使用相同的代码，而不用担心表前缀不一样。

二、query() 方法：SQL 语句的司令部

query() 方法是 wpdb 类中最核心的方法之一。 它的作用很简单：接收一个 SQL 查询语句，然后执行它。 但看似简单的背后，却隐藏着许多玄机。

让我们看看 query() 方法的基本结构（简化版）：

public function query( $query ) {

    // 1. 初始化
    $this->result = null;
    $this->rows_affected = false;
    $this->last_error = '';
    $this->num_queries++;

    // 2. 检查是否开启了数据库错误显示
    if ( WP_DEBUG ) {
        $this->timer_start();
    }

    // 3. 实际执行 SQL 查询
    $this->last_query = $query; // 记录最后一次执行的查询
    $this->result = @mysqli_query( $this->dbh, $query ); // 执行查询

    // 4. 处理查询结果和错误
    if ( $this->result ) {
        if ( preg_match( '/^s*(create|alter|truncate|drop)s/i', $query ) ) {
            // CREATE, ALTER, TRUNCATE, DROP 语句
            $this->rows_affected = true;
        } elseif ( preg_match( '/^s*(insert|delete|update|replace)s/i', $query ) ) {
            // INSERT, DELETE, UPDATE, REPLACE 语句
            $this->rows_affected = @mysqli_affected_rows( $this->dbh );
            if ( $this->rows_affected === -1 ) {
                $this->rows_affected = false;
            }
        } else {
            // SELECT 语句
            if ( @mysqli_num_rows( $this->result ) ) {
                $this->num_rows = @mysqli_num_rows( $this->result );
            } else {
                $this->num_rows = 0;
            }
            @mysqli_data_seek( $this->result, 0 ); // 将结果指针重置到开始位置
        }
        $this->last_error = '';
    } else {
        // 查询出错
        $this->last_error = @mysqli_error( $this->dbh );
        $this->print_error();
        return false; // 返回 false 表示查询失败
    }

    // 5. 记录查询时间和内存使用情况 (如果开启了 WP_DEBUG)
    if ( WP_DEBUG ) {
        $this->timer_stop();
        $this->queries[] = array( $query, $this->timer_get_time(), $this->timer_get_memory() );
    }

    return $this->result; // 返回查询结果
}

这个简化版的代码已经展示了 query() 方法的核心逻辑：

1. 初始化： 清空之前的查询结果和错误信息，增加查询计数器。
2. 调试： 如果开启了 WP_DEBUG，记录查询开始时间。
3. 执行查询： 使用 mysqli_query() 函数执行 SQL 查询。
4. 处理结果和错误：
   • 根据 SQL 语句的类型（CREATE, ALTER, TRUNCATE, DROP, INSERT, DELETE, UPDATE, REPLACE, SELECT）采取不同的处理方式。
   • 如果是 SELECT 语句，获取结果集中的行数。
   • 如果查询出错，记录错误信息并返回 false。
5. 记录调试信息： 如果开启了 WP_DEBUG，记录查询时间和内存使用情况。
6. 返回结果： 返回查询结果。

三、SQL 语句类型识别：火眼金睛的正则表达式

query() 方法如何判断 SQL 语句的类型呢？ 答案是：正则表达式。 仔细观察上面的代码，你会发现 preg_match() 函数的身影。

if ( preg_match( '/^s*(create|alter|truncate|drop)s/i', $query ) ) {
    // CREATE, ALTER, TRUNCATE, DROP 语句
} elseif ( preg_match( '/^s*(insert|delete|update|replace)s/i', $query ) ) {
    // INSERT, DELETE, UPDATE, REPLACE 语句
} else {
    // SELECT 语句
}

这里使用了两个正则表达式来判断 SQL 语句的类型：

• '/^s*(create|alter|truncate|drop)s/i'：匹配以 CREATE, ALTER, TRUNCATE, DROP 开头的 SQL 语句。 s* 匹配零个或多个空白字符， /i 表示忽略大小写。
• '/^s*(insert|delete|update|replace)s/i'：匹配以 INSERT, DELETE, UPDATE, REPLACE 开头的 SQL 语句。

如果 SQL 语句既不是 CREATE, ALTER, TRUNCATE, DROP，也不是 INSERT, DELETE, UPDATE, REPLACE，那么就认为是 SELECT 语句。

四、不同类型 SQL 语句的处理方式：各有千秋

query() 方法根据 SQL 语句的类型采取不同的处理方式。 让我们分别来看一下：

• CREATE, ALTER, TRUNCATE, DROP 语句：

  这类语句通常用于修改数据库结构。 query() 方法会将 $this->rows_affected 设置为 true，表示执行成功。 至于实际影响的行数，这类语句通常不返回。

• INSERT, DELETE, UPDATE, REPLACE 语句：

  这类语句用于修改数据。 query() 方法会使用 mysqli_affected_rows() 函数获取受影响的行数，并将其赋值给 $this->rows_affected。 如果 mysqli_affected_rows() 返回 -1，则将 $this->rows_affected 设置为 false。

• SELECT 语句：

  这类语句用于查询数据。 query() 方法会使用 mysqli_num_rows() 函数获取结果集中的行数，并将其赋值给 $this->num_rows。 然后，使用 mysqli_data_seek() 函数将结果指针重置到开始位置，以便后续可以从头开始读取结果集。

SQL 语句类型	处理方式
CREATE	$this->rows_affected = true; 表示执行成功，但不返回实际影响的行数。
ALTER	$this->rows_affected = true; 表示执行成功，但不返回实际影响的行数。
TRUNCATE	$this->rows_affected = true; 表示执行成功，但不返回实际影响的行数。
DROP	$this->rows_affected = true; 表示执行成功，但不返回实际影响的行数。
INSERT	使用 mysqli_affected_rows() 获取受影响的行数，赋值给 $this->rows_affected。 如果返回 -1，则 $this->rows_affected = false;
DELETE	使用 mysqli_affected_rows() 获取受影响的行数，赋值给 $this->rows_affected。 如果返回 -1，则 $this->rows_affected = false;
UPDATE	使用 mysqli_affected_rows() 获取受影响的行数，赋值给 $this->rows_affected。 如果返回 -1，则 $this->rows_affected = false;
REPLACE	使用 mysqli_affected_rows() 获取受影响的行数，赋值给 $this->rows_affected。 如果返回 -1，则 $this->rows_affected = false;
SELECT	使用 mysqli_num_rows() 获取结果集中的行数，赋值给 $this->num_rows。 使用 mysqli_data_seek() 将结果指针重置到开始位置。

五、错误处理：容错机制的基石

query() 方法还包含了错误处理机制。 如果 mysqli_query() 函数执行失败，query() 方法会：

1. 使用 mysqli_error() 函数获取错误信息，并将其赋值给 $this->last_error。
2. 调用 $this->print_error() 方法打印错误信息。
3. 返回 false，表示查询失败。

$this->print_error() 方法的具体实现会根据 WordPress 的配置来决定如何显示错误信息。 在开发环境中，通常会将错误信息打印到屏幕上，方便调试。 在生产环境中，为了安全起见，通常会禁止显示错误信息，而是将错误信息记录到日志文件中。

六、性能调试：WP_DEBUG 的妙用

query() 方法还支持性能调试。 如果开启了 WP_DEBUG，query() 方法会在查询前后记录时间戳和内存使用情况，并将这些信息存储到 $this->queries 数组中。

if ( WP_DEBUG ) {
    $this->timer_start();
}

// ... 执行查询 ...

if ( WP_DEBUG ) {
    $this->timer_stop();
    $this->queries[] = array( $query, $this->timer_get_time(), $this->timer_get_memory() );
}

你可以通过 $wpdb->queries 数组查看所有执行过的 SQL 查询语句、执行时间和内存使用情况。 这对于优化数据库查询性能非常有帮助。

七、安全问题：SQL 注入的防范

虽然 wpdb 类本身并没有直接防止 SQL 注入的功能，但是 WordPress 提供了其他函数来帮助你构建安全的 SQL 查询。 比如，$wpdb->prepare() 函数可以用来预处理 SQL 查询语句，防止 SQL 注入。

global $wpdb;
$post_id = 123;
$sql = $wpdb->prepare( "SELECT post_title FROM {$wpdb->posts} WHERE ID = %d", $post_id );
$results = $wpdb->get_results( $sql );

$wpdb->prepare() 函数会将 %d 替换为整数类型的 $post_id 变量，并对 $post_id 进行转义，防止 SQL 注入。

八、总结：query() 方法的价值

wpdb 类的 query() 方法是 WordPress 操作数据库的核心。 它负责执行 SQL 查询，处理查询结果和错误，支持性能调试，并为构建安全的 SQL 查询提供了基础。

理解 query() 方法的工作原理，可以帮助你更好地理解 WordPress 的数据库操作，优化数据库查询性能，并编写更安全的代码。

九、举一反三：其他相关方法

除了 query() 方法之外，wpdb 类还提供了许多其他方法来方便你操作数据库。 比如：

• get_results()：执行 SELECT 查询，并返回结果集。
• get_row()：执行 SELECT 查询，并返回第一行结果。
• get_col()：执行 SELECT 查询，并返回第一列结果。
• get_var()：执行 SELECT 查询，并返回第一个单元格的值。
• insert()：执行 INSERT 语句。
• update()：执行 UPDATE 语句。
• delete()：执行 DELETE 语句。
• replace(): 执行 REPLACE 语句。

这些方法都是基于 query() 方法实现的，它们只是对 query() 方法进行了封装，使其更易于使用。

十、结束语：深入学习永无止境

今天我们深入探讨了 WordPress wpdb 类的 query() 方法。 希望通过今天的讲解，你对 query() 方法有了更深入的理解。 记住，学习是一个永无止境的过程。 只有不断学习，才能不断进步。

希望今天的讲座对你有所帮助。 谢谢大家！