wiki.techinc.nl/includes/UserArray.php

78 lines
1.6 KiB
PHP
Raw Normal View History

<?php
abstract class UserArray implements Iterator {
static function newFromResult( $res ) {
$userArray = null;
if ( !wfRunHooks( 'UserArrayFromResult', array( &$userArray, $res ) ) ) {
return null;
}
if ( $userArray === null ) {
$userArray = self::newFromResult_internal( $res );
}
return $userArray;
}
static function newFromIDs( $ids ) {
$ids = array_map( 'intval', (array)$ids ); // paranoia
if ( !$ids )
// Database::select() doesn't like empty arrays
return new ArrayIterator(array());
$dbr = wfGetDB( DB_SLAVE );
$res = $dbr->select( 'user', '*', array( 'user_id' => $ids ),
__METHOD__ );
return self::newFromResult( $res );
}
protected static function newFromResult_internal( $res ) {
$userArray = new UserArrayFromResult( $res );
return $userArray;
}
}
class UserArrayFromResult extends UserArray {
var $res;
var $key, $current;
function __construct( $res ) {
$this->res = $res;
$this->key = 0;
$this->setCurrent( $this->res->current() );
}
protected function setCurrent( $row ) {
if ( $row === false ) {
$this->current = false;
} else {
$this->current = User::newFromRow( $row );
}
}
public function count() {
return $this->res->numRows();
}
function current() {
return $this->current;
}
function key() {
return $this->key;
}
function next() {
$row = $this->res->next();
$this->setCurrent( $row );
$this->key++;
}
function rewind() {
$this->res->rewind();
$this->key = 0;
$this->setCurrent( $this->res->current() );
}
function valid() {
return $this->current !== false;
}
}