Merge pull request #216 from marmeladema/std-io-write

Implement `std::io::Write` for `Vec<'bump, u8>`

Cargo.toml

@@ -46,6 +46,7 @@ default = []
 collections = []
 boxed = []
 allocator_api = []
+std = []
 
 # [profile.bench]
 # debug = true

README.md

@@ -155,7 +155,14 @@ in its space itself.
 
 ### `#![no_std]` Support
 
-Bumpalo is a `no_std` crate. It depends only on the `alloc` and `core` crates.
+Bumpalo is a `no_std` crate by default. It depends only on the `alloc` and `core` crates.
+
+### `std` Support
+
+You can optionally decide to enable the `std` feature in order to enable some
+std only trait implementations for some collections:
+
+* `std::io::Write` for `Vec<'bump, u8>`
 
 ### Thread support
 

src/collections/vec.rs

@@ -104,6 +104,8 @@ use core::ops::{Index, IndexMut, RangeBounds};
 use core::ptr;
 use core::ptr::NonNull;
 use core::slice;
+#[cfg(feature = "std")]
+use std::io;
 
 unsafe fn arith_offset<T>(p: *const T, offset: isize) -> *const T {
     p.offset(offset)
@@ -2612,3 +2614,23 @@ where
         }
     }
 }
+
+#[cfg(feature = "std")]
+impl<'bump> io::Write for Vec<'bump, u8> {
+    #[inline]
+    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
+        self.extend_from_slice(buf);
+        Ok(buf.len())
+    }
+
+    #[inline]
+    fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
+        self.extend_from_slice(buf);
+        Ok(())
+    }
+
+    #[inline]
+    fn flush(&mut self) -> io::Result<()> {
+        Ok(())
+    }
+}

src/lib.rs

@@ -1,7 +1,7 @@
 #![doc = include_str!("../README.md")]
 #![deny(missing_debug_implementations)]
 #![deny(missing_docs)]
-#![no_std]
+#![cfg_attr(not(feature = "std"), no_std)]
 #![cfg_attr(feature = "allocator_api", feature(allocator_api))]
 
 #[doc(hidden)]

tests/all/vec.rs

@@ -105,3 +105,30 @@ fn test_vec_items_get_dropped() {
     }
     assert_eq!("Dropped!Dropped!", buffer.borrow().deref());
 }
+
+#[cfg(feature = "std")]
+#[test]
+fn test_vec_write() {
+    use std::io::Write;
+
+    let b = Bump::new();
+    let mut v = bumpalo::vec![in &b];
+
+    assert_eq!(v.write(&[]).unwrap(), 0);
+
+    v.flush().unwrap();
+
+    assert_eq!(v.write(&[1]).unwrap(), 1);
+
+    v.flush().unwrap();
+
+    v.write_all(&[]).unwrap();
+
+    v.flush().unwrap();
+
+    v.write_all(&[2, 3]).unwrap();
+
+    v.flush().unwrap();
+
+    assert_eq!(v, &[1, 2, 3]);
+}