php.lab

⌂ Home ▲ Previous: First class callable syntax ▼ Next: Fibers

Generators

Availability

PHP 5 >= 5.5.0, PHP 7, PHP 8

Definition

In computer science, a generator is a routine that can be used to control the iteration behaviour of a loop. All generators are also iterators. A generator is very similar to a function that returns an array, in that a generator has parameters, can be called, and generates a sequence of values. However, instead of building an array containing all the values and returning them all at once, a generator yields the values one at a time, which requires less memory and allows the caller to get started processing the first few values immediately. In short, a generator looks like a function but behaves like an iterator.

Generators can be implemented in terms of more expressive control flow constructs, such as coroutines or first-class continuations.Generators, also known as semicoroutines, are a special case of (and weaker than) coroutines, in that they always yield control back to the caller (when passing a value back), rather than specifying a coroutine to jump to.

Wikipedia

Description

Generators provide an easy way to implement simple iterators without the overhead or complexity of implementing a class that implements the Iterator interface.

A generator offers a convenient way to provide data to foreach loops without having to build an array in memory ahead of time, which may cause the program to exceed a memory limit or require a considerable amount of processing time to generate. Instead, a generator function can be used, which is the same as a normal function, except that instead of returning once, a generator can yield as many times as it needs to in order to provide the values to be iterated over. Like with iterators, random data access is not possible.

PHP Reference

Example: Generator

<?php

function someGenerator(int $value, int $quantity, callable $algorithm): Generator {
    for ($i = 1; $i < $quantity; $i++) {
        $value = $algorithm($value);

        yield $value;
    }
}

$values = someGenerator(0, 10, function(int $value) {
    return $value + 2;
});

print("Information:\n");
var_dump($values);
print('Type: ' . gettype($values) . PHP_EOL . PHP_EOL);

foreach ($values as $value) {
    print($value . ' ');
}

print(PHP_EOL);

Result (PHP 8.4):

Information:
object(Generator)#2 (1) {
  ["function"]=>
  string(13) "someGenerator"
}
Type: object

2 4 6 8 10 12 14 16 18

Source code: Example

A simple example of this is to reimplement the range() function as a generator. The standard range() function has to generate an array with every value in it and return it, which can result in large arrays: for example, calling range(0, 1000000) will result in well over 100 MB of memory being used.

As an alternative, we can implement an xrange() generator, which will only ever need enough memory to create an Iterator object and track the current state of the generator internally, which turns out to be less than 1 kilobyte.

Example: Implementing range() as a generator

<?php
function xrange($start, $limit, $step = 1) {
    if ($start <= $limit) {
        if ($step <= 0) {
            throw new LogicException('Step must be positive');
        }

        for ($i = $start; $i <= $limit; $i += $step) {
            yield $i;
        }
    } else {
        if ($step >= 0) {
            throw new LogicException('Step must be negative');
        }

        for ($i = $start; $i >= $limit; $i += $step) {
            yield $i;
        }
    }
}

/*
 * Note that both range() and xrange() result in the same
 * output below.
 */

echo 'Single digit odd numbers from range():  ';
foreach (range(1, 9, 2) as $number) {
    echo "$number ";
}
echo "\n";

echo 'Single digit odd numbers from xrange(): ';
foreach (xrange(1, 9, 2) as $number) {
    echo "$number ";
}
?>

The above example will output:

Single digit odd numbers from range():  1 3 5 7 9
Single digit odd numbers from xrange(): 1 3 5 7 9

PHP Reference

Generator objects

When a generator function is called, a new object of the internal Generator class is returned. This object implements the Iterator interface in much the same way as a forward-only iterator object would, and provides methods that can be called to manipulate the state of the generator, including sending values to and returning values from it.

PHP Reference

Generator syntax

A generator function looks just like a normal function, except that instead of returning a value, a generator yields as many values as it needs to. Any function containing yield is a generator function.

Yielding values by a generator

When a generator function is called, it returns an object that can be iterated over. When you iterate over that object (for instance, via a foreach loop), PHP will call the object’s iteration methods each time it needs a value, then saves the state of the generator when the generator yields a value so that it can be resumed when the next value is required.

Once there are no more values to be yielded, then the generator can simply return, and the calling code continues just as if an array has run out of values.

The heart of a generator function is the yield keyword. In its simplest form, a yield statement looks much like a return statement, except that instead of stopping execution of the function and returning, yield instead provides a value to the code looping over the generator and pauses execution of the generator function.

Example: A simple example of yielding values

<?php
function gen_one_to_three() {
    for ($i = 1; $i <= 3; $i++) {
        // Note that $i is preserved between yields.
        yield $i;
    }
}

$generator = gen_one_to_three();
foreach ($generator as $value) {
    echo "$value\n";
}
?>

The above example will output:

1
2
3

Note:

Internally, sequential integer keys will be paired with the yielded values, just as with a non-associative array.

PHP Reference

Example: Generator yielding values

<?php

function oneValueGenerator()
{
    yield 1;
}

foreach (oneValueGenerator() as $value) {
    print($value . ' ');
}

print(PHP_EOL);

function limitedValuesGenerator()
{
    yield 1;
    yield 3;
    yield 5;
}

foreach (limitedValuesGenerator() as $value) {
    print($value . ' ');
}

print(PHP_EOL);

function unlimitedValuesGenerator()
{
    $value = 0;

    while(true) {
        yield $value++;
    }
}

$anchor = unlimitedValuesGenerator();

print($anchor->current() . ' ');
$anchor->next();
print($anchor->current() . ' ');
$anchor->next();
print($anchor->current() . ' ');
$anchor->next();
print($anchor->current() . ' ');
$anchor->next();
print($anchor->current() . ' ');

print(PHP_EOL);

Result (PHP 8.4):

1
1 3 5
0 1 2 3 4

Source code: Example

Yielding values with keys

PHP also supports associative arrays, and generators are no different. In addition to yielding simple values, as shown above, you can also yield a key at the same time.

The syntax for yielding a key/value pair is very similar to that used to define an associative array, as shown below.

Example: Yielding a key/value pair

<?php
/*
 * The input is semi-colon separated fields, with the first
 * field being an ID to use as a key.
 */

$input = <<<'EOF'
1;PHP;Likes dollar signs
2;Python;Likes whitespace
3;Ruby;Likes blocks
EOF;

function input_parser($input) {
    foreach (explode("\n", $input) as $line) {
        $fields = explode(';', $line);
        $id = array_shift($fields);

        yield $id => $fields;
    }
}

foreach (input_parser($input) as $id => $fields) {
    echo "$id:\n";
    echo "    $fields[0]\n";
    echo "    $fields[1]\n";
}
?>

The above example will output:

1:
    PHP
    Likes dollar signs
2:
    Python
    Likes whitespace
3:
    Ruby
    Likes blocks

PHP Reference

Example: Generator yielding key - value pairs

<?php

function oneValueGenerator()
{
    yield 1 => 1;
}

foreach (oneValueGenerator() as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

function limitedValuesGenerator()
{
    yield 'one' => 1;
    yield 'three' => 3;
    yield 'five' => 5;
}

foreach (limitedValuesGenerator() as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

function unlimitedValuesGenerator()
{
    $key = 0;
    $value = 0;

    while(true) {
        yield $key-- => $value++;
    }
}

$anchor = unlimitedValuesGenerator();

print($anchor->key() . ': ' . $anchor->current() . '; ');
$anchor->next();
print($anchor->key() . ': ' . $anchor->current() . '; ');
$anchor->next();
print($anchor->key() . ': ' . $anchor->current() . '; ');
$anchor->next();
print($anchor->key() . ': ' . $anchor->current() . '; ');
$anchor->next();
print($anchor->key() . ': ' . $anchor->current() . '; ');

print(PHP_EOL);

Result (PHP 8.4):

1: 1;
one: 1; three: 3; five: 5;
0: 0; -1: 1; -2: 2; -3: 3; -4: 4;

Source code: Example

Yielding null values

yield can be called without an argument to yield a null value with an automatic key.

Example: Yielding nulls

<?php
function gen_three_nulls() {
    foreach (range(1, 3) as $i) {
        yield;
    }
}

var_dump(iterator_to_array(gen_three_nulls()));
?>

The above example will output:

array(3) {
  [0]=>
  NULL
  [1]=>
  NULL
  [2]=>
  NULL
}

PHP Reference

Example: Generator yielding null

<?php

function nullGenerator()
{
    while(true) {
        yield;
    }
}

$nulls = nullGenerator();

for ($i = 0; $i < 10; $i++) {
    print($nulls->key() . ': ' . gettype($nulls->current()) . '; ');
    $nulls->next();
}

print(PHP_EOL);

Result (PHP 8.4):

0: NULL; 1: NULL; 2: NULL; 3: NULL; 4: NULL; 5: NULL; 6: NULL; 7: NULL; 8: NULL; 9: NULL;

Source code: Example

Yielding by value

Example: Yielding element by value

<?php

function generatorYieldingByValue(int $value, int $quantity): Generator {
    for ($i = 1; $i < $quantity; $i++) {
        $value *= 2;

        yield $value;
    }
}

foreach (generatorYieldingByValue(1, 6) as $value) {
    print($value . ' ');
    $value += 1;
}

print(PHP_EOL);

Result (PHP 8.4):

2 4 8 16 32

Source code: Example

Yielding by reference

Generator functions are able to yield values by reference as well as by value. This is done in the same way as returning references from functions: by prepending an ampersand to the function name.

Example: Yielding values by reference

<?php
function &gen_reference() {
    $value = 3;

    while ($value > 0) {
        yield $value;
    }
}

/*
 * Note that we can change $number within the loop, and
 * because the generator is yielding references, $value
 * within gen_reference() changes.
 */
foreach (gen_reference() as &$number) {
    echo (--$number).'... ';
}
?>

The above example will output:

2... 1... 0...

PHP Reference

Example: Yielding element by reference

<?php

function &generatorYieldingByValue(int $value, int $quantity): Generator {
    for ($i = 1; $i < $quantity; $i++) {
        $value *= 2;

        yield $value;
    }
}

foreach (generatorYieldingByValue(1, 6) as &$value) {
    print($value . ' ');
    $value += 1;
}

print(PHP_EOL);

Result (PHP 8.4):

2 6 14 30 62

Source code: Example

Generator calling generator

Example: Generator calling generator

<?php

function oddValuesGenerator(int $number)
{
    $value = -1;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value;
    }
}

function evenValueGenerator(int $number)
{
    $value = 0;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value;
    }
}

function someGenerator(int $number)
{
    $odds = oddValuesGenerator($number);
    $evens = evenValueGenerator($number);

    $number *= 2;

    for ($i = 1; $i <= $number; $i++) {
        if ($i % 2) {
            yield $odds->current();
            $odds->next();
        } else {
            yield $evens->current();
            $evens->next();
        }
    }
}

foreach (someGenerator(3) as $value) {
    print($value . ' ');
}

print(PHP_EOL);

Result (PHP 8.4):

1 2 3 4 5 6

Source code: Example

Yielding from array

Example: Yielding from array

<?php

global $oddValues, $evenValues;

$oddValues = [1, 3, 5];
$evenValues = [2, 4, 6];

function someGenerator()
{
    global $oddValues, $evenValues;

    yield from [0];
    yield from $oddValues;
    yield from $evenValues;
}

foreach (someGenerator() as $value) {
    print($value . ' ');
}

print(PHP_EOL);

Result (PHP 8.4):

0 1 3 5 2 4 6

Source code: Example

Example: Yielding from array with keys

<?php

global $oddValues, $evenValues;

$oddValues = ['one' => 1, 'three' => 3, 'five' => 5];
$evenValues = ['two' => 2, 'four' => 4, 'six' => 6];

function someGenerator()
{
    global $oddValues, $evenValues;

    yield from [0 => 0];
    yield from $oddValues;
    yield from $evenValues;
}

foreach (someGenerator() as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

Result (PHP 8.4):

0: 0; one: 1; three: 3; five: 5; two: 2; four: 4; six: 6;

Source code: Example

Yielding from generator

Generator delegation allows you to yield values from another generator, Traversable object, or array by using the yield from keyword. The outer generator will then yield all values from the inner generator, object, or array until that is no longer valid, after which execution will continue in the outer generator.

If a generator is used with yield from, the yield from expression will also return any value returned by the inner generator.

Example: Yielding from generator

<?php

function oddValuesGenerator(int $number)
{
    $value = -1;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value;
    }
}

function evenValueGenerator(int $number)
{
    $value = 0;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value;
    }
}

function someGenerator(int $number)
{
    yield 0;
    yield from oddValuesGenerator($number);
    yield from evenValueGenerator($number);
}

foreach (someGenerator(3) as $value) {
    print($value . ' ');
}

print(PHP_EOL);

Result (PHP 8.4):

0 1 3 5 2 4 6

Source code: Example

Example: Yielding from generator with keys

<?php

function oddValuesGenerator(int $number)
{
    $value = -1;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value => $value;
    }
}

function evenValueGenerator(int $number)
{
    $value = 0;

    for($i = 0; $i < $number; $i++) {
        $value += 2;
        yield $value => $value;
    }
}

function someGenerator(int $number)
{
    yield 0 => 0;
    yield from oddValuesGenerator($number);
    yield from evenValueGenerator($number);
}

foreach (someGenerator(3) as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

Result (PHP 8.4):

0: 0; 1: 1; 3: 3; 5: 5; 2: 2; 4: 4; 6: 6;

Source code: Example

Yielding from and keys collision

Caution

Storing into an array (e.g. with iterator_to_array()) yield from does not reset the keys. It preserves the keys returned by the Traversable object, or array. Thus some values may share a common key with another yield or yield from, which, upon insertion into an array, will overwrite former values with that key.

A common case where this matters is iterator_to_array() returning a keyed array by default, leading to possibly unexpected results. iterator_to_array() has a second parameter preserve_keys which can be set to false to collect all the values while ignoring the keys returned by the generator.

Example: yield from with iterator_to_array()

<?php
function inner() {
    yield 1; // key 0
    yield 2; // key 1
    yield 3; // key 2
}
function gen() {
    yield 0; // key 0
    yield from inner(); // keys 0-2
    yield 4; // key 1
}
// pass false as second parameter to get an array [0, 1, 2, 3, 4]
var_dump(iterator_to_array(gen()));
?>

The above example will output:

array(3) {
  [0]=>
  int(1)
  [1]=>
  int(4)
  [2]=>
  int(3)
}

PHP Reference

Example: Yielding from and keys collision

<?php

global $oddValues;

$oddValues = ['one' => 1, 'two' => 3, 'three' => 5];

function evenValueGenerator()
{
    yield 'one' => 2;
    yield 'two' => 4;
    yield 'three' => 6;
}

function someGenerator()
{
    global $oddValues;

    yield from $oddValues;
    yield from evenValueGenerator();
}

foreach (someGenerator() as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

$someArray = iterator_to_array(someGenerator());

foreach ($someArray as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

$otherArray = iterator_to_array(someGenerator(), false);

foreach ($otherArray as $key => $value) {
    print($key . ': ' . $value . '; ');
}

print(PHP_EOL);

Result (PHP 8.4):

one: 1; two: 3; three: 5; one: 2; two: 4; three: 6;
one: 2; two: 4; three: 6;
0: 1; 1: 3; 2: 5; 3: 2; 4: 4; 5: 6;

Source code: Example

Yielding from array and generator

Example: Basic use of yield from

<?php
function count_to_ten() {
    yield 1;
    yield 2;
    yield from [3, 4];
    yield from new ArrayIterator([5, 6]);
    yield from seven_eight();
    yield 9;
    yield 10;
}

function seven_eight() {
    yield 7;
    yield from eight();
}

function eight() {
    yield 8;
}

foreach (count_to_ten() as $num) {
    echo "$num ";
}
?>

The above example will output:

1 2 3 4 5 6 7 8 9 10

Example: yield from and return values

<?php
function count_to_ten() {
    yield 1;
    yield 2;
    yield from [3, 4];
    yield from new ArrayIterator([5, 6]);
    yield from seven_eight();
    return yield from nine_ten();
}

function seven_eight() {
    yield 7;
    yield from eight();
}

function eight() {
    yield 8;
}

function nine_ten() {
    yield 9;
    return 10;
}

$gen = count_to_ten();
foreach ($gen as $num) {
    echo "$num ";
}
echo $gen->getReturn();
?>

The above example will output:

1 2 3 4 5 6 7 8 9 10

PHP Reference

Returning values by a generator

Note:

A generator can return values, which can be retrieved using Generator::getReturn().

PHP Reference

Example: Generator returning value

<?php

function someGenerator(int $number): Generator
{
    $result = 0;

    for ($i = 0; $i < $number; $i++) {
        $result += $i;
        yield $i;
    }

    return $result;
}

$values = someGenerator(5);

foreach($values as $value) {
    print($value . ' ');
}

print(PHP_EOL);

$result = $values->getReturn();
print($result . PHP_EOL);

Result (PHP 8.4):

0 1 2 3 4
10

Source code: Example

Comparing generators with Iterator objects

The primary advantage of generators is their simplicity. Much less boilerplate code has to be written compared to implementing an Iterator class, and the code is generally much more readable. For example, the following function and class are equivalent:

<?php
function getLinesFromFile($fileName) {
    if (!$fileHandle = fopen($fileName, 'r')) {
        return;
    }

    while (false !== $line = fgets($fileHandle)) {
        yield $line;
    }

    fclose($fileHandle);
}

// versus...

class LineIterator implements Iterator {
    protected $fileHandle;

    protected $line;
    protected $i;

    public function __construct($fileName) {
        if (!$this->fileHandle = fopen($fileName, 'r')) {
            throw new RuntimeException('Couldn\'t open file "' . $fileName . '"');
        }
    }

    public function rewind() {
        fseek($this->fileHandle, 0);
        $this->line = fgets($this->fileHandle);
        $this->i = 0;
    }

    public function valid() {
        return false !== $this->line;
    }

    public function current() {
        return $this->line;
    }

    public function key() {
        return $this->i;
    }

    public function next() {
        if (false !== $this->line) {
            $this->line = fgets($this->fileHandle);
            $this->i++;
        }
    }

    public function __destruct() {
        fclose($this->fileHandle);
    }
}
?>

This flexibility does come at a cost, however: generators are forward-only iterators, and cannot be rewound once iteration has started. This also means that the same generator can’t be iterated over multiple times: the generator will need to be rebuilt by calling the generator function again.

PHP Reference

▵ Up ⌂ Home ▲ Previous: Fibers ▼ Next: Classes