zig

commit 9129fb28dcc3c94a709cefd8040f41b9125693ee (tree)
parent 2df3de1e209076cb27d6573a0cdd2bed075d88b3
Author: Andrew Kelley <andrew@ziglang.org>
Date:   Tue, 20 Feb 2024 03:16:02 -0700

std.ArrayList: add writerAssumeCapacity

Useful when you want to use an ArrayList to operate on a static buffer.

Diffstat:
M
lib/std/array_list.zig
 | 
21
+++++++++++++++++++--

1 file changed, 19 insertions(+), 2 deletions(-)
diff --git a/lib/std/array_list.zig b/lib/std/array_list.zig
@@ -937,14 +937,31 @@ pub fn ArrayListAlignedUnmanaged(comptime T: type, comptime alignment: ?u29) typ
             return .{ .context = .{ .self = self, .allocator = allocator } };
         }
 
-        /// Same as `append` except it returns the number of bytes written, which is always the same
-        /// as `m.len`. The purpose of this function existing is to match `std.io.Writer` API.
+        /// Same as `append` except it returns the number of bytes written,
+        /// which is always the same as `m.len`. The purpose of this function
+        /// existing is to match `std.io.Writer` API.
         /// Invalidates element pointers if additional memory is needed.
         fn appendWrite(context: WriterContext, m: []const u8) Allocator.Error!usize {
             try context.self.appendSlice(context.allocator, m);
             return m.len;
         }
 
+        pub const WriterAssumeCapacity = std.io.Writer(*Self, error{}, appendWriteAssumeCapacity);
+
+        /// Initializes a Writer which will append to the list, asserting the
+        /// list can hold the additional bytes.
+        pub fn writerAssumeCapacity(self: *Self) WriterAssumeCapacity {
+            return .{ .context = self };
+        }
+
+        /// Same as `appendSliceAssumeCapacity` except it returns the number of bytes written,
+        /// which is always the same as `m.len`. The purpose of this function
+        /// existing is to match `std.io.Writer` API.
+        fn appendWriteAssumeCapacity(self: *Self, m: []const u8) error{}!usize {
+            self.appendSliceAssumeCapacity(m);
+            return m.len;
+        }
+
         /// Append a value to the list `n` times.
         /// Allocates more memory as necessary.
         /// Invalidates element pointers if additional memory is needed.