Specialize single-element writes to buffer
copy_from_slice generally falls back to memcpy/memmove, which is much more expensive than we need to write a single element in. This saves 0.26% instructions on the diesel benchmark.
This commit is contained in:
parent
9f75dbfa69
commit
299ac75894
@ -91,6 +91,21 @@ pub(super) fn extend_from_slice(&mut self, xs: &[T]) {
|
||||
let b = self.take();
|
||||
*self = (b.extend_from_slice)(b, Slice::from(xs));
|
||||
}
|
||||
|
||||
pub(super) fn push(&mut self, v: T) {
|
||||
// Fast path to avoid going through an FFI call.
|
||||
if let Some(final_len) = self.len.checked_add(1) {
|
||||
if final_len <= self.capacity {
|
||||
unsafe {
|
||||
*self.data.add(self.len) = v;
|
||||
}
|
||||
self.len = final_len;
|
||||
return;
|
||||
}
|
||||
}
|
||||
let b = self.take();
|
||||
*self = (b.extend_from_slice)(b, Slice::from(std::slice::from_ref(&v)));
|
||||
}
|
||||
}
|
||||
|
||||
impl Write for Buffer<u8> {
|
||||
|
@ -114,7 +114,7 @@ fn decode(_: &mut Reader<'_>, _: &mut S) -> Self {}
|
||||
|
||||
impl<S> Encode<S> for u8 {
|
||||
fn encode(self, w: &mut Writer, _: &mut S) {
|
||||
w.write_all(&[self]).unwrap();
|
||||
w.push(self);
|
||||
}
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user