std.ArrayList: add writerAssumeCapacity

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

1 files changed, 19 insertions, 2 deletions

diff --git a/lib/std/array_list.zig b/lib/std/array_list.zig
index 79ed5c192c..e5fcb8f4fa 100644
--- 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.