authorgravatar for jhc@dismail.deJimmi Holst Christensen <jhc@dismail.de> 2022-01-22 13:24:09+01:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2022-01-23 14:41:15-05:00
log7287c7482a2c694c7c7f56b9f7c1744a7ae7905f
tree9a8a3ace5afc9d45a17b9e07f6d1bdf3055e2a78
parentfa84625aeb852cc921cb596f4baccd452c3ce3b6

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.

1 files changed, 12 insertions(+), 11 deletions(-)

lib/std/io/buffered_writer.zig+12-11
......@@ -1,24 +1,22 @@
11const std = @import("../std.zig");
2
23const io = std.io;
4const mem = std.mem;
35
46pub fn BufferedWriter(comptime buffer_size: usize, comptime WriterType: type) type {
57 return struct {
68 unbuffered_writer: WriterType,
7 fifo: FifoType = FifoType.init(),
9 buf: [buffer_size]u8 = undefined,
10 end: usize = 0,
811
912 pub const Error = WriterType.Error;
1013 pub const Writer = io.Writer(*Self, Error, write);
1114
1215 const Self = @This();
13 const FifoType = std.fifo.LinearFifo(u8, std.fifo.LinearFifoBufferType{ .Static = buffer_size });
1416
1517 pub fn flush(self: *Self) !void {
16 while (true) {
17 const slice = self.fifo.readableSlice(0);
18 if (slice.len == 0) break;
19 try self.unbuffered_writer.writeAll(slice);
20 self.fifo.discard(slice.len);
21 }
18 try self.unbuffered_writer.writeAll(self.buf[0..self.end]);
19 self.end = 0;
2220 }
2321
2422 pub fn writer(self: *Self) Writer {
......@@ -26,11 +24,14 @@ pub fn BufferedWriter(comptime buffer_size: usize, comptime WriterType: type) ty
2624 }
2725
2826 pub fn write(self: *Self, bytes: []const u8) Error!usize {
29 if (bytes.len >= self.fifo.writableLength()) {
27 if (self.end + bytes.len > self.buf.len) {
3028 try self.flush();
31 return self.unbuffered_writer.write(bytes);
29 if (bytes.len > self.buf.len)
30 return self.unbuffered_writer.write(bytes);
3231 }
33 self.fifo.writeAssumeCapacity(bytes);
32
33 mem.copy(u8, self.buf[self.end..], bytes);
34 self.end += bytes.len;
3435 return bytes.len;
3536 }
3637 };