Prime caches

Author: mikejolley

includes/class-wc-product-tables-query.php

@@ -226,22 +226,100 @@ public function custom_price_filter_post_clauses( $args, $wp_query ) {
 	 * @return array
 	 */
 	public function prime_product_table_caches( $posts ) {
-		$prime_ids = array();
+		$prime_ids     = array();
+		$variation_ids = array();
 
 		foreach ( $posts as $post ) {
 			if ( ! in_array( $post->post_type, array( 'product', 'product_variation' ), true ) ) {
 				continue;
 			}
 			$prime_ids[] = (int) $post->ID;
+
+			if ( 'product_variation' === $post->post_type ) {
+				$variation_ids[] = (int) $post->ID;
+			}
 		}
 
 		if ( ! empty( $prime_ids ) ) {
 			global $wpdb;
 
-			$products = $wpdb->get_results( 'SELECT * FROM ' . $wpdb->prefix . 'wc_products WHERE product_id IN (' . implode( ',', $prime_ids ) . ');' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+			$prime_id_sql             = implode( ',', $prime_ids );
+			$prime_non_variations_sql = implode( ',', array_diff( $prime_ids, $variation_ids ) );
+			$prime_variations_sql     = implode( ',', $variation_ids );
+
+			// Prime product rows.
+			$rows = $wpdb->get_results( 'SELECT * FROM ' . $wpdb->prefix . 'wc_products WHERE product_id IN (' . $prime_id_sql . ');' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+
+			foreach ( $rows as $row ) {
+				wp_cache_set( 'woocommerce_product_' . $row->product_id, $row, 'product' );
+			}
+
+			// Prime attribute rows.
+			if ( $prime_non_variations_sql ) {
+				$rows  = $wpdb->get_results( 'SELECT * FROM ' . $wpdb->prefix . 'wc_product_attributes WHERE product_id IN (' . $prime_non_variations_sql . ');' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+				$cache = array_fill_keys( $prime_ids, array() );
+
+				foreach ( $rows as $row ) {
+					$cache[ $row->product_id ][] = $row;
+				}
+
+				foreach ( $prime_ids as $id ) {
+					wp_cache_set( 'woocommerce_product_attributes_' . $id, $cache[ $id ], 'product' );
+				}
+			}
+
+			// Prime attribute values.
+			$rows  = $wpdb->get_results( '
+				SELECT `product_attribute_id`, `value`, `is_default`
+				FROM ' . $wpdb->prefix . 'wc_product_attribute_values
+				WHERE product_id IN (' . $prime_non_variations_sql . ');
+			' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+			$cache = array_fill_keys( $prime_ids, array() );
+
+			foreach ( $rows as $row ) {
+				$cache[ $row->product_attribute_id ][] = $row;
+			}
+
+			foreach ( $prime_ids as $id ) {
+				wp_cache_set( 'woocommerce_product_attribute_values_' . $id, $cache[ $id ], 'product' );
+			}
+
+			// Prime variation attribute values.
+			if ( $prime_variations_sql ) {
+				$rows  = $wpdb->get_results( 'SELECT * FROM ' . $wpdb->prefix . 'wc_product_variation_attribute_values WHERE product_id IN (' . $prime_variations_sql . ');' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+				$cache = array_fill_keys( $prime_ids, array() );
+
+				foreach ( $rows as $row ) {
+					$cache[ $row->product_id ][] = $row;
+				}
+
+				foreach ( $prime_ids as $id ) {
+					wp_cache_set( 'woocommerce_product_variation_attribute_values_' . $id, $cache[ $id ], 'product' );
+				}
+			}
+
+			// Prime relationships.
+			$rows  = $wpdb->get_results( 'SELECT `product_id`, `relationship_id`, `object_id`, `type` FROM ' . $wpdb->prefix . 'wc_product_relationships WHERE product_id IN (' . $prime_id_sql . ') ORDER BY `priority` ASC;' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+			$cache = array_fill_keys( $prime_ids, array() );
+
+			foreach ( $rows as $row ) {
+				$cache[ $row->product_id ][] = $row;
+			}
+
+			foreach ( $prime_ids as $id ) {
+				wp_cache_set( 'woocommerce_product_relationships_' . $id, $cache[ $id ], 'product' );
+			}
+
+			// Prime downloads.
+			$rows  = $wpdb->get_results( 'SELECT `product_id`, `download_id`, `name`, `file`, `priority` FROM ' . $wpdb->prefix . 'wc_product_downloads WHERE product_id IN (' . $prime_id_sql . ') ORDER BY `priority` ASC;' ); // WPCS: db call ok, cache ok, unprepared SQL OK.
+			$cache = array_fill_keys( $prime_ids, array() );
+
+			foreach ( $rows as $row ) {
+				$cache[ $row->product_id ][] = $row;
+			}
 
-			foreach ( $products as $product ) {
-				wp_cache_set( 'woocommerce_product_' . $product->product_id, $product, 'product' );
+			foreach ( $prime_ids as $id ) {
+				wp_cache_set( 'woocommerce_product_downloads_' . $id, $cache[ $id ], 'product' );
 			}
 		}
 

includes/data-stores/class-wc-product-variation-data-store-custom-table.php

@@ -331,7 +331,7 @@ public function update( &$product ) {
 	public function read_attributes( &$product ) {
 		global $wpdb;
 
-		$product_attributes = wp_cache_get( 'woocommerce_product_attributes_' . $product->get_id(), 'product' );
+		$product_attributes = wp_cache_get( 'woocommerce_product_variation_attribute_values_' . $product->get_id(), 'product' );
 
 		if ( false === $product_attributes ) {
 			$product_attributes = $wpdb->get_results(
@@ -341,7 +341,7 @@ public function read_attributes( &$product ) {
 				)
 			); // WPCS: db call ok, cache ok.
 
-			wp_cache_set( 'woocommerce_product_attributes_' . $product->get_id(), $product_attributes, 'product' );
+			wp_cache_set( 'woocommerce_product_variation_attribute_values_' . $product->get_id(), $product_attributes, 'product' );
 		}
 
 		if ( ! empty( $product_attributes ) ) {
@@ -669,6 +669,7 @@ protected function update_attributes( &$product, $force = false ) {
 	 */
 	protected function clear_caches( &$product ) {
 		wp_cache_delete( 'woocommerce_product_children_stock_status_' . $product->get_parent_id(), 'product' );
+		wp_cache_delete( 'woocommerce_product_variation_attribute_values_' . $product->get_id(), 'product' );
 		parent::clear_caches( $product );
 	}
 }