Ticket #38280: 38280.11.diff

src/wp-includes/taxonomy.php

@@ -3091 +3091 @@
         return true;
 }
 
+/**
+ * Retrieves the term count for a specific object type.
+ *
+ * @since 5.0.0
+ *
+ * @param int    $term_id     Term ID.
+ * @param string $taxonomy    Taxonomy name.
+ * @param string $object_type Object type.
+ *
+ * @return WP_Error|bool|int WP_Error if invalid taxonomy is passed, false if
+ *                           object is not in taxonomy, object term count otherwise.
+ */
+function wp_get_term_count_for_object_type( $term_id, $taxonomy, $object_type ) {
+        if ( ! taxonomy_exists( $taxonomy ) ) {
+                return new WP_Error( 'invalid_taxonomy', __( 'Invalid taxonomy.' ) );
+        }
+
+        if ( ! is_object_in_taxonomy( $object_type, $taxonomy ) ) {
+                return new WP_Error(
+                        'invalid_object_type',
+                        __( 'Object type is not in taxonomy.' ),
+                        array(
+                                'object_type' => $object_type,
+                                'taxonomy'    => $taxonomy,
+                        )
+                );
+        }
+
+        $term = get_term( $term_id, $taxonomy );
+
+        if ( is_wp_error( $term ) ) {
+                return $term;
+        }
+
+        if ( 0 === $term->count ) {
+                return 0;
+        }
+
+        $taxonomy_object = get_taxonomy( $taxonomy );
+
+        if ( 1 >= count( $taxonomy_object->object_type ) ) {
+                return $term->count;
+        }
+
+        $term_object_count = get_term_meta( $term_id, '_wp_object_count_' . $object_type, true );
+        if ( $term_object_count ) {
+                return (int) $term_object_count;
+        }
+
+        $counted_object_types = (array) get_term_meta( $term_id, '_wp_counted_object_types', true );
+
+        // When the object type is marked counted and no meta key exists, the count is 0.
+        if ( in_array( $object_type, $counted_object_types, true ) ) {
+                return 0;
+        }
+
+        // Term has not been counted for the object type. Calculate and fetch count.
+        wp_update_term_count_now( array( $term->term_taxonomy_id ), $taxonomy );
+        return wp_get_term_count_for_object_type( $term_id, $taxonomy, $object_type );
+}
+
 //
 // Cache
 //
@@ -3605 +3666 @@
  *
  * @access private
  * @since 2.3.0
+ * @since 5.0.0 Store term counts on a per object type basis in meta.
  *
  * @global wpdb $wpdb WordPress database abstraction object.
  *
@@ -3628 +3690 @@
         }
 
         if ( $object_types ) {
-                $object_types = esc_sql( array_filter( $object_types, 'post_type_exists' ) );
+                $object_types = array_filter( $object_types, 'post_type_exists' );
         }
 
         foreach ( (array) $terms as $term ) {
                 $count = 0;
 
+                $term_id = (int) $wpdb->get_var( $wpdb->prepare( "SELECT term_id FROM $wpdb->term_taxonomy WHERE term_taxonomy_id = %d", $term ) );
+
+                // Remove previous counts to prevent stale data if an object type is removed from a taxonomy.
+                $counted_object_types = (array) get_term_meta( $term_id, '_wp_counted_object_types', true );
+
+                foreach ( $counted_object_types as $o_type ) {
+                        delete_term_meta( $term_id, '_wp_object_count_' . $o_type );
+                }
+
+                delete_term_meta( $term_id, '_wp_counted_object_types' );
+
+                $term_count_meta = array();
+
+                if ( $object_types ) {
+                        foreach ( $object_types as $type ) {
+                                $current_count = (int) $wpdb->get_var( $wpdb->prepare( "SELECT COUNT(*) FROM $wpdb->term_relationships, $wpdb->posts WHERE $wpdb->posts.ID = $wpdb->term_relationships.object_id AND post_status = 'publish' AND post_type = %s AND term_taxonomy_id = %d", $type, $term ) );
+
+                                $count += $current_count;
+
+                                $term_count_meta[ $type ] = $current_count;
+                        }
+                }
+
                 // Attachments can be 'inherit' status, we need to base count off the parent's status if so.
                 if ( $check_attachments ) {
-                        $count += (int) $wpdb->get_var( $wpdb->prepare( "SELECT COUNT(*) FROM $wpdb->term_relationships, $wpdb->posts p1 WHERE p1.ID = $wpdb->term_relationships.object_id AND ( post_status = 'publish' OR ( post_status = 'inherit' AND post_parent > 0 AND ( SELECT post_status FROM $wpdb->posts WHERE ID = p1.post_parent ) = 'publish' ) ) AND post_type = 'attachment' AND term_taxonomy_id = %d", $term ) );
+                        $attachment_count = (int) $wpdb->get_var( $wpdb->prepare( "SELECT COUNT(*) FROM $wpdb->term_relationships, $wpdb->posts p1 WHERE p1.ID = $wpdb->term_relationships.object_id AND ( post_status = 'publish' OR ( post_status = 'inherit' AND post_parent > 0 AND ( SELECT post_status FROM $wpdb->posts WHERE ID = p1.post_parent ) = 'publish' ) ) AND post_type = 'attachment' AND term_taxonomy_id = %d", $term ) );
+
+                        $count += $attachment_count;
+
+                        $term_count_meta['attachment'] = $attachment_count;
+
+                        // Re-add attachment so the meta gets saved below.
+                        $object_types[] = 'attachment';
                 }
 
-                if ( $object_types ) {
-                        $count += (int) $wpdb->get_var( $wpdb->prepare( "SELECT COUNT(*) FROM $wpdb->term_relationships, $wpdb->posts WHERE $wpdb->posts.ID = $wpdb->term_relationships.object_id AND post_status = 'publish' AND post_type IN ('" . implode( "', '", $object_types ) . "') AND term_taxonomy_id = %d", $term ) );
+                // Save individual counts for each object type in term meta.
+                if ( 1 < count( $term_count_meta ) ) {
+                        foreach ( $object_types as $type ) {
+                                if ( ! empty( $term_count_meta[ $type ] ) ) {
+                                        update_term_meta( $term_id, '_wp_object_count_' . $type, (int) $term_count_meta[ $type ] );
+                                }
+                        }
+
+                        update_term_meta( $term_id, '_wp_counted_object_types', $object_types );
                 }
 
                 /** This action is documented in wp-includes/taxonomy.php */