lib/picotoml: use heapless::vec to allow usage without alloc (#1)

Signed-off-by: Junxing Zhu <[email protected]>
rustsbi · Feb 10, 2025 · 478750c · 478750c
2 parents cf06989 + 449cf7d
commit 478750c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/lib/picotoml/Cargo.toml b/lib/picotoml/Cargo.toml
@@ -9,7 +9,7 @@ edition = "2021"
 [dependencies]
 logos = { version = "0.13.0", default-features = false, features = ["export_derive"] }
 serde = { version = "1.0.188", default-features = false }
-arrayvec = "0.7.6"
+heapless = "0.8.0"
 
 [features]
 default = []

diff --git a/lib/picotoml/src/peeking.rs b/lib/picotoml/src/peeking.rs
@@ -15,7 +15,7 @@ pub struct PeekingIterator<I: Iterator> {
     #[cfg(feature = "alloc")]
     queue: alloc::vec::Vec<Option<I::Item>>,
     #[cfg(not(feature = "alloc"))]
-    queue: arrayvec::ArrayVec<Option<I::Item>, DEFAULT_STACK_SIZE>,
+    queue: heapless::Vec<Option<I::Item>, DEFAULT_STACK_SIZE>,
 
     /// The cursor points to the element we are currently peeking at.
     ///
@@ -85,7 +85,10 @@ impl<I: Iterator> PeekingIterator<I> {
     #[inline]
     fn push_next_to_queue(&mut self) {
         let item = self.iterator.next();
+        #[cfg(feature = "alloc")]
         self.queue.push(item);
+        #[cfg(not(feature = "alloc"))]
+        self.queue.push(item).ok();
     }
 
     /// Increment the cursor which points to the current peekable item.