[e3d4e0a] | 1 | <?php
|
---|
| 2 |
|
---|
| 3 | declare(strict_types=1);
|
---|
| 4 |
|
---|
| 5 | namespace GuzzleHttp\Psr7;
|
---|
| 6 |
|
---|
| 7 | use Psr\Http\Message\StreamInterface;
|
---|
| 8 |
|
---|
| 9 | /**
|
---|
| 10 | * Provides a buffer stream that can be written to to fill a buffer, and read
|
---|
| 11 | * from to remove bytes from the buffer.
|
---|
| 12 | *
|
---|
| 13 | * This stream returns a "hwm" metadata value that tells upstream consumers
|
---|
| 14 | * what the configured high water mark of the stream is, or the maximum
|
---|
| 15 | * preferred size of the buffer.
|
---|
| 16 | */
|
---|
| 17 | final class BufferStream implements StreamInterface
|
---|
| 18 | {
|
---|
| 19 | /** @var int */
|
---|
| 20 | private $hwm;
|
---|
| 21 |
|
---|
| 22 | /** @var string */
|
---|
| 23 | private $buffer = '';
|
---|
| 24 |
|
---|
| 25 | /**
|
---|
| 26 | * @param int $hwm High water mark, representing the preferred maximum
|
---|
| 27 | * buffer size. If the size of the buffer exceeds the high
|
---|
| 28 | * water mark, then calls to write will continue to succeed
|
---|
| 29 | * but will return 0 to inform writers to slow down
|
---|
| 30 | * until the buffer has been drained by reading from it.
|
---|
| 31 | */
|
---|
| 32 | public function __construct(int $hwm = 16384)
|
---|
| 33 | {
|
---|
| 34 | $this->hwm = $hwm;
|
---|
| 35 | }
|
---|
| 36 |
|
---|
| 37 | public function __toString(): string
|
---|
| 38 | {
|
---|
| 39 | return $this->getContents();
|
---|
| 40 | }
|
---|
| 41 |
|
---|
| 42 | public function getContents(): string
|
---|
| 43 | {
|
---|
| 44 | $buffer = $this->buffer;
|
---|
| 45 | $this->buffer = '';
|
---|
| 46 |
|
---|
| 47 | return $buffer;
|
---|
| 48 | }
|
---|
| 49 |
|
---|
| 50 | public function close(): void
|
---|
| 51 | {
|
---|
| 52 | $this->buffer = '';
|
---|
| 53 | }
|
---|
| 54 |
|
---|
| 55 | public function detach()
|
---|
| 56 | {
|
---|
| 57 | $this->close();
|
---|
| 58 |
|
---|
| 59 | return null;
|
---|
| 60 | }
|
---|
| 61 |
|
---|
| 62 | public function getSize(): ?int
|
---|
| 63 | {
|
---|
| 64 | return strlen($this->buffer);
|
---|
| 65 | }
|
---|
| 66 |
|
---|
| 67 | public function isReadable(): bool
|
---|
| 68 | {
|
---|
| 69 | return true;
|
---|
| 70 | }
|
---|
| 71 |
|
---|
| 72 | public function isWritable(): bool
|
---|
| 73 | {
|
---|
| 74 | return true;
|
---|
| 75 | }
|
---|
| 76 |
|
---|
| 77 | public function isSeekable(): bool
|
---|
| 78 | {
|
---|
| 79 | return false;
|
---|
| 80 | }
|
---|
| 81 |
|
---|
| 82 | public function rewind(): void
|
---|
| 83 | {
|
---|
| 84 | $this->seek(0);
|
---|
| 85 | }
|
---|
| 86 |
|
---|
| 87 | public function seek($offset, $whence = SEEK_SET): void
|
---|
| 88 | {
|
---|
| 89 | throw new \RuntimeException('Cannot seek a BufferStream');
|
---|
| 90 | }
|
---|
| 91 |
|
---|
| 92 | public function eof(): bool
|
---|
| 93 | {
|
---|
| 94 | return strlen($this->buffer) === 0;
|
---|
| 95 | }
|
---|
| 96 |
|
---|
| 97 | public function tell(): int
|
---|
| 98 | {
|
---|
| 99 | throw new \RuntimeException('Cannot determine the position of a BufferStream');
|
---|
| 100 | }
|
---|
| 101 |
|
---|
| 102 | /**
|
---|
| 103 | * Reads data from the buffer.
|
---|
| 104 | */
|
---|
| 105 | public function read($length): string
|
---|
| 106 | {
|
---|
| 107 | $currentLength = strlen($this->buffer);
|
---|
| 108 |
|
---|
| 109 | if ($length >= $currentLength) {
|
---|
| 110 | // No need to slice the buffer because we don't have enough data.
|
---|
| 111 | $result = $this->buffer;
|
---|
| 112 | $this->buffer = '';
|
---|
| 113 | } else {
|
---|
| 114 | // Slice up the result to provide a subset of the buffer.
|
---|
| 115 | $result = substr($this->buffer, 0, $length);
|
---|
| 116 | $this->buffer = substr($this->buffer, $length);
|
---|
| 117 | }
|
---|
| 118 |
|
---|
| 119 | return $result;
|
---|
| 120 | }
|
---|
| 121 |
|
---|
| 122 | /**
|
---|
| 123 | * Writes data to the buffer.
|
---|
| 124 | */
|
---|
| 125 | public function write($string): int
|
---|
| 126 | {
|
---|
| 127 | $this->buffer .= $string;
|
---|
| 128 |
|
---|
| 129 | if (strlen($this->buffer) >= $this->hwm) {
|
---|
| 130 | return 0;
|
---|
| 131 | }
|
---|
| 132 |
|
---|
| 133 | return strlen($string);
|
---|
| 134 | }
|
---|
| 135 |
|
---|
| 136 | /**
|
---|
| 137 | * @return mixed
|
---|
| 138 | */
|
---|
| 139 | public function getMetadata($key = null)
|
---|
| 140 | {
|
---|
| 141 | if ($key === 'hwm') {
|
---|
| 142 | return $this->hwm;
|
---|
| 143 | }
|
---|
| 144 |
|
---|
| 145 | return $key ? null : [];
|
---|
| 146 | }
|
---|
| 147 | }
|
---|