blob: 2a330c343d2ea5fc17dfb2689799e0c12b0d360d (
plain)
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
|
use std::{sync::Arc};
const INLINE_CAP: usize = 22;
#[derive(Clone, Debug)]
pub(crate) enum SmolStr {
Heap(Arc<str>),
Inline {
len: u8,
buf: [u8; INLINE_CAP],
},
}
impl SmolStr {
pub fn new(text: &str) -> SmolStr {
let len = text.len();
if len <= INLINE_CAP {
let mut buf = [0; INLINE_CAP];
buf[..len].copy_from_slice(text.as_bytes());
SmolStr::Inline { len: len as u8, buf }
} else {
SmolStr::Heap(
text.to_string().into_boxed_str().into()
)
}
}
pub fn as_str(&self) -> &str {
match self {
SmolStr::Heap(data) => &*data,
SmolStr::Inline { len, buf } => {
let len = *len as usize;
let buf = &buf[..len];
unsafe { ::std::str::from_utf8_unchecked(buf) }
}
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
#[cfg(target_pointer_width = "64")]
fn smol_str_is_smol() {
assert_eq!(::std::mem::size_of::<SmolStr>(), 8 + 8 + 8)
}
#[test]
fn test_round_trip() {
let mut text = String::new();
for n in 0..256 {
let smol = SmolStr::new(&text);
assert_eq!(smol.as_str(), text.as_str());
text.push_str(&n.to_string());
}
}
}
|
