use crate::*;
use std::{
collections::HashMap,
hash::{Hash, Hasher},
};
#[derive(Clone, Debug, Default, PartialEq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "facet", derive(Facet))]
#[cfg_attr(feature = "facet", facet(opaque))]
#[cfg_attr(feature = "rkyv", derive(Archive, RkyvSerialize, RkyvDeserialize))]
pub struct TokenValueMap {
attributes: HashMap<Token, Value>,
}
impl Hash for TokenValueMap {
fn hash<H: Hasher>(&self, state: &mut H) {
self.attributes.len().hash(state);
let mut items: Vec<_> = self.attributes.iter().collect();
items.sort_by_key(|(k, _)| *k);
for (token, value) in items {
token.hash(state);
value.hash(state);
}
}
}
impl TokenValueMap {
pub fn new() -> Self {
Self::default()
}
pub fn with_capacity(capacity: usize) -> Self {
Self {
attributes: HashMap::with_capacity(capacity),
}
}
pub fn insert<V: Into<Value>>(&mut self, token: impl Into<Token>, value: V) {
self.attributes.insert(token.into(), value.into());
}
pub fn get(&self, token: &Token) -> Option<&Value> {
self.attributes.get(token)
}
pub fn get_mut(&mut self, token: &Token) -> Option<&mut Value> {
self.attributes.get_mut(token)
}
pub fn remove(&mut self, token: &Token) -> Option<Value> {
self.attributes.remove(token)
}
pub fn contains(&self, token: &Token) -> bool {
self.attributes.contains_key(token)
}
pub fn len(&self) -> usize {
self.attributes.len()
}
pub fn is_empty(&self) -> bool {
self.attributes.is_empty()
}
pub fn clear(&mut self) {
self.attributes.clear();
}
pub fn iter(&self) -> impl Iterator<Item = (&Token, &Value)> {
self.attributes.iter()
}
pub fn iter_mut(&mut self) -> impl Iterator<Item = (&Token, &mut Value)> {
self.attributes.iter_mut()
}
pub fn tokens(&self) -> impl Iterator<Item = &Token> {
self.attributes.keys()
}
pub fn values(&self) -> impl Iterator<Item = &Value> {
self.attributes.values()
}
pub fn values_mut(&mut self) -> impl Iterator<Item = &mut Value> {
self.attributes.values_mut()
}
pub fn extend<I>(&mut self, iter: I)
where
I: IntoIterator<Item = (Token, Value)>,
{
self.attributes.extend(iter);
}
pub fn retain<F>(&mut self, mut f: F)
where
F: FnMut(&Token, &Value) -> bool,
{
self.attributes.retain(|k, v| f(k, v));
}
}
impl Eq for TokenValueMap {}
impl FromIterator<(Token, Value)> for TokenValueMap {
fn from_iter<T: IntoIterator<Item = (Token, Value)>>(iter: T) -> Self {
Self {
attributes: HashMap::from_iter(iter),
}
}
}
impl<const N: usize> From<[(Token, Value); N]> for TokenValueMap {
fn from(arr: [(Token, Value); N]) -> Self {
Self {
attributes: HashMap::from(arr),
}
}
}