WooCommerce 根据用户角色隐藏指定分类的产品

function wholeseller_role_cat( $query ) {

    // Get the current user

    $current_user = wp_get_current_user();

    if ( $query->is_main_query() ) {

        // Displaying only "vip" category products to "whole seller" user role

        if ( in_array( 'vip', $current_user->roles ) ) {

            // Set here the ID for vip category 

            $query->set( 'cat', '123' ); 

        // Displaying All products (except "123" category products) 

        // to all other users roles (except "vip" user role)

        // and to non logged user.

        } else {

            // Set here the ID for vip category (with minus sign before)

            $query->set( 'cat', '-123' ); // negative number

        }

    }

}

add_action( 'pre_get_posts', 'wholeseller_role_cat' );

function wholeseller_role_cat( $q ) {

    // Get the current user

    $current_user = wp_get_current_user();

    // Displaying only "123" category products to "whole seller" user role

    if ( in_array( 'vip', $current_user->roles ) ) {

        // Set here the ID for vip category 

        $q->set( 'tax_query', array(

            array(

                'taxonomy' => 'product_cat',

                'field' => 'term_id',

                'terms' => '123', // your category ID

            )

        ) ); 

    // Displaying All products (except "123" category products) 

    // to all other users roles (except "vip" user role)

    // and to non logged user.

    } else {

        // Set here the ID for vip category

        $q->set( 'tax_query', array(

            array(

                'taxonomy' => 'product_cat',

                'field' => 'term_id',

                'terms' => '123', // your category ID

                'operator' => 'NOT IN'

            )

        ) ); 

    }

}

add_action( 'woocommerce_product_query', 'wholeseller_role_cat' );

            array(

                'taxonomy' => 'product_cat',

                'field' => 'slug',

                'terms' => 'array( 'PUT YOUR CATEGORY HERE' ),

                // your category slug (to use the slug see below)