UserArray.php

Go to the documentation of this file.

<?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 ) {
        return new UserArrayFromResult( $res );
    }
}

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;
    }
}