1
0
mirror of https://github.com/danog/loop.git synced 2024-12-12 09:19:45 +01:00
loop/lib/Generic/PeriodicLoop.php
2020-07-23 13:26:16 +02:00

106 lines
2.5 KiB
PHP

<?php
/**
* Periodic loop.
*
* @author Daniil Gentili <daniil@daniil.it>
* @copyright 2016-2020 Daniil Gentili <daniil@daniil.it>
* @license https://opensource.org/licenses/MIT MIT
*/
namespace danog\Loop\Generic;
use Amp\Promise;
use danog\Loop\ResumableSignalLoop;
/**
* Periodic loop.
*
* Runs a callback at a periodic interval.
*
* The loop can be stopped from the outside or
* from the inside by signaling or returning `true`.
*
* @template T as bool
* @template TGenerator as \Generator<mixed,Promise|array<array-key,Promise>,mixed,Promise<T>|T>
* @template TPromise as Promise<T>
*
* @template TCallable as T|TPromise|TGenerator
*
* @author Daniil Gentili <daniil@daniil.it>
*/
class PeriodicLoop extends ResumableSignalLoop
{
/**
* Callback.
*
* @var callable
*
* @psalm-var callable():TCallable
*/
private $callback;
/**
* Loop name.
*
* @var string
*/
private $name;
/**
* Loop interval.
*
* @var ?int
*/
private $interval;
/**
* Constructor.
*
* @param callable $callback Callback to call
* @param string $name Loop name
* @param ?int $interval Loop interval
*
* @psalm-param callable():TCallable $callback Callable to run
*/
public function __construct(callable $callback, string $name, ?int $interval)
{
$this->callback = $callback;
$this->name = $name;
$this->interval = $interval;
}
/**
* Loop implementation.
*
* @return \Generator
*/
public function loop(): \Generator
{
$callback = $this->callback;
while (true) {
/** @psalm-suppress MixedAssignment */
$result = $callback();
if ($result instanceof \Generator) {
/** @psalm-var TGenerator */
$result = yield from $result;
} elseif ($result instanceof Promise) {
/** @psalm-var TPromise */
$result = yield $result;
}
if ($result === true) {
return;
}
/** @psalm-suppress MixedAssignment */
$result = yield $this->waitSignal($this->pause($this->interval));
if ($result === true) {
return;
}
}
}
/**
* Get name of the loop.
*
* @return string
*/
public function __toString(): string
{
return $this->name;
}
}