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

Reject non-strings in IteratorStream

Browse Source
This commit is contained in:
Niklas Keller 2017-05-22 10:41:35 +02:00
parent 2049e76771
commit 6b369657a1
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib/IteratorStream.php
View File

@ -3,11 +3,13 @@
namespace Amp\ByteStream;
use Amp\Deferred;
use Amp\Failure;
use Amp\Iterator;
use Amp\Promise;
class IteratorStream implements InputStream {
private $iterator;
private $exception;
public function __construct(Iterator $iterator) {
$this->iterator = $iterator;
@ -15,13 +17,31 @@ class IteratorStream implements InputStream {
/** @inheritdoc */
public function read(): Promise {
if ($this->exception) {
return new Failure($this->exception);
}
$deferred = new Deferred;
$this->iterator->advance()->onResolve(function ($error, $hasNextElement) use ($deferred) {
if ($error) {
$this->exception = $error;
$deferred->fail($error);
} elseif ($hasNextElement) {
$deferred->resolve($this->iterator->getCurrent());
$chunk = $this->iterator->getCurrent();
if (!\is_string($chunk)) {
$this->exception = new StreamException(\sprintf(
"Unexpected iterator value of type '%s', expected string",
\is_object($chunk) ? \get_class($chunk) : \gettype($chunk)
));
$deferred->fail($this->exception);
return;
}
$deferred->resolve($chunk);
} else {
$deferred->resolve(null);
}