Impl container traits on Cow

Author: YuhanLiin

micropb/src/container.rs

@@ -353,7 +353,7 @@ mod impl_heapless {
 mod impl_alloc {
     use super::*;
 
-    use alloc::{collections::BTreeMap, string::String, vec::Vec};
+    use alloc::{borrow::Cow, collections::BTreeMap, string::String, vec::Vec};
 
     impl PbBytes for Vec<u8> {}
     impl PbString for Vec<u8> {
@@ -378,22 +378,28 @@ mod impl_alloc {
         }
     }
 
-    //impl<'a, T> PbContainer for alloc::borrow::Cow<'a, [T]>
-    //where
-    //[T]: ToOwned<Owned = Vec<T>>,
-    //{
-    //unsafe fn pb_set_len(&mut self, len: usize) {
-    //self.to_mut().set_len(len);
-    //}
+    impl PbBytes for Cow<'_, [u8]> {}
+    impl PbString for Cow<'_, [u8]> {
+        #[inline]
+        unsafe fn pb_set_len(&mut self, len: usize) {
+            self.to_mut().set_len(len);
+        }
 
-    //fn pb_clear(&mut self) {
-    //self.to_mut().clear()
-    //}
+        #[inline]
+        fn pb_clear(&mut self) {
+            self.to_mut().clear()
+        }
 
-    //fn pb_reserve(&mut self, additional: usize) {
-    //self.to_mut().reserve(additional)
-    //}
-    //}
+        #[inline]
+        fn pb_reserve(&mut self, additional: usize) {
+            self.to_mut().reserve(additional)
+        }
+
+        #[inline]
+        fn pb_spare_cap(&mut self) -> &mut [MaybeUninit<u8>] {
+            self.to_mut().spare_capacity_mut()
+        }
+    }
 
     impl PbString for String {
         #[inline]
@@ -418,19 +424,28 @@ mod impl_alloc {
         }
     }
 
-    //impl<'a> PbContainer for alloc::borrow::Cow<'a, str> {
-    //unsafe fn pb_set_len(&mut self, len: usize) {
-    //self.to_mut().as_mut_vec().set_len(len);
-    //}
+    impl PbString for Cow<'_, str> {
+        #[inline]
+        unsafe fn pb_set_len(&mut self, len: usize) {
+            self.to_mut().as_mut_vec().set_len(len);
+        }
 
-    //fn pb_clear(&mut self) {
-    //self.to_mut().clear()
-    //}
+        #[inline]
+        fn pb_clear(&mut self) {
+            self.to_mut().clear()
+        }
+
+        #[inline]
+        fn pb_reserve(&mut self, additional: usize) {
+            self.to_mut().reserve(additional)
+        }
 
-    //fn pb_reserve(&mut self, additional: usize) {
-    //self.to_mut().reserve(additional)
-    //}
-    //}
+        #[inline]
+        fn pb_spare_cap(&mut self) -> &mut [MaybeUninit<u8>] {
+            // SAFETY: spare_capacity_mut() is a safe call, since it doesn't change any bytes
+            unsafe { self.to_mut().as_mut_vec() }.spare_capacity_mut()
+        }
+    }
 
     impl<T> PbVec<T> for Vec<T> {
         #[inline]
@@ -440,26 +455,15 @@ mod impl_alloc {
         }
     }
 
-    //impl<'a, T> PbVec<T> for alloc::borrow::Cow<'a, [T]>
-    //where
-    //[T]: ToOwned<Owned = Vec<T>>,
-    //{
-    //fn pb_push(&mut self, elem: T) -> Result<(), ()> {
-    //self.to_mut().push(elem);
-    //Ok(())
-    //}
-
-    //fn pb_spare_cap(&mut self) -> &mut [MaybeUninit<T>] {
-    //self.to_mut().spare_capacity_mut()
-    //}
-    //}
-
-    //impl<'a> PbString for alloc::borrow::Cow<'a, str> {
-    //fn pb_spare_cap(&mut self) -> &mut [MaybeUninit<u8>] {
-    // // SAFETY: spare_capacity_mut() is a safe call, since it doesn't change any bytes
-    //unsafe { self.to_mut().as_mut_vec().spare_capacity_mut() }
-    //}
-    //}
+    impl<T> PbVec<T> for Cow<'_, [T]>
+    where
+        [T]: ToOwned<Owned = Vec<T>>,
+    {
+        fn pb_push(&mut self, elem: T) -> Result<(), ()> {
+            self.to_mut().push(elem);
+            Ok(())
+        }
+    }
 
     impl<K: Ord, V> PbMap<K, V> for BTreeMap<K, V> {
         #[inline]