|
|
|
|
@ -9,12 +9,12 @@ use std::{
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
|
pub struct StackString<const N: usize> {
|
|
|
|
|
pub struct InlineString<const N: usize> {
|
|
|
|
|
buf: [u8; N],
|
|
|
|
|
len: u8,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> StackString<N> {
|
|
|
|
|
impl<const N: usize> InlineString<N> {
|
|
|
|
|
const MAX_LEN: u8 = {
|
|
|
|
|
#[allow(clippy::cast_possible_truncation, clippy::checked_conversions)]
|
|
|
|
|
if N <= u8::MAX as usize {
|
|
|
|
|
@ -24,7 +24,7 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
/// Creates a new `StackString` from a given byte buffer and length.
|
|
|
|
|
/// Creates a new `InlineString` from a given byte buffer and length.
|
|
|
|
|
///
|
|
|
|
|
/// # Safety
|
|
|
|
|
///
|
|
|
|
|
@ -35,7 +35,7 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
Self { buf, len }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Returns a new empty `StackString`.
|
|
|
|
|
/// Returns a new empty `InlineString`.
|
|
|
|
|
#[inline]
|
|
|
|
|
#[must_use]
|
|
|
|
|
pub const fn empty() -> Self {
|
|
|
|
|
@ -46,7 +46,7 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
pub fn new<S>(s: &S) -> Result<Self, StackStringError>
|
|
|
|
|
pub fn new<S>(s: &S) -> Result<Self, InlineStringError>
|
|
|
|
|
where
|
|
|
|
|
S: AsRef<str> + ?Sized,
|
|
|
|
|
{
|
|
|
|
|
@ -55,11 +55,11 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
let s = <S as AsRef<str>>::as_ref(s).as_bytes();
|
|
|
|
|
|
|
|
|
|
// If the length of the string is greater than `Self::MAX_LEN`, it will not fit in the
|
|
|
|
|
// stack buffer so return `None`.
|
|
|
|
|
// buffer so return `None`.
|
|
|
|
|
let len = u8::try_from(s.len())
|
|
|
|
|
.ok()
|
|
|
|
|
.and_then(|len| (len <= Self::MAX_LEN).then_some(len))
|
|
|
|
|
.ok_or(StackStringError {
|
|
|
|
|
.ok_or(InlineStringError {
|
|
|
|
|
max_len: N,
|
|
|
|
|
actual_len: s.len(),
|
|
|
|
|
})?;
|
|
|
|
|
@ -76,24 +76,24 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
/// Returns a string slice pointing to the underlying string data.
|
|
|
|
|
pub fn as_str(&self) -> &str {
|
|
|
|
|
// SAFETY:
|
|
|
|
|
// `len` being less than or equal to `N` is an invariant of `StackString`, so it is
|
|
|
|
|
// `len` being less than or equal to `N` is an invariant of `InlineString`, so it is
|
|
|
|
|
// always within the bounds of `buf`.
|
|
|
|
|
let slice = unsafe { self.buf.get_unchecked(..usize::from(self.len)) };
|
|
|
|
|
|
|
|
|
|
// SAFETY:
|
|
|
|
|
// The first `len` bytes of `buf` being valid UTF-8 is an invariant of `StackString`.
|
|
|
|
|
// The first `len` bytes of `buf` being valid UTF-8 is an invariant of `InlineString`.
|
|
|
|
|
unsafe { str::from_utf8_unchecked(slice) }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Returns a mutable string slice pointing to the underlying string data.
|
|
|
|
|
pub fn as_str_mut(&mut self) -> &mut str {
|
|
|
|
|
// SAFETY:
|
|
|
|
|
// `len` being less than or equal to `N` is an invariant of `StackString`, so it is
|
|
|
|
|
// `len` being less than or equal to `N` is an invariant of `InlineString`, so it is
|
|
|
|
|
// always within the bounds of `buf`.
|
|
|
|
|
let slice = unsafe { self.buf.get_unchecked_mut(..usize::from(self.len)) };
|
|
|
|
|
|
|
|
|
|
// SAFETY:
|
|
|
|
|
// The first `len` bytes of `buf` being valid UTF-8 is an invariant of `StackString`.
|
|
|
|
|
// The first `len` bytes of `buf` being valid UTF-8 is an invariant of `InlineString`.
|
|
|
|
|
unsafe { str::from_utf8_unchecked_mut(slice) }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -110,14 +110,14 @@ impl<const N: usize> StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> Default for StackString<N> {
|
|
|
|
|
impl<const N: usize> Default for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn default() -> Self {
|
|
|
|
|
Self::empty()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> ops::Deref for StackString<N> {
|
|
|
|
|
impl<const N: usize> ops::Deref for InlineString<N> {
|
|
|
|
|
type Target = str;
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
@ -126,43 +126,43 @@ impl<const N: usize> ops::Deref for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> ops::DerefMut for StackString<N> {
|
|
|
|
|
impl<const N: usize> ops::DerefMut for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn deref_mut(&mut self) -> &mut Self::Target {
|
|
|
|
|
self.as_str_mut()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> AsRef<str> for StackString<N> {
|
|
|
|
|
impl<const N: usize> AsRef<str> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn as_ref(&self) -> &str {
|
|
|
|
|
self
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> AsMut<str> for StackString<N> {
|
|
|
|
|
impl<const N: usize> AsMut<str> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn as_mut(&mut self) -> &mut str {
|
|
|
|
|
self
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> borrow::Borrow<str> for StackString<N> {
|
|
|
|
|
impl<const N: usize> borrow::Borrow<str> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn borrow(&self) -> &str {
|
|
|
|
|
self
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> borrow::BorrowMut<str> for StackString<N> {
|
|
|
|
|
impl<const N: usize> borrow::BorrowMut<str> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn borrow_mut(&mut self) -> &mut str {
|
|
|
|
|
self
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<'a, const N: usize> TryFrom<&'a str> for StackString<N> {
|
|
|
|
|
type Error = StackStringError;
|
|
|
|
|
impl<'a, const N: usize> TryFrom<&'a str> for InlineString<N> {
|
|
|
|
|
type Error = InlineStringError;
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
fn try_from(s: &'a str) -> Result<Self, Self::Error> {
|
|
|
|
|
@ -170,8 +170,8 @@ impl<'a, const N: usize> TryFrom<&'a str> for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> TryFrom<String> for StackString<N> {
|
|
|
|
|
type Error = StackStringError;
|
|
|
|
|
impl<const N: usize> TryFrom<String> for InlineString<N> {
|
|
|
|
|
type Error = InlineStringError;
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
fn try_from(s: String) -> Result<Self, Self::Error> {
|
|
|
|
|
@ -179,8 +179,8 @@ impl<const N: usize> TryFrom<String> for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<'a, const N: usize> TryFrom<Cow<'a, str>> for StackString<N> {
|
|
|
|
|
type Error = StackStringError;
|
|
|
|
|
impl<'a, const N: usize> TryFrom<Cow<'a, str>> for InlineString<N> {
|
|
|
|
|
type Error = InlineStringError;
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
fn try_from(s: Cow<'a, str>) -> Result<Self, Self::Error> {
|
|
|
|
|
@ -188,45 +188,45 @@ impl<'a, const N: usize> TryFrom<Cow<'a, str>> for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> From<StackString<N>> for String {
|
|
|
|
|
impl<const N: usize> From<InlineString<N>> for String {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn from(s: StackString<N>) -> Self {
|
|
|
|
|
fn from(s: InlineString<N>) -> Self {
|
|
|
|
|
s.into_string()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize, const M: usize> PartialEq<StackString<M>> for StackString<N> {
|
|
|
|
|
impl<const N: usize, const M: usize> PartialEq<InlineString<M>> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn eq(&self, other: &StackString<M>) -> bool {
|
|
|
|
|
fn eq(&self, other: &InlineString<M>) -> bool {
|
|
|
|
|
**self == **other
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> Eq for StackString<N> {}
|
|
|
|
|
impl<const N: usize> Eq for InlineString<N> {}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize, const M: usize> PartialOrd<StackString<M>> for StackString<N> {
|
|
|
|
|
impl<const N: usize, const M: usize> PartialOrd<InlineString<M>> for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn partial_cmp(&self, other: &StackString<M>) -> Option<Ordering> {
|
|
|
|
|
fn partial_cmp(&self, other: &InlineString<M>) -> Option<Ordering> {
|
|
|
|
|
(**self).partial_cmp(&**other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> Ord for StackString<N> {
|
|
|
|
|
impl<const N: usize> Ord for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn cmp(&self, other: &Self) -> Ordering {
|
|
|
|
|
(**self).cmp(&**other)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> Hash for StackString<N> {
|
|
|
|
|
impl<const N: usize> Hash for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn hash<H: Hasher>(&self, state: &mut H) {
|
|
|
|
|
(**self).hash(state);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> str::FromStr for StackString<N> {
|
|
|
|
|
type Err = StackStringError;
|
|
|
|
|
impl<const N: usize> str::FromStr for InlineString<N> {
|
|
|
|
|
type Err = InlineStringError;
|
|
|
|
|
|
|
|
|
|
#[inline]
|
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
|
|
|
@ -234,14 +234,14 @@ impl<const N: usize> str::FromStr for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> fmt::Debug for StackString<N> {
|
|
|
|
|
impl<const N: usize> fmt::Debug for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
|
fmt::Debug::fmt(&**self, f)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl<const N: usize> fmt::Display for StackString<N> {
|
|
|
|
|
impl<const N: usize> fmt::Display for InlineString<N> {
|
|
|
|
|
#[inline]
|
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
|
fmt::Display::fmt(&**self, f)
|
|
|
|
|
@ -249,12 +249,12 @@ impl<const N: usize> fmt::Display for StackString<N> {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[derive(Debug)]
|
|
|
|
|
pub struct StackStringError {
|
|
|
|
|
pub struct InlineStringError {
|
|
|
|
|
max_len: usize,
|
|
|
|
|
actual_len: usize,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl StackStringError {
|
|
|
|
|
impl InlineStringError {
|
|
|
|
|
pub fn max_len(&self) -> usize {
|
|
|
|
|
self.max_len
|
|
|
|
|
}
|
|
|
|
|
@ -264,15 +264,15 @@ impl StackStringError {
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl fmt::Display for StackStringError {
|
|
|
|
|
impl fmt::Display for InlineStringError {
|
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
|
|
|
write!(
|
|
|
|
|
f,
|
|
|
|
|
"string of length {} exceeds limit for `StackString<{}>`",
|
|
|
|
|
"string of length {} exceeds limit for `InlineString<{}>`",
|
|
|
|
|
self.actual_len,
|
|
|
|
|
self.max_len
|
|
|
|
|
)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl error::Error for StackStringError {}
|
|
|
|
|
impl error::Error for InlineStringError {}
|