From 7287c7482a2c694c7c7f56b9f7c1744a7ae7905f Mon Sep 17 00:00:00 2001 From: Jimmi Holst Christensen Date: Sat, 22 Jan 2022 13:24:09 +0100 Subject: [PATCH] Stop using LinearFifo in BufferedWriter Looking at the BufferedWriter assembly generated, one can see that is has to do a lot of work, just to copy over some bytes and increase an offset. This is because the LinearFifo is a much more general construct than what BufferedWriter needs and the optimizer cannot prove that we don't need to do this extra work. --- lib/std/io/buffered_writer.zig | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/lib/std/io/buffered_writer.zig b/lib/std/io/buffered_writer.zig index 77f556f2d2..497a1810a7 100644 --- a/lib/std/io/buffered_writer.zig +++ b/lib/std/io/buffered_writer.zig @@ -1,24 +1,22 @@ const std = @import("../std.zig"); + const io = std.io; +const mem = std.mem; pub fn BufferedWriter(comptime buffer_size: usize, comptime WriterType: type) type { return struct { unbuffered_writer: WriterType, - fifo: FifoType = FifoType.init(), + buf: [buffer_size]u8 = undefined, + end: usize = 0, pub const Error = WriterType.Error; pub const Writer = io.Writer(*Self, Error, write); const Self = @This(); - const FifoType = std.fifo.LinearFifo(u8, std.fifo.LinearFifoBufferType{ .Static = buffer_size }); pub fn flush(self: *Self) !void { - while (true) { - const slice = self.fifo.readableSlice(0); - if (slice.len == 0) break; - try self.unbuffered_writer.writeAll(slice); - self.fifo.discard(slice.len); - } + try self.unbuffered_writer.writeAll(self.buf[0..self.end]); + self.end = 0; } pub fn writer(self: *Self) Writer { @@ -26,11 +24,14 @@ pub fn BufferedWriter(comptime buffer_size: usize, comptime WriterType: type) ty } pub fn write(self: *Self, bytes: []const u8) Error!usize { - if (bytes.len >= self.fifo.writableLength()) { + if (self.end + bytes.len > self.buf.len) { try self.flush(); - return self.unbuffered_writer.write(bytes); + if (bytes.len > self.buf.len) + return self.unbuffered_writer.write(bytes); } - self.fifo.writeAssumeCapacity(bytes); + + mem.copy(u8, self.buf[self.end..], bytes); + self.end += bytes.len; return bytes.len; } };