Reduce allocations on string.

yewstack · Aug 12, 2022 · 3ac02fd · 3ac02fd
1 parent a470862
commit 3ac02fd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/yew/src/platform/fmt.rs b/packages/yew/src/platform/fmt.rs
@@ -132,7 +132,10 @@ impl Stream for BufStream {
         let mut inner = self.inner.borrow_mut();
 
         if !inner.buf.is_empty() {
-            return Poll::Ready(Some(inner.buf.split_off(0)));
+            let mut buf = String::new();
+            std::mem::swap(&mut buf, &mut inner.buf);
+
+            return Poll::Ready(Some(buf));
         }
 
         if let BufStreamState::Done = inner.state {
@@ -189,6 +192,7 @@ where
 {
     type Item = String;
 
+    #[inline]
     fn poll_next(
         self: std::pin::Pin<&mut Self>,
         cx: &mut std::task::Context<'_>,