Add support for defmt

Author: darkwater

Cargo.toml

@@ -22,6 +22,7 @@ extract_if = []
 bytes = { version = "1", optional = true, default-features = false }
 serde = { version = "1", optional = true, default-features = false }
 malloc_size_of = { version = "0.1.1", optional = true, default-features = false }
+defmt = { version = "1", optional = true }
 
 [dev-dependencies]
 bincode = "1.0.1"

src/lib.rs

@@ -105,6 +105,7 @@ use serde::{
 use std::io;
 
 /// Error type for APIs with fallible heap allocation
+#[cfg_attr(feature = "defmt", derive(defmt::Format))]
 #[derive(Debug)]
 pub enum CollectionAllocErr {
     /// Overflow `usize::MAX` or other error during size computation
@@ -562,6 +563,18 @@ where
     }
 }
 
+#[cfg(all(feature = "extract_if", feature = "defmt"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "defmt")))]
+impl<T, const N: usize, F> defmt::Format for ExtractIf<'_, T, N, F>
+where
+    F: FnMut(&mut T) -> bool,
+    T: defmt::Format,
+{
+    fn format(&self, fmt: defmt::Formatter) {
+        defmt::write!(fmt, "ExtractIf({:?})", self.vec.as_slice());
+    }
+}
+
 #[cfg(feature = "extract_if")]
 impl<T, F, const N: usize> Iterator for ExtractIf<'_, T, N, F>
 where
@@ -638,6 +651,17 @@ where
     }
 }
 
+#[cfg(feature = "defmt")]
+impl<'a, I, const N: usize> defmt::Format for Splice<'a, I, N>
+where
+    I: Iterator + 'a,
+    <I as Iterator>::Item: defmt::Format,
+{
+    fn format(&self, fmt: defmt::Formatter) {
+        defmt::write!(fmt, "Splice({:?})", self.drain);
+    }
+}
+
 impl<I: Iterator, const N: usize> Iterator for Splice<'_, I, N> {
     type Item = I::Item;
 
@@ -2851,6 +2875,33 @@ impl<T: Debug, const N: usize> Debug for Drain<'_, T, N> {
     }
 }
 
+#[cfg(feature = "defmt")]
+#[cfg_attr(docsrs, doc(cfg(feature = "defmt")))]
+impl<T: defmt::Format, const N: usize> defmt::Format for SmallVec<T, N> {
+    fn format(&self, fmt: defmt::Formatter) {
+        defmt::write!(fmt, "{=[?]}", self.as_slice());
+    }
+}
+
+#[cfg(feature = "defmt")]
+#[cfg_attr(docsrs, doc(cfg(feature = "defmt")))]
+impl<T: defmt::Format, const N: usize> defmt::Format for IntoIter<T, N> {
+    fn format(&self, fmt: defmt::Formatter) {
+        defmt::write!(fmt, "IntoIter({=[?]})", self.as_slice());
+    }
+}
+
+#[cfg(feature = "defmt")]
+#[cfg_attr(docsrs, doc(cfg(feature = "defmt")))]
+impl<T: defmt::Format, const N: usize> defmt::Format for Drain
+where
+    T: defmt::Format,
+{
+    fn format(&self, fmt: defmt::Formatter) {
+        defmt::write!(fmt, "Drain({=[?]})", self.iter.as_slice());
+    }
+}
+
 #[cfg(feature = "serde")]
 #[cfg_attr(docsrs, doc(cfg(feature = "serde")))]
 impl<T, const N: usize> Serialize for SmallVec<T, N>