| ... | @@ -0,0 +1,43 @@ |
| 1 | const std = @import("std"); |
| 2 | |
| 3 | const nio = @import("./nio.zig"); |
| 4 | |
| 5 | pub fn FixedBufferStream(comptime Buffer: type) type { |
| 6 | comptime std.debug.assert(Buffer == []u8 or Buffer == []const u8); |
| 7 | return struct { |
| 8 | buffer: Buffer, |
| 9 | pos: usize, |
| 10 | |
| 11 | const Self = @This(); |
| 12 | |
| 13 | pub fn init(buffer: Buffer) Self { |
| 14 | return .{ |
| 15 | .buffer = buffer, |
| 16 | .pos = 0, |
| 17 | }; |
| 18 | } |
| 19 | |
| 20 | pub const ReadError = error{}; |
| 21 | pub usingnamespace nio.Readable(@This()); |
| 22 | pub fn read(self: *Self, dest: []u8) ReadError!usize { |
| 23 | const size = @min(dest.len, self.buffer.len - self.pos); |
| 24 | const end = self.pos + size; |
| 25 | @memcpy(dest[0..size], self.buffer[self.pos..end]); |
| 26 | self.pos = end; |
| 27 | return size; |
| 28 | } |
| 29 | |
| 30 | pub fn anyReadable(self: *Self) nio.AnyReadable { |
| 31 | const S = struct { |
| 32 | fn foo(s: *anyopaque, buffer: []u8) anyerror!usize { |
| 33 | const fbs: *Self = @ptrCast(@alignCast(s)); |
| 34 | return fbs.read(buffer); |
| 35 | } |
| 36 | }; |
| 37 | return .{ |
| 38 | .readFn = S.foo, |
| 39 | .state = @ptrCast(self), |
| 40 | }; |
| 41 | } |
| 42 | }; |
| 43 | } |