From 1c9a241148e5586fbf19df5b70647e1dcf5f7fd3 Mon Sep 17 00:00:00 2001 From: scribu Date: Tue, 26 Oct 2010 19:01:55 +0000 Subject: [PATCH] Introduce 'fields' query var to WP_Query. See #14777 git-svn-id: https://develop.svn.wordpress.org/trunk@15982 602fd350-edb4-49c9-b593-d223f7449a82 --- wp-includes/query.php | 36 +++++++++++++++++++++++++++++++++--- 1 file changed, 33 insertions(+), 3 deletions(-) diff --git a/wp-includes/query.php b/wp-includes/query.php index c265a5008c..821bd146a6 100644 --- a/wp-includes/query.php +++ b/wp-includes/query.php @@ -1179,10 +1179,11 @@ class WP_Query extends WP_Object_Query { , 'preview' , 's' , 'sentence' + , 'fields' ); foreach ( $keys as $key ) { - if ( !isset($array[$key])) + if ( !isset($array[$key]) ) $array[$key] = ''; } @@ -1190,7 +1191,7 @@ class WP_Query extends WP_Object_Query { 'tag__in', 'tag__not_in', 'tag__and', 'tag_slug__in', 'tag_slug__and'); foreach ( $array_keys as $key ) { - if ( !isset($array[$key])) + if ( !isset($array[$key]) ) $array[$key] = array(); } return $array; @@ -1640,7 +1641,7 @@ class WP_Query extends WP_Object_Query { $join = ''; $search = ''; $groupby = ''; - $fields = "$wpdb->posts.*"; + $fields = ''; $post_status_join = false; $page = 1; @@ -1721,6 +1722,17 @@ class WP_Query extends WP_Object_Query { else $q['no_found_rows'] = false; + switch ( $q['fields'] ) { + case 'ids': + $fields = "$wpdb->posts.ID"; + break; + case 'id=>parent': + $fields = "$wpdb->posts.ID, $wpdb->posts.post_parent"; + break; + default: + $fields = "$wpdb->posts.*"; + } + // If a month is specified in the querystring, load that month if ( $q['m'] ) { $q['m'] = '' . preg_replace('|[^0-9]|', '', $q['m']); @@ -2257,6 +2269,7 @@ class WP_Query extends WP_Object_Query { $groupby = 'GROUP BY ' . $groupby; if ( !empty( $orderby ) ) $orderby = 'ORDER BY ' . $orderby; + $found_rows = ''; if ( !$q['no_found_rows'] && !empty($limits) ) $found_rows = 'SQL_CALC_FOUND_ROWS'; @@ -2265,7 +2278,24 @@ class WP_Query extends WP_Object_Query { if ( !$q['suppress_filters'] ) $this->request = apply_filters_ref_array('posts_request', array( $this->request, &$this ) ); + if ( 'ids' == $q['fields'] ) { + $this->posts = $wpdb->get_col($this->request); + + return $this->posts; + } + + if ( 'id=>parent' == $q['fields'] ) { + $this->posts = $wpdb->get_results($this->request); + + $r = array(); + foreach ( $this->posts as $post ) + $r[ $post->ID ] = $post->post_parent; + + return $r; + } + $this->posts = $wpdb->get_results($this->request); + // Raw results filter. Prior to status checks. if ( !$q['suppress_filters'] ) $this->posts = apply_filters_ref_array('posts_results', array( $this->posts, &$this ) );