Ticket #18536: 18536.4.diff

wp-includes/functions.php

@@ -3654 +3654 @@
 
         return $protocols;
 }
+
+/**
+ * Retrieve ids that are not already present in the cache
+ *
+ * @since 3.4.0
+ *
+ * @param array $object_ids ID list
+ * @param string $cache_key The cache bucket to check against
+ *
+ * @return array
+ */
+function _get_non_cached_ids( $object_ids, $cache_key ) {
+        $clean = array();
+        foreach ( $object_ids as $id ) {
+                $id = (int) $id;
+                if ( !wp_cache_get( $id, $cache_key ) ) {
+                        $clean[] = $id;
+                }
+        }
+
+        return $clean;
+}
+

wp-includes/pluggable.php

@@ -139 +139 @@
 function cache_users( $user_ids ) {
         global $wpdb;
 
-        $clean = array();
-        foreach ( $user_ids as $id ) {
-                $id = (int) $id;
-                if ( !wp_cache_get( $id, 'users' ) ) {
-                        $clean[] = $id;
-                }
-        }
+        $clean = _get_non_cached_ids( $user_ids, 'users' );
 
         if ( empty( $clean ) )
                 return;

wp-includes/query.php

@@ -1951 +1951 @@
                 $fields = '';
                 $post_status_join = false;
                 $page = 1;
+                $expand_ids = false;
 
                 if ( isset( $q['caller_get_posts'] ) ) {
                         _deprecated_argument( 'WP_Query', '3.1', __( '"caller_get_posts" is deprecated. Use "ignore_sticky_posts" instead.' ) );
@@ -2597 +2598 @@
                                 $$piece = isset( $clauses[ $piece ] ) ? $clauses[ $piece ] : '';
                 }
 
+                if ( "$wpdb->posts.*" == $fields ) {
+                        $fields = "$wpdb->posts.ID";
+                        $expand_ids = true;
+                }
+
                 if ( ! empty($groupby) )
                         $groupby = 'GROUP BY ' . $groupby;
                 if ( !empty( $orderby ) )
@@ -2626 +2632 @@
                         return $r;
                 }
 
-                $this->posts = $wpdb->get_results($this->request);
+                if ( $expand_ids ) {
+                        $ids = $wpdb->get_col( $this->request );
+
+                        if ( $ids ) {
+                                $this->set_found_posts( $q, $limits );
+
+                                $non_cached_ids = _get_non_cached_ids( $ids, 'posts' );
+
+                                if ( !empty( $non_cached_ids ) ) {
+                                        $fresh_posts = $wpdb->get_results( sprintf( "SELECT $wpdb->posts.* FROM $wpdb->posts WHERE ID IN (%s)", join( ",", $non_cached_ids ) ) );
+
+                                        update_post_caches($fresh_posts, $post_type, $q['update_post_term_cache'], $q['update_post_meta_cache']);
+                                }
+
+                                $this->posts = array_map( 'get_post', $ids );
+                        } else {
+                                $this->found_posts = $this->max_num_pages = 0;
+                                $this->posts = array();
+                        }
+                } else {
+                        $this->posts = $wpdb->get_results( $this->request );
+                        $this->set_found_posts( $q, $limits );
+                }
 
                 // Raw results filter. Prior to status checks.
                 if ( !$q['suppress_filters'] )
@@ -2645 +2673 @@
                         $this->comment_count = count($this->comments);
                 }
 
-                if ( !$q['no_found_rows'] && !empty($limits) ) {
-                        $found_posts_query = apply_filters_ref_array( 'found_posts_query', array( 'SELECT FOUND_ROWS()', &$this ) );
-                        $this->found_posts = $wpdb->get_var( $found_posts_query );
-                        $this->found_posts = apply_filters_ref_array( 'found_posts', array( $this->found_posts, &$this ) );
-                        $this->max_num_pages = ceil($this->found_posts / $q['posts_per_page']);
-                }
-
                 // Check post status to determine if post should be displayed.
                 if ( !empty($this->posts) && ($this->is_single || $this->is_page) ) {
                         $status = get_post_status($this->posts[0]);
@@ -2754 +2775 @@
                 return $this->posts;
         }
 
+        function set_found_posts( $q, $limits ) {
+                global $wpdb;
+
+                if ( $q['no_found_rows'] || empty( $limits ) )
+                        return;
+
+                $found_posts_query = apply_filters_ref_array( 'found_posts_query', array( 'SELECT FOUND_ROWS()', $this ) );
+                $this->found_posts = $wpdb->get_var( $found_posts_query );
+                $this->found_posts = apply_filters_ref_array( 'found_posts', array( $this->found_posts, $this ) );
+                $this->max_num_pages = ceil( $this->found_posts / $q['posts_per_page'] );
+        }
+
         /**
          * Set up the next post and iterate current post index.
          *