| ... | ... | @@ -234,6 +234,18 @@ pub fn InStream( |
| 234 | 234 | } |
| 235 | 235 | } |
| 236 | 236 | |
| 237 | /// Reads `slice.len` bytes from the stream and returns if they are the same as the passed slice |
| 238 | pub fn isBytes(self: Self, slice: []const u8) !bool { |
| 239 | var i: usize = 0; |
| 240 | var matches = true; |
| 241 | while (i < slice.len) : (i += 1) { |
| 242 | if (slice[i] != try self.readByte()) { |
| 243 | matches = false; |
| 244 | } |
| 245 | } |
| 246 | return matches; |
| 247 | } |
| 248 | |
| 237 | 249 | pub fn readStruct(self: Self, comptime T: type) !T { |
| 238 | 250 | // Only extern and packed structs have defined in-memory layout. |
| 239 | 251 | comptime assert(@typeInfo(T).Struct.layout != builtin.TypeInfo.ContainerLayout.Auto); |
| ... | ... | @@ -276,3 +288,9 @@ test "InStream" { |
| 276 | 288 | }, undefined)) == .c); |
| 277 | 289 | testing.expectError(error.EndOfStream, in_stream.readByte()); |
| 278 | 290 | } |
| 291 | |
| 292 | test "InStream.isBytes" { |
| 293 | const in_stream = std.io.fixedBufferStream("foobar").inStream(); |
| 294 | testing.expectEqual(true, try in_stream.isBytes("foo")); |
| 295 | testing.expectEqual(false, try in_stream.isBytes("qux")); |
| 296 | } |
daurnimator <quae@daurnimator.com> 2020-04-25 20:42:13+10:00
Commit is signed but in an unrecognized format.
lib