1
0
mirror of https://github.com/danog/amp.git synced 2024-12-13 09:57:25 +01:00
amp/lib/Internal/ResolutionQueue.php

85 lines
2.7 KiB
PHP
Raw Normal View History

<?php
2016-08-16 06:46:26 +02:00
2016-05-24 05:48:28 +02:00
namespace Amp\Internal;
2016-05-21 16:44:52 +02:00
use Amp\Loop;
use Amp\Promise;
2016-05-21 16:44:52 +02:00
2016-06-01 19:18:11 +02:00
/**
2016-11-14 20:59:21 +01:00
* Stores a set of functions to be invoked when a promise is resolved.
2016-06-01 19:18:11 +02:00
*
* @internal
2020-03-28 20:27:42 +01:00
* @psalm-internal Amp\Internal
2016-06-01 19:18:11 +02:00
*/
final class ResolutionQueue
2018-06-18 20:00:01 +02:00
{
/** @var array<array-key, callable(\Throwable|null, mixed): (Promise|\React\Promise\PromiseInterface|\Generator<mixed,
* Promise|\React\Promise\PromiseInterface|array<array-key, Promise|\React\Promise\PromiseInterface>, mixed,
* mixed>|null) | callable(\Throwable|null, mixed): void> */
2020-09-24 18:52:22 +02:00
private array $queue = [];
2016-05-21 16:44:52 +02:00
/**
* @param callable|null $callback Initial callback to add to queue.
*
* @psalm-param null|callable(\Throwable|null, mixed): (Promise|\React\Promise\PromiseInterface|\Generator<mixed,
* Promise|\React\Promise\PromiseInterface|array<array-key, Promise|\React\Promise\PromiseInterface>, mixed,
* mixed>|null) | callable(\Throwable|null, mixed): void $callback
2016-05-21 16:44:52 +02:00
*/
2018-06-18 20:00:01 +02:00
public function __construct(callable $callback = null)
{
if ($callback !== null) {
2016-05-21 16:44:52 +02:00
$this->push($callback);
}
}
/**
* Unrolls instances of self to avoid blowing up the call stack on resolution.
*
* @param callable $callback
*
* @psalm-param callable(\Throwable|null, mixed): (Promise|\React\Promise\PromiseInterface|\Generator<mixed,
* Promise|\React\Promise\PromiseInterface|array<array-key, Promise|\React\Promise\PromiseInterface>, mixed,
* mixed>|null) | callable(\Throwable|null, mixed): void $callback
*
* @return void
*/
2020-09-24 18:52:22 +02:00
public function push(callable $callback): void
2018-06-18 20:00:01 +02:00
{
if ($callback instanceof self) {
$this->queue = \array_merge($this->queue, $callback->queue);
return;
}
$this->queue[] = $callback;
}
2016-05-21 16:44:52 +02:00
/**
* Calls each callback in the queue, passing the provided values to the function.
*
2016-08-11 21:35:58 +02:00
* @param \Throwable|null $exception
* @param mixed $value
*
* @return void
2016-05-21 16:44:52 +02:00
*/
2020-09-24 18:52:22 +02:00
public function __invoke(?\Throwable $exception, mixed $value): void
2018-06-18 20:00:01 +02:00
{
2016-05-21 16:44:52 +02:00
foreach ($this->queue as $callback) {
try {
$result = $callback($exception, $value);
if ($result instanceof \Awaitable) {
$result->onResolve(static function (?\Throwable $exception): void {
if ($exception) {
throw $exception;
}
});
}
2016-05-21 16:44:52 +02:00
} catch (\Throwable $exception) {
Loop::defer(static function () use ($exception): void {
throw $exception;
});
2016-05-21 16:44:52 +02:00
}
}
}
}