ProductCountCacheService.php
<?php
declare( strict_types=1 );
namespace Automattic\WooCommerce\Caches;
use Automattic\WooCommerce\Internal\Utilities\ProductUtil;
use WC_Product;
use WP_Post;
/**
* A service class to help with updates to the aggregate product counts cache.
*
* @internal
*/
class ProductCountCacheService {
public const BACKGROUND_EVENT_HOOK = 'woocommerce_refresh_product_count_cache';
/**
* ProductCountCache instance.
*
* @var ProductCountCache
*/
private ProductCountCache $product_count_cache;
/**
* Array of product IDs with their last transitioned status as key value pairs.
* Guarantees idempotency for product status transitions when multiple hooks fire for the same product.
*
* @var array<int,string>
*/
private array $product_statuses = array();
/**
* Array of product IDs with their initial status as key value pairs.
* Guarantees idempotency for product status transitions when multiple hooks fire for the same product.
*
* @var array<int,string>
*/
private array $initial_product_statuses = array();
/**
* Set of product IDs currently being created in this request (keyed by ID; detected via old_status='new').
*
* @var array<int,true>
*/
private array $products_in_creation = array();
/**
* Class initialization, invoked by the DI container.
*
* @internal
*/
final public function init(): void {
$this->product_count_cache = new ProductCountCache();
add_action( 'action_scheduler_ensure_recurring_actions', array( $this, 'unschedule_background_actions' ) );
add_action( self::BACKGROUND_EVENT_HOOK, array( $this, 'prime_cache_if_cold' ) );
if ( defined( 'WC_PLUGIN_BASENAME' ) ) {
add_action( 'deactivate_' . WC_PLUGIN_BASENAME, array( $this, 'unschedule_background_actions' ) );
}
// Until persistent counters reactivated, disable callbacks for woocommerce_new_product, transition_post_status, before_delete_post hooks.
}
/**
* Primes the product count cache for a given post type when it is cold.
*
* @param string $product_type The product post type.
* @return void
*/
public function prime_cache_if_cold( string $product_type = 'product' ): void {
// Until persistent counters reactivated, this task is no-op.
}
/**
* Register background caching for each product type.
*
* @return void
*/
public function schedule_background_actions(): void {
$frequency = HOUR_IN_SECONDS * 12;
$timestamp = time() + $frequency;
as_schedule_recurring_action( $timestamp, $frequency, self::BACKGROUND_EVENT_HOOK, array( 'product' ), 'count', true );
}
/**
* Unschedules background actions.
*
* @return void
*/
public function unschedule_background_actions(): void {
WC()->queue()->cancel_all( self::BACKGROUND_EVENT_HOOK );
}
/**
* Update the cache when a new product is created.
*
* @param int $product_id Product ID.
* @param WC_Product $product The product.
* @return void
*/
public function update_on_new_product( int $product_id, WC_Product $product ): void {
// transition_post_status already counted this product — reverse any errant decrement from a cold step 1 and stop.
// In-memory status may diverge from DB after a mid-creation wp_update_post; do not increment here.
if ( isset( $this->product_statuses[ $product_id ] ) ) {
$this->maybe_restore_initial_status_count( $product_id );
unset( $this->products_in_creation[ $product_id ] );
return;
}
// Cache was cold throughout creation — transition_post_status never fired; use in-memory status as the sole count.
$product_status = $product->get_status();
if ( $this->product_count_cache->is_cached( 'product', $product_status ) ) {
$this->product_statuses[ $product_id ] = $product_status;
$this->product_count_cache->increment( 'product', $product_status );
}
unset( $this->products_in_creation[ $product_id ] );
}
/**
* Update the cache whenever a product status changes.
*
* @param string $new_status The new post status.
* @param string $old_status The previous post status.
* @param WP_Post $post The post object.
*
* @return void
*/
public function update_on_product_status_changed( string $new_status, string $old_status, WP_Post $post ): void {
if ( 'product' !== $post->post_type ) {
return;
}
$product_id = $post->ID;
// WordPress uses 'new' as old_status exclusively on the first transition_post_status of a newly inserted post.
if ( 'new' === $old_status ) {
$this->products_in_creation[ $product_id ] = true;
}
$is_new_cached = $this->product_count_cache->is_cached( 'product', $new_status );
$is_old_cached = $this->product_count_cache->is_cached( 'product', $old_status );
if ( ! $is_new_cached && ! $is_old_cached ) {
return;
}
// If the status count has already been incremented for this product, skip.
if ( ( $this->product_statuses[ $product_id ] ?? null ) === $new_status ) {
return;
}
$previously_tracked = isset( $this->product_statuses[ $product_id ] );
$this->product_statuses[ $product_id ] = $new_status;
$was_decremented = $is_old_cached && false !== $this->product_count_cache->decrement( 'product', $old_status );
if ( $is_new_cached ) {
$this->product_count_cache->increment( 'product', $new_status );
}
// Record old status for creation-time correction only; existing-product decrements are correct and must not be reversed.
// If $previously_tracked, an earlier transition already counted the old status — decrement is legitimate.
if ( ! $previously_tracked && $was_decremented && ! isset( $this->initial_product_statuses[ $product_id ] ) && isset( $this->products_in_creation[ $product_id ] ) ) {
$this->initial_product_statuses[ $product_id ] = $old_status;
} elseif ( ( $this->initial_product_statuses[ $product_id ] ?? null ) === $new_status ) {
unset( $this->initial_product_statuses[ $product_id ] );
}
}
/**
* Update the cache when a product is permanently deleted.
*
* @param int $post_id Post ID.
* @param WP_Post $post The post object.
*
* @return void
*/
public function update_on_product_deleted( int $post_id, WP_Post $post ): void {
if ( 'product' === $post->post_type ) {
// Reverse any errant decrement from a mid-creation status transition that update_on_new_product will never get to correct.
$this->maybe_restore_initial_status_count( $post_id );
$product_status = $post->post_status;
if ( $this->product_count_cache->is_cached( 'product', $product_status ) ) {
$this->product_count_cache->decrement( 'product', $product_status );
}
unset( $this->product_statuses[ $post_id ], $this->products_in_creation[ $post_id ] );
}
}
/**
* Reverses an errant decrement recorded in initial_product_statuses for a given product, if any.
*
* @param int $product_id Product ID.
*
* @return void
*/
private function maybe_restore_initial_status_count( int $product_id ): void {
if ( isset( $this->initial_product_statuses[ $product_id ] ) ) {
$initial_status = $this->initial_product_statuses[ $product_id ];
unset( $this->initial_product_statuses[ $product_id ] );
if ( $this->product_count_cache->is_cached( 'product', $initial_status ) ) {
$this->product_count_cache->increment( 'product', $initial_status );
}
}
}
}
