WordPress核心函数wp_get_current_user如何在登录态中加载全局对象 - 智猿学院-前后端，数据库，人工智能，云计算等领域前沿技术讲座

好的，下面开始我们的讲座。

讲座主题：WordPress核心函数wp_get_current_user：登录态中全局对象的加载机制

大家好，今天我们来深入探讨WordPress核心函数wp_get_current_user，重点分析其在登录状态下如何加载全局对象，以及背后的原理和机制。理解这一过程对于我们进行WordPress主题和插件开发至关重要，能帮助我们更好地处理用户认证和授权相关的问题。

一、wp_get_current_user的基本功能与作用

wp_get_current_user函数是WordPress中用于获取当前用户信息的关键函数。无论用户是否登录，它都会尝试返回一个WP_User对象，代表当前用户。

未登录用户： 如果用户未登录，该函数会返回一个WP_User对象，但其ID属性为0，其他属性为空。
已登录用户： 如果用户已登录，该函数会返回一个包含该用户所有信息的WP_User对象，例如用户名、邮箱、角色、权限等。

二、wp_get_current_user的源代码分析（简化版）

为了更好地理解其工作原理，我们先看一个wp_get_current_user的简化版代码：

<?php

/**
 * 获取当前用户对象.
 *
 * @return WP_User 当前用户对象.
 */
function my_wp_get_current_user() {
    static $current_user = null;

    if ( null === $current_user ) {
        if ( function_exists( 'get_current_user_id' ) && get_current_user_id() ) {
            $user_id = get_current_user_id();
            $current_user = get_user( $user_id );
        } else {
            $current_user = new WP_User( 0 ); // 未登录用户
        }
    }

    return $current_user;
}

?>

这段代码的关键点在于：

静态变量缓存： 使用 static $current_user 来缓存用户对象。这意味着函数只会执行一次用户信息的获取过程，后续调用直接返回缓存的对象，避免重复查询数据库，提高性能。
get_current_user_id()判断： 依赖于get_current_user_id()函数来判断用户是否已登录。get_current_user_id()函数通常依赖于cookie或者session来确定当前用户的ID。
get_user()加载用户对象： 如果用户已登录，则使用get_user( $user_id )函数通过用户ID从数据库中加载完整的WP_User对象。
未登录用户的处理： 如果用户未登录，则创建一个ID为0的WP_User对象。

三、登录态判断的机制

wp_get_current_user能正确判断登录态，依赖于WordPress的身份验证机制。WordPress 使用 cookies 来跟踪用户的登录状态。具体过程如下：

登录过程： 当用户成功登录时，WordPress会设置几个 cookies。其中最重要的两个是：
- wordpress_[hash]：存储用户的认证信息。
- wordpress_logged_in_[hash]：指示用户已登录。
请求处理： 在每个请求中，WordPress会检查这些 cookies。如果 cookie 存在且有效，WordPress 会尝试根据 cookie 中的信息来恢复用户的会话。
get_current_user_id()的实现： get_current_user_id()函数会读取这些 cookies，并根据 cookies 中的信息来确定当前用户的 ID。如果 cookies 不存在或无效，get_current_user_id() 会返回 0。

四、全局对象加载的深入分析

现在，我们来深入分析wp_get_current_user如何加载全局对象，这里涉及几个关键的全局变量：

$current_user：这是一个全局变量，用于存储当前用户对象。wp_get_current_user 函数的目的是确保这个全局变量被正确初始化。
$wpdb： WordPress数据库连接对象，用户数据存储在此数据库中.

4.1 初始化$current_user全局变量

在WordPress的核心启动过程中（通常是wp-settings.php），会调用wp_get_current_user来初始化$current_user全局变量。

<?php
// In wp-settings.php or similar core file:

global $current_user;

if ( !isset($current_user) ) {
    $current_user = wp_get_current_user();
}
?>

这段代码确保了 $current_user 全局变量在 WordPress 加载的早期阶段就被初始化，并且只初始化一次。

4.2 get_user()函数：从数据库加载用户数据

当用户已登录时，wp_get_current_user 会调用 get_user() 函数来加载完整的用户数据。get_user() 函数的实现大致如下：

<?php
function get_user( $id ) {
    global $wpdb;

    $id = (int) $id;
    if ( !$id ) {
        return false;
    }

    // Check cache first (omitted for brevity)

    $data = wp_cache_get( $id, 'users' );
    if ( false === $data ) {
        $user = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$wpdb->users} WHERE ID = %d", $id ) );

        if ( empty( $user ) ) {
            return false;
        }

        $data = new WP_User( $user );
        wp_cache_set( $id, $data, 'users' );

    } else {

        $data = new WP_User($data);
    }
    return $data;
}
?>

get_user() 函数的关键步骤：

连接数据库： 通过全局变量 $wpdb 访问 WordPress 数据库连接对象。
查询数据库： 使用 $wpdb->get_row() 执行 SQL 查询，从 wp_users 表中检索用户数据。SQL查询语句会使用$wpdb->prepare()进行预处理，防止SQL注入。
创建WP_User对象： 使用检索到的用户数据创建一个 WP_User 对象。
缓存用户数据： 将用户数据缓存起来，以便后续访问，提高性能。wp_cache_get()和wp_cache_set()函数用于缓存操作。

4.3 WP_User对象：存储用户信息

WP_User 类是 WordPress 中用于表示用户的核心类。它包含了用户的各种属性，例如 ID、用户名、邮箱、角色、权限等。

<?php
class WP_User {
    public $ID = 0;
    public $user_login;
    public $user_pass;
    public $user_nicename;
    public $user_email;
    public $user_url;
    public $user_registered;
    public $user_activation_key;
    public $user_status;
    public $display_name;
    public $roles = array();
    public $caps = array();
    public $allcaps = array();
    public $filter;

    public function __construct( $id = 0, $name = '', $site_id = '' ) {
        if ( ! empty( $id ) ) {
            $this->init( $id, $name, $site_id );
        }
    }

    public function init( $id, $name = '', $site_id = '' ) {

        if ( is_object( $id ) ) {
            foreach ( get_object_vars( $id ) as $key => $value ) {
                $this->$key = $value;
            }
            $this->filter = 'raw';
        } elseif ( is_array( $id ) ) {
            foreach ( $id as $key => $value ) {
                $this->$key = $value;
            }
            $this->filter = 'raw';
        } else {
            $id = (int) $id;
            if ( ! $id ) {
                return;
            }

            $data = get_userdata( $id );
            if ( false === $data ) {
                return;
            }

            foreach ( get_object_vars( $data ) as $key => $value ) {
                $this->$key = $value;
            }
        }

        $this->get_role_caps();
    }

    public function get_role_caps() {
        global $wpdb;

        $this->caps    = apply_filters( 'user_has_cap', $this->caps, $this->roles, $this );
        $this->allcaps = array_merge( (array) $this->caps );
    }

    public function has_cap( $cap ) {
        $args = array_slice( func_get_args(), 1 );
        $args = array_merge( array( $cap ), $args );

        $caps = apply_filters( 'map_meta_cap', $args, $this->ID );

        // If we have no cap to check, assume it is fine.
        if ( empty( $caps ) ) {
            return true;
        }

        foreach ( (array) $caps as $cap ) {
            if ( ! isset( $this->allcaps[ $cap ] ) || true !== $this->allcaps[ $cap ] ) {
                return false;
            }
        }

        return true;
    }
}
?>

WP_User 类的关键属性：

$ID：用户的唯一标识符。
$user_login：用户的登录名。
$user_email：用户的邮箱地址。
$roles：用户的角色列表（例如，administrator, editor, author, contributor, subscriber）。
$caps：用户的权限列表（例如，edit_posts, delete_posts, manage_options）。

WP_User对象初始化时，会从数据库中加载用户的所有信息，包括角色和权限。角色和权限信息用于进行用户授权判断。 get_role_caps() 函数用于获取用户角色的权限。has_cap() 函数用于判断用户是否具有特定的权限。

五、代码示例：在插件中使用wp_get_current_user

假设我们正在开发一个插件，需要在插件中获取当前用户的用户名和邮箱地址：

<?php
/**
 * Plugin Name: 获取当前用户信息插件
 */

add_action( 'wp_footer', 'my_plugin_get_user_info' );

function my_plugin_get_user_info() {
    $current_user = wp_get_current_user();

    if ( $current_user->exists() ) {
        $username = $current_user->user_login;
        $email = $current_user->user_email;

        echo '<div style="border: 1px solid #ccc; padding: 10px;">';
        echo '<p>当前用户名：' . esc_html( $username ) . '</p>';
        echo '<p>当前用户邮箱：' . esc_html( $email ) . '</p>';
        echo '</div>';
    } else {
        echo '<div style="border: 1px solid #ccc; padding: 10px;">';
        echo '<p>您尚未登录</p>';
        echo '</div>';
    }
}
?>

这段代码的关键点：

wp_footer action hook： 将 my_plugin_get_user_info 函数绑定到 wp_footer action hook，以便在页面的底部输出用户信息。
wp_get_current_user()： 调用 wp_get_current_user() 函数获取当前用户对象。
$current_user->exists()： 使用 $current_user->exists() 方法检查用户是否已登录。如果用户未登录，$current_user 对象仍然存在，但 $current_user->ID 为 0，$current_user->exists() 返回 false。
esc_html()： 使用 esc_html() 函数对用户名和邮箱地址进行转义，以防止 XSS 攻击。

六、优化建议

缓存： WordPress已经对用户数据进行了缓存，但我们仍然可以根据实际情况，在我们的插件或主题中进行额外的缓存，以提高性能。
权限检查： 在访问用户数据之前，始终进行权限检查，以确保当前用户有权访问这些数据。
安全： 对用户输入进行验证和转义，以防止安全漏洞。

七、常见问题解答

为什么wp_get_current_user在某些情况下返回错误的用户信息？

这可能是由于缓存问题、cookie 问题或插件冲突导致的。尝试清除缓存、检查 cookie 设置、禁用插件进行排查。
如何强制刷新wp_get_current_user返回的用户信息？

可以使用wp_cache_delete( $user_id, 'users' )删除缓存，然后再次调用wp_get_current_user。
wp_get_current_user和get_userdata有什么区别？

wp_get_current_user 获取当前用户对象，而 get_userdata 根据用户 ID 获取用户对象。wp_get_current_user 内部会调用 get_userdata (实际上经过 get_user()函数) 。

八、总结：wp_get_current_user，用户认证与全局对象加载的关键

wp_get_current_user是WordPress中获取当前用户信息的基石。它通过cookie判断登录状态，利用get_user函数从数据库加载用户数据，最终返回一个WP_User对象，用于后续的权限验证和用户数据展示。理解这一过程对于开发安全、高效的WordPress插件和主题至关重要。

发表回复 取消回复

发表回复取消回复