| 1 |
<?php |
| 2 |
|
| 3 |
declare (strict_types=1); |
| 4 |
namespace Dudlewebs\WPMCS\s3\GuzzleHttp\Psr7; |
| 5 |
|
| 6 |
use Dudlewebs\WPMCS\s3\Psr\Http\Message\StreamInterface; |
| 7 |
/** |
| 8 |
* Stream decorator that begins dropping data once the size of the underlying |
| 9 |
* stream becomes too full. |
| 10 |
*/ |
| 11 |
final class DroppingStream implements StreamInterface |
| 12 |
{ |
| 13 |
use StreamDecoratorTrait; |
| 14 |
/** @var int */ |
| 15 |
private $maxLength; |
| 16 |
/** @var StreamInterface */ |
| 17 |
private $stream; |
| 18 |
/** |
| 19 |
* @param StreamInterface $stream Underlying stream to decorate. |
| 20 |
* @param int $maxLength Maximum size before dropping data. |
| 21 |
*/ |
| 22 |
public function __construct(StreamInterface $stream, int $maxLength) |
| 23 |
{ |
| 24 |
$this->stream = $stream; |
| 25 |
$this->maxLength = $maxLength; |
| 26 |
} |
| 27 |
public function write($string) : int |
| 28 |
{ |
| 29 |
$diff = $this->maxLength - $this->stream->getSize(); |
| 30 |
// Begin returning 0 when the underlying stream is too large. |
| 31 |
if ($diff <= 0) { |
| 32 |
return 0; |
| 33 |
} |
| 34 |
// Write the stream or a subset of the stream if needed. |
| 35 |
if (\strlen($string) < $diff) { |
| 36 |
return $this->stream->write($string); |
| 37 |
} |
| 38 |
return $this->stream->write(\substr($string, 0, $diff)); |
| 39 |
} |
| 40 |
} |
| 41 |
|
