#!/usr/bin/env php
<?php

require __DIR__ . '/../../vendor/autoload.php';

use Amp\AsyncGenerator;
use Amp\Delayed;
use Amp\Loop;

Loop::run(function () {
    try {
        /** @psalm-var AsyncGenerator<int, void, void> $pipeline */
        $pipeline = new AsyncGenerator(function (callable $emit): \Generator {
            yield $emit(1);
            yield $emit(yield new Delayed(500, 2));
            yield $emit(3);
            yield $emit(yield new Delayed(300, 4));
            yield $emit(5);
            yield $emit(6);
            yield $emit(yield new Delayed(1000, 7));
            yield $emit(8);
            yield $emit(9);
            yield $emit(yield new Delayed(600, 10));
        });

        // Flow listener attempts to consume 11 values at once. Only 10 will be emitted.
        $promises = [];
        for ($i = 0; $i < 11 && ($promises[] = $pipeline->continue()); ++$i);

        foreach ($promises as $key => $promise) {
            if (null === $yielded = yield $promise) {
                \printf("Async generator completed after yielding %d values\n", $key);
                break;
            }

            \printf("Async generator yielded %d\n", $yielded);
        }
    } catch (\Exception $exception) {
        \printf("Exception: %s\n", (string) $exception);
    }
});