nsq/nsqphp
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Receive Reader as dependency in Subscriber instead of extending it

Browse Source
This commit is contained in:
Konstantin Grachev
2021-01-23 01:24:35 +03:00
parent 6509753bd9
commit 7873a9f010
4 changed files with 25 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/Subscriber.php
View File

@@ -9,24 +9,31 @@ use function get_debug_type;
use function microtime;
use function sprintf;
final class Subscriber extends Reader
final class Subscriber
{
public const STOP = 0;
public const CHANGE_TIMEOUT = 1;
private Reader $reader;
public function __construct(Reader $reader)
{
$this->reader = $reader;
}
/**
* @psalm-return Generator<int, Envelope|null, int|float|null, void>
*/
public function subscribe(string $topic, string $channel, float $timeout = 0): Generator
{
$this->sub($topic, $channel);
$this->reader->sub($topic, $channel);
while (true) {
$this->rdy(1);
$this->reader->rdy(1);
$message = $this->consume($timeout);
$command = yield null === $message ? null : new Envelope($message, $this);
$command = yield null === $message ? null : new Envelope($message, $this->reader);
if (self::STOP === $command) {
break;
@@ -43,20 +50,20 @@ final class Subscriber extends Reader
}
}
$this->disconnect();
$this->reader->disconnect();
}
private function consume(float $timeout): ?Message
{
$deadline = microtime(true) + $timeout;
$response = $this->receive($timeout);
$response = $this->reader->receive($timeout);
if (null === $response) {
return null;
}
if ($response->isHeartBeat()) {
$this->send('NOP'.PHP_EOL);
$this->reader->nop();
return $this->consume(
($currentTime = microtime(true)) > $deadline ? 0 : $deadline - $currentTime