From 9e78dd03cbfd66c11d93669b193f79b11606e6e6 Mon Sep 17 00:00:00 2001
From: Andrii Savytskyi <contact@qroc.pro>
Date: Mon, 30 Dec 2024 10:39:53 +0200
Subject: [PATCH] feat: impl `WriteExt` for `&mut BytesMut`

---
 src/writer.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/writer.rs b/src/writer.rs
index 1418b5c..a3d5c41 100644
--- a/src/writer.rs
+++ b/src/writer.rs
@@ -114,6 +114,24 @@ impl WriteExt for Writer<BytesMut> {
     }
 }
 
+impl WriteExt for Writer<&mut BytesMut> {
+    #[inline(always)]
+    unsafe fn flush_len(&mut self, additional: usize) -> io::Result<()> {
+        let new_len = self.get_ref().len() + additional;
+        self.get_mut().set_len(new_len);
+        Ok(())
+    }
+
+    #[inline(always)]
+    fn reserve_with(&mut self, additional: usize) -> io::Result<&mut [MaybeUninit<u8>]> {
+        self.get_mut().reserve(additional);
+        unsafe {
+            let ptr = self.get_mut().as_mut_ptr().add(self.get_ref().len()) as *mut MaybeUninit<u8>;
+            Ok(from_raw_parts_mut(ptr, additional))
+        }
+    }
+}
+
 impl<W: WriteExt + ?Sized> WriteExt for IoBufWriter<W> {
     fn reserve_with(&mut self, additional: usize) -> io::Result<&mut [MaybeUninit<u8>]> {
         self.get_mut().reserve_with(additional)