WooCommerce下单后自动新增用户角色实现及现有代码不生效排查

问题排查及修复方案 #

你提供的代码存在以下几个常见疏漏： #

• 触发时机不匹配：你使用的woocommerce_order_status_completed钩子仅在订单标记为「已完成」时执行，若你需要下单后立即触发逻辑，可更换为woocommerce_thankyou（订单提交成功跳转到感谢页触发）或woocommerce_order_status_processing（订单进入处理中状态时触发）。
• 高版本WooCommerce兼容问题：WooCommerce 3.0+版本已经废弃了直接访问订单、订单项属性的写法，$order->user_id、$item['product_id']这类写法会取不到对应值，导致后续判断不生效，必须使用官方提供的get_*系列方法取值。
• 缺少必要的校验逻辑：没有校验订单、用户实例是否有效，也没有判断用户是否已经添加过目标角色，可能会出现重复操作、PHP报错的问题。
• 额外提示：如果你只是要给用户叠加额外角色而非替换原有角色，可以直接删除remove_role('customer')代码，WordPress原生支持一个用户绑定多个角色。

修正后可直接使用的代码： #

// 可根据你的实际触发需求更换下方钩子
add_action( 'woocommerce_thankyou', 'change_role_on_purchase' );
function change_role_on_purchase( $order_id ) {
    // 校验订单是否有效
    $order = wc_get_order( $order_id );
    if( !$order ) return;

    // 校验下单用户是否为注册用户
    $user_id = $order->get_user_id();
    if( !$user_id ) return;

    $target_product_id = 73; // 替换为你的目标产品ID
    $order_has_target_product = false;

    // 遍历订单商品判断是否包含目标产品
    foreach ( $order->get_items() as $item ) {
        if( $item->get_product_id() === $target_product_id ) {
            $order_has_target_product = true;
            break;
        }
    }

    if( $order_has_target_product ) {
        $user = get_user_by( 'id', $user_id );
        // 校验用户有效且未拥有premium角色，避免重复添加
        if( $user && !in_array( 'premium', (array)$user->roles, true ) ) {
            // 不需要删除customer角色可注释掉下一行
            $user->remove_role( 'customer' );
            $user->add_role( 'premium' );
        }
    }
}

内容的提问来源于stack exchange，提问作者B.M. Rafiul Alam