WooCommerce Rest API仅用于登录用户

Question

我想允许WooCommerce API仅登录用户，并为非登录用户显示一条消息。我该怎么办？

我找到了适用于WordPress REST API的代码。如何将其用于WooCommerce？

add_filter( 'rest_authentication_errors', function( $result ) {
    if ( ! empty( $result ) ) {
        return $result;
    }
    if ( ! is_user_logged_in() ) {
        return new WP_Error( 'rest_not_logged_in', 'You are not currently logged in.', array( 'status' => 401 ) );
    }
    return $result;
});