| ... | @@ -143,6 +143,23 @@ pub fn Reader( | ... | @@ -143,6 +143,23 @@ pub fn Reader( |
| 143 | return array_list.toOwnedSlice(); | 143 | return array_list.toOwnedSlice(); |
| 144 | } | 144 | } |
| 145 | | 145 | |
| | 146 | /// Reads from the stream until specified byte is found. If the buffer is not |
| | 147 | /// large enough to hold the entire contents, `error.StreamTooLong` is returned. |
| | 148 | /// Returns a slice of the stream data, with ptr equal to `buf.ptr`. The |
| | 149 | /// delimiter byte is not included in the returned slice. |
| | 150 | pub fn readUntilDelimiter(self: Self, buf: []u8, delimiter: u8) ![]u8 { |
| | 151 | var index: usize = 0; |
| | 152 | while (true) { |
| | 153 | const byte = try self.readByte(); |
| | 154 | |
| | 155 | if (byte == delimiter) return buf[0..index]; |
| | 156 | if (index >= buf.len) return error.StreamTooLong; |
| | 157 | |
| | 158 | buf[index] = byte; |
| | 159 | index += 1; |
| | 160 | } |
| | 161 | } |
| | 162 | |
| 146 | /// Allocates enough memory to read until `delimiter` or end-of-stream. | 163 | /// Allocates enough memory to read until `delimiter` or end-of-stream. |
| 147 | /// If the allocated memory would be greater than `max_size`, returns | 164 | /// If the allocated memory would be greater than `max_size`, returns |
| 148 | /// `error.StreamTooLong`. If end-of-stream is found, returns the rest | 165 | /// `error.StreamTooLong`. If end-of-stream is found, returns the rest |
Samadi van Koten <samadi@vktec.org.uk> 2021-06-07 22:09:53+01:00
Veikka Tuominen <git@vexu.eu> 2021-08-20 12:20:42+03:00
lib