Upstream/WordPress
1
0
Fork 0
mirror of https://github.com/WordPress/WordPress.git synced 2024-11-13 22:26:06 +01:00
Code Issues Projects Releases Wiki Activity

Add keyed object output to wpdb::get_results. Fixes #5286 props andy.

Browse Source 
git-svn-id: http://svn.automattic.com/wordpress/trunk@7028 1a063a9b-81f0-0310-95a4-ce76da25c4cd
This commit is contained in:
westi 2008-02-25 22:00:27 +00:00
parent 2229af8cb6
commit 35a2e26b4d
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
wp-includes/wp-db.php
View File

@ -7,6 +7,7 @@
define('EZSQL_VERSION', 'WP1.25'); define('EZSQL_VERSION', 'WP1.25');
define('OBJECT', 'OBJECT', true); define('OBJECT', 'OBJECT', true);
define('OBJECT_K', 'OBJECT_K', false);
define('ARRAY_A', 'ARRAY_A', false); define('ARRAY_A', 'ARRAY_A', false);
define('ARRAY_N', 'ARRAY_N', false); define('ARRAY_N', 'ARRAY_N', false);
@ -396,7 +397,7 @@ class wpdb {
/** /**
* Return an entire result set from the database * Return an entire result set from the database
* @param string $query (can also be null to pull from the cache) * @param string $query (can also be null to pull from the cache)
* @param string $output ARRAY_A | ARRAY_N | OBJECT * @param string $output ARRAY_A | ARRAY_N | OBJECT_K | OBJECT
* @return mixed results * @return mixed results
*/ */
function get_results($query = null, $output = OBJECT) { function get_results($query = null, $output = OBJECT) {
@ -407,22 +408,33 @@ class wpdb {
else else
return null; return null;
// Send back array of objects. Each row is an object
if ( $output == OBJECT ) { if ( $output == OBJECT ) {
// Return an integer-keyed array of row objects
return $this->last_result; return $this->last_result;
} elseif ( $output == OBJECT_K ) {
// Return an array of row objects with keys from column 1
// (Duplicates are discarded)
foreach ( $this->last_result as $row ) {
$key = array_shift( get_object_vars( $row ) );
if ( !isset( $new_array[ $key ] ) )
$new_array[ $key ] = $row;
}
return $new_array;
} elseif ( $output == ARRAY_A || $output == ARRAY_N ) { } elseif ( $output == ARRAY_A || $output == ARRAY_N ) {
// Return an integer-keyed array of...
if ( $this->last_result ) { if ( $this->last_result ) {
$i = 0; $i = 0;
foreach( $this->last_result as $row ) { foreach( $this->last_result as $row ) {
$new_array[$i] = (array) $row;
if ( $output == ARRAY_N ) { if ( $output == ARRAY_N ) {
$new_array[$i] = array_values($new_array[$i]); // ...integer-keyed row arrays
$new_array[$i] = array_values( get_object_vars( $row ) );
} else {
// ...column name-keyed row arrays
$new_array[$i] = get_object_vars( $row );
} }
$i++; ++$i;
} }
return $new_array; return $new_array;
} else {
return null;
} }
} }
} }