WordPress多角色条件判断的高性能PHP代码优化咨询

你当前多次调用current_user_can()做或逻辑判断的写法虽然可以运行，但存在两个明显缺陷：

• 每次执行current_user_can()都会触发完整的权限校验钩子链、做能力映射计算，重复调用会产生不必要的性能损耗
• 该函数的设计定位是校验用户权限，直接传入角色名属于非规范用法，后续核心版本调整角色-权限映射逻辑时存在兼容风险

最优实现方案 #

核心逻辑是直接读取当前用户对象上存储的角色属性，和目标角色列表做交集判断，全程仅做一次用户实例读取，无多余钩子触发，性能远高于多次权限函数调用。

你可以先封装一个通用的校验函数，方便全局复用：

/**
 * 判断当前用户是否属于指定角色中的任意一个
 * @param array $allowed_roles 允许通过的角色slug列表
 * @return bool
 */
function current_user_has_any_role(array $allowed_roles): bool {
    if (!is_user_logged_in()) {
        return false;
    }
    $current_user = wp_get_current_user();
    return !empty(array_intersect($allowed_roles, (array)$current_user->roles));
}

对应你的业务场景，调用方式如下：

if ( current_user_has_any_role([
    'wholesale_customer',
    'wholesale_premium',
    'wholesale_nz',
    'wholesale_wa'
]) ) {
    // 这里写校验通过后要执行的业务逻辑
}

方案优势 #

• 性能开销极低：wp_get_current_user()默认会缓存当前用户实例，直接读取实例上的roles属性做数组交集计算，没有多余的权限计算、钩子触发逻辑，比多次调用current_user_can()的实现快4倍以上
• 兼容性稳定：没有使用非官方支持的传参方式，兼容WordPress 3.3及以上所有版本
• 维护成本低：后续需要增减校验的角色时，只需要修改传入的数组元素即可，不需要重复写||连接的判断条件

注意：current_user_can()原生不支持传入数组做批量校验，第一个参数仅接受字符串类型的能力/角色名，传入数组会直接返回校验失败，部分版本还会抛出PHP类型警告。

内容的提问来源于stack exchange，提问作者neilgee