<?php
// This file is part of Moodle - http://moodle.org/
//
// Moodle is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Moodle is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Moodle. If not, see <http://www.gnu.org/licenses/>.
/**
* Applies the same callback to all recorset records.
*
* @since Moodle 2.9
* @package core
* @category dml
* @copyright 2015 David Monllao
* @license http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
*/
namespace core\dml;
defined('MOODLE_INTERNAL') || die();
/**
* Iterator that walks through a moodle_recordset applying the provided function.
*
* The internal recordset can be closed using the close() function.
*
* Note that consumers of this class are responsible of closing the recordset,
* although there are some implicit closes under some ciscumstances:
* - Once all recordset records have been iterated
* - The object is destroyed
*
* @since Moodle 2.9
* @package core
* @category dml
* @copyright 2015 David Monllao
* @license http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
*/
class recordset_walk implements \Iterator {
/**
* @var \moodle_recordset The recordset.
*/
protected $recordset;
/**
* @var callable The callback.
*/
protected $callback;
/**
* @var mixed|null Extra param for the callback.
*/
protected $callbackextra;
/**
* Create a new iterator applying the callback to each record.
*
* @param \moodle_recordset $recordset Recordset to iterate.
* @param callable $callback Apply this function to each record. If using a method, it should be public.
* @param mixed $callbackextra An extra single parameter to pass to the callback. Use a container to pass multiple values.
*/
public function __construct(\moodle_recordset $recordset, callable $callback, $callbackextra = null) {
$this->recordset = $recordset;
$this->callback = $callback;
$this->callbackextra = $callbackextra;
}
/**
* Closes the recordset.
*
* @return void
*/
public function __destruct() {
$this->close();
}
/**
* Returns the current element after applying the callback.
*
* @return mixed|bool The returned value type will depend on the callback.
*/
> #[\ReturnTypeWillChange]
public function current() {
if (!$this->recordset->valid()) {
return false;
}
if (!$record = $this->recordset->current()) {
return false;
}
// Apply callback and return.
if (!is_null($this->callbackextra)) {
return call_user_func($this->callback, $record, $this->callbackextra);
} else {
return call_user_func($this->callback, $record);
}
}
/**
* Moves the internal pointer to the next record.
*
* @return void
*/
< public function next() {
< return $this->recordset->next();
> public function next(): void {
> $this->recordset->next();
}
/**
* Returns current record key.
*
* @return int
*/
> #[\ReturnTypeWillChange]
public function key() {
return $this->recordset->key();
}
/**
* Returns whether the current position is valid or not.
*
* If we reached the end of the recordset we close as we
* don't allow rewinds. Doing do so we reduce the chance
* of unclosed recordsets.
*
* @return bool
*/
< public function valid() {
> public function valid(): bool {
if (!$valid = $this->recordset->valid()) {
$this->close();
}
return $valid;
}
/**
* Rewind is not supported.
*
* @return void
*/
< public function rewind() {
> public function rewind(): void {
// No rewind as it is not implemented in moodle_recordset.
return;
}
/**
* Closes the recordset.
*
* @return void
*/
public function close() {
$this->recordset->close();
}
}