如何修复WooCommerce代码，使其精准定位批发客户角色？

算力阿峰

2026-8-24

修复批发客户角色识别及数量输入设置问题

问题分析

你的代码存在两个核心问题：

第一个函数中判断用户角色的if语句未闭合，导致逻辑执行异常
变量产品的数量限制函数未添加批发客户角色判断，会对所有用户生效

另外需要确认当前登录用户确实被分配了wholesale_customer角色，避免因角色名称错误或未分配导致的判断失效。

修复后的完整代码

/**
 * Adjust the quantity input values for simple products
 */
add_filter( 'woocommerce_quantity_input_args', 'jk_woocommerce_quantity_input_args', 10, 2 );
function jk_woocommerce_quantity_input_args( $args, $product ) {
    // 获取当前登录用户
    $user = wp_get_current_user();
    
    // 仅对批发客户生效
    if ( in_array( 'wholesale_customer', $user->roles ) ) {
        if ( is_singular( 'product' ) ) {
            $args['input_value'] = 2; // 产品页面起始数量
        }
        $args['max_value'] = 80; // 最大数量
        $args['min_value'] = 2; // 最小数量
        $args['step']      = 2; // 数量步长
    }
    
    return $args;
}

/**
 * Adjust the quantity input values for variable products
 */
add_filter( 'woocommerce_available_variation', 'jk_woocommerce_available_variation' );
function jk_woocommerce_available_variation( $args ) {
    // 获取当前登录用户
    $user = wp_get_current_user();
    
    // 仅对批发客户生效
    if ( in_array( 'wholesale_customer', $user->roles ) ) {
        $args['max_qty'] = 80; // 变量产品最大数量
        $args['min_qty'] = 2; // 变量产品最小数量
    }
    
    return $args;
}