danog/byte-stream
1
0
Fork 0
mirror of https://github.com/danog/byte-stream.git synced 2024-11-26 20:04:51 +01:00
Code Issues Projects Releases Wiki Activity

Throw correct PendingReadError for IteratorStream

Browse Source
This commit is contained in:
Niklas Keller 2017-12-22 09:31:37 +01:00
parent 926fa6d24d
commit 5dffe33692
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lib/IteratorStream.php
View File

@ -10,6 +10,7 @@ use Amp\Promise;
final class IteratorStream implements InputStream {
private $iterator;
private $exception;
private $pending = false;
public function __construct(Iterator $iterator) {
$this->iterator = $iterator;
@ -21,9 +22,16 @@ final class IteratorStream implements InputStream {
return new Failure($this->exception);
}
if ($this->pending) {
throw new PendingReadError;
}
$this->pending = true;
$deferred = new Deferred;
$this->iterator->advance()->onResolve(function ($error, $hasNextElement) use ($deferred) {
$this->pending = false;
if ($error) {
$this->exception = $error;
$deferred->fail($error);
@ -43,7 +51,7 @@ final class IteratorStream implements InputStream {
$deferred->resolve($chunk);
} else {
$deferred->resolve(null);
$deferred->resolve();
}
});