parallel/lib/Context/Internal/ParallelHub.php

<?php

namespace Amp\Parallel\Context\Internal;

use Amp\Loop;
use Amp\Parallel\Sync\ChannelledSocket;
use parallel\Events;
use parallel\Events\Error\Timeout;
use parallel\Future;

class ParallelHub extends ProcessHub
{
    const EXIT_CHECK_FREQUENCY = 250;

    /** @var ChannelledSocket[] */
    private $channels;

    /** @var string */
    private $watcher;

    /** @var Events */
    private $events;

    public function __construct()
    {
        parent::__construct();

        $events = $this->events = new Events;
        $this->events->setTimeout(0);

        $channels = &$this->channels;
        $this->watcher = Loop::repeat(self::EXIT_CHECK_FREQUENCY, static function () use (&$channels, $events): void {
            try {
                while ($event = $events->poll()) {
                    $id = (int) $event->source;
                    \assert(isset($channels[$id]), 'Channel for context ID not found');
                    $channels[$id]->close();
                }
            } catch (Timeout $exception) {
                // Ignore poll timeout.
            }
        });
        Loop::disable($this->watcher);
        Loop::unreference($this->watcher);
    }

    public function add(int $id, ChannelledSocket $channel, Future $future): void
    {
        $this->channels[$id] = $channel;
        $this->events->addFuture((string) $id, $future);

        Loop::enable($this->watcher);
    }

    public function remove(int $id): void
    {
        if (!isset($this->channels[$id])) {
            return;
        }

        unset($this->channels[$id]);
        $this->events->remove((string) $id);

        if (empty($this->channels)) {
            Loop::disable($this->watcher);
        }
    }
}