1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
use bytes::Bytes; use std::{ops, str}; #[derive(Debug, Clone, Eq, PartialEq, Ord, PartialOrd, Hash)] pub(crate) struct ByteStr { bytes: Bytes, } impl ByteStr { #[inline] pub fn new() -> ByteStr { ByteStr { bytes: Bytes::new(), } } #[inline] pub fn from_static(val: &'static str) -> ByteStr { ByteStr { bytes: Bytes::from_static(val.as_bytes()), } } #[inline] pub unsafe fn from_utf8_unchecked(bytes: Bytes) -> ByteStr { if cfg!(debug_assertions) { match str::from_utf8(&bytes) { Ok(_) => (), Err(err) => panic!( "ByteStr::from_utf8_unchecked() with invalid bytes; error = {}, bytes = {:?}", err, bytes ), } } ByteStr { bytes: bytes } } } impl ops::Deref for ByteStr { type Target = str; #[inline] fn deref(&self) -> &str { let b: &[u8] = self.bytes.as_ref(); unsafe { str::from_utf8_unchecked(b) } } } impl From<String> for ByteStr { #[inline] fn from(src: String) -> ByteStr { ByteStr { bytes: Bytes::from(src), } } } impl<'a> From<&'a str> for ByteStr { #[inline] fn from(src: &'a str) -> ByteStr { ByteStr { bytes: Bytes::copy_from_slice(src.as_bytes()), } } } impl From<ByteStr> for Bytes { fn from(src: ByteStr) -> Self { src.bytes } }