use std::collections::HashMap;
use std::collections::hash_map::RandomState;
use std::convert::TryFrom;
use std::hash::{BuildHasher, Hash, Hasher};
use std::iter::{FromIterator, FusedIterator};
use std::marker::PhantomData;
use std::{fmt, mem, ops, ptr, vec};
use crate::Error;
use super::HeaderValue;
use super::name::{HdrName, HeaderName, InvalidHeaderName};
pub use self::as_header_name::AsHeaderName;
pub use self::into_header_name::IntoHeaderName;
#[derive(Clone)]
pub struct HeaderMap<T = HeaderValue> {
mask: Size,
indices: Box<[Pos]>,
entries: Vec<Bucket<T>>,
extra_values: Vec<ExtraValue<T>>,
order: Vec<Option<Link>>,
order_holes: usize,
danger: Danger,
}
#[derive(Debug)]
pub struct Iter<'a, T> {
map: &'a HeaderMap<T>,
entry: usize,
cursor: Option<Cursor>,
}
#[derive(Debug)]
pub struct IterMut<'a, T> {
entries: *mut Bucket<T>,
entries_len: usize,
extra_values: *mut ExtraValue<T>,
entry: usize,
cursor: Option<Cursor>,
lt: PhantomData<&'a mut HeaderMap<T>>,
}
#[derive(Debug)]
pub struct IntoIter<T> {
next: Option<usize>,
entries: vec::IntoIter<Bucket<T>>,
extra_values: Vec<ExtraValue<T>>,
}
#[derive(Debug)]
pub struct OrderedIter<'a, T> {
map: &'a HeaderMap<T>,
index: usize,
}
#[derive(Debug)]
pub struct IntoOrderedIter<T> {
index: usize,
order: Vec<Option<Link>>,
entries: Vec<Bucket<T>>,
extra_values: Vec<ExtraValue<T>>,
}
#[derive(Debug)]
pub struct Keys<'a, T> {
inner: ::std::slice::Iter<'a, Bucket<T>>,
}
#[derive(Debug)]
pub struct Values<'a, T> {
inner: Iter<'a, T>,
}
#[derive(Debug)]
pub struct ValuesMut<'a, T> {
inner: IterMut<'a, T>,
}
#[derive(Debug)]
pub struct Drain<'a, T> {
idx: usize,
len: usize,
entries: *mut [Bucket<T>],
next: Option<usize>,
extra_values: *mut Vec<ExtraValue<T>>,
lt: PhantomData<&'a mut HeaderMap<T>>,
}
#[derive(Debug)]
pub struct GetAll<'a, T> {
map: &'a HeaderMap<T>,
index: Option<usize>,
}
#[derive(Debug)]
pub enum Entry<'a, T: 'a> {
Occupied(OccupiedEntry<'a, T>),
Vacant(VacantEntry<'a, T>),
}
#[derive(Debug)]
pub struct VacantEntry<'a, T> {
map: &'a mut HeaderMap<T>,
key: HeaderName,
hash: HashValue,
probe: usize,
danger: bool,
}
#[derive(Debug)]
pub struct OccupiedEntry<'a, T> {
map: &'a mut HeaderMap<T>,
probe: usize,
index: usize,
}
#[derive(Debug)]
pub struct ValueIter<'a, T> {
map: &'a HeaderMap<T>,
index: usize,
front: Option<Cursor>,
back: Option<Cursor>,
}
#[derive(Debug)]
pub struct ValueIterMut<'a, T> {
entries: *mut Bucket<T>,
extra_values: *mut ExtraValue<T>,
index: usize,
front: Option<Cursor>,
back: Option<Cursor>,
lt: PhantomData<&'a mut HeaderMap<T>>,
}
#[derive(Debug)]
pub struct ValueDrain<'a, T> {
first: Option<T>,
next: Option<::std::vec::IntoIter<T>>,
lt: PhantomData<&'a mut HeaderMap<T>>,
}
pub struct MaxSizeReached {
_priv: (),
}
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
enum Cursor {
Head,
Values(usize),
}
type Size = u16;
const MAX_SIZE: usize = 1 << 15;
#[derive(Copy, Clone)]
struct Pos {
index: Size,
hash: HashValue,
}
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
struct HashValue(u16);
#[derive(Debug, Clone)]
struct Bucket<T> {
hash: HashValue,
key: HeaderName,
value: T,
links: Option<Links>,
order: usize,
}
#[derive(Debug, Copy, Clone)]
struct Links {
next: usize,
tail: usize,
}
#[derive(Debug)]
struct RawLinks<T>(*mut [Bucket<T>]);
#[derive(Debug, Clone)]
struct ExtraValue<T> {
key: HeaderName,
value: T,
prev: Link,
next: Link,
order: usize,
}
#[derive(Debug, Copy, Clone, Eq, PartialEq)]
enum Link {
Entry(usize),
Extra(usize),
}
#[derive(Clone)]
enum Danger {
Green,
Yellow,
Red(RandomState),
}
const DISPLACEMENT_THRESHOLD: usize = 128;
const FORWARD_SHIFT_THRESHOLD: usize = 512;
const LOAD_FACTOR_THRESHOLD: usize = 5;
macro_rules! probe_loop {
($label:tt: $probe_var: ident < $len: expr, $body: expr) => {
debug_assert!($len > 0);
$label:
loop {
if $probe_var < $len {
$body
$probe_var += 1;
} else {
$probe_var = 0;
}
}
};
($probe_var: ident < $len: expr, $body: expr) => {
debug_assert!($len > 0);
loop {
if $probe_var < $len {
$body
$probe_var += 1;
} else {
$probe_var = 0;
}
}
};
}
macro_rules! insert_phase_one {
($map:ident,
$key:expr,
$probe:ident,
$pos:ident,
$hash:ident,
$danger:ident,
$vacant:expr,
$occupied:expr,
$robinhood:expr) =>
{{
let $hash = hash_elem_using(&$map.danger, &$key);
let mut $probe = desired_pos($map.mask, $hash);
let mut dist = 0;
let ret;
probe_loop!('probe: $probe < $map.indices.len(), {
if let Some(($pos, entry_hash)) = $map.indices[$probe].resolve() {
let their_dist = probe_distance($map.mask, entry_hash, $probe);
if their_dist < dist {
let $danger =
dist >= FORWARD_SHIFT_THRESHOLD && !$map.danger.is_red();
ret = $robinhood;
break 'probe;
} else if entry_hash == $hash && $map.entries[$pos].key == $key {
ret = $occupied;
break 'probe;
}
} else {
let $danger =
dist >= FORWARD_SHIFT_THRESHOLD && !$map.danger.is_red();
ret = $vacant;
break 'probe;
}
dist += 1;
});
ret
}}
}
impl HeaderMap {
#[inline]
pub fn new() -> Self {
Self::default()
}
}
impl<T> Default for HeaderMap<T> {
fn default() -> Self {
HeaderMap {
mask: 0,
indices: Box::new([]), entries: Vec::new(),
extra_values: Vec::new(),
order: Vec::new(),
order_holes: 0,
danger: Danger::Green,
}
}
}
impl<T> HeaderMap<T> {
pub fn with_capacity(capacity: usize) -> HeaderMap<T> {
Self::try_with_capacity(capacity).expect("size overflows MAX_SIZE")
}
pub fn try_with_capacity(capacity: usize) -> Result<HeaderMap<T>, MaxSizeReached> {
if capacity == 0 {
Ok(Self::default())
} else {
let raw_cap = to_raw_capacity(capacity)?;
let raw_cap = match raw_cap.checked_next_power_of_two() {
Some(c) => c,
None => return Err(MaxSizeReached { _priv: () }),
};
if raw_cap > MAX_SIZE {
return Err(MaxSizeReached { _priv: () });
}
debug_assert!(raw_cap > 0);
Ok(HeaderMap {
mask: (raw_cap - 1) as Size,
indices: vec![Pos::none(); raw_cap].into_boxed_slice(),
entries: Vec::with_capacity(usable_capacity(raw_cap)),
extra_values: Vec::new(),
order: Vec::with_capacity(capacity),
order_holes: 0,
danger: Danger::Green,
})
}
}
pub fn len(&self) -> usize {
self.entries.len() + self.extra_values.len()
}
pub fn keys_len(&self) -> usize {
self.entries.len()
}
pub fn is_empty(&self) -> bool {
self.entries.len() == 0
}
pub fn clear(&mut self) {
self.entries.clear();
self.extra_values.clear();
self.order.clear();
self.order_holes = 0;
self.danger = Danger::Green;
for e in self.indices.iter_mut() {
*e = Pos::none();
}
}
pub fn capacity(&self) -> usize {
usable_capacity(self.indices.len())
}
pub fn reserve(&mut self, additional: usize) {
self.try_reserve(additional)
.expect("size overflows MAX_SIZE")
}
pub fn try_reserve(&mut self, additional: usize) -> Result<(), MaxSizeReached> {
let cap = self
.entries
.len()
.checked_add(additional)
.ok_or_else(MaxSizeReached::new)?;
let raw_cap = to_raw_capacity(cap)?;
if raw_cap > self.indices.len() {
let raw_cap = raw_cap
.checked_next_power_of_two()
.ok_or_else(MaxSizeReached::new)?;
if raw_cap > MAX_SIZE {
return Err(MaxSizeReached::new());
}
if self.entries.is_empty() {
self.mask = raw_cap as Size - 1;
self.indices = vec![Pos::none(); raw_cap].into_boxed_slice();
let cap = usable_capacity(raw_cap);
self.entries = Vec::with_capacity(cap);
self.order.reserve_exact(cap);
} else {
self.try_grow(raw_cap)?;
}
}
self.order.reserve(additional);
Ok(())
}
pub fn get<K>(&self, key: K) -> Option<&T>
where
K: AsHeaderName,
{
self.get2(&key)
}
fn get2<K>(&self, key: &K) -> Option<&T>
where
K: AsHeaderName,
{
match key.find(self) {
Some((_, found)) => {
let entry = &self.entries[found];
Some(&entry.value)
}
None => None,
}
}
pub fn get_mut<K>(&mut self, key: K) -> Option<&mut T>
where
K: AsHeaderName,
{
match key.find(self) {
Some((_, found)) => {
let entry = &mut self.entries[found];
Some(&mut entry.value)
}
None => None,
}
}
pub fn get_all<K>(&self, key: K) -> GetAll<'_, T>
where
K: AsHeaderName,
{
GetAll {
map: self,
index: key.find(self).map(|(_, i)| i),
}
}
pub fn contains_key<K>(&self, key: K) -> bool
where
K: AsHeaderName,
{
key.find(self).is_some()
}
pub fn iter(&self) -> Iter<'_, T> {
Iter {
map: self,
entry: 0,
cursor: self.entries.first().map(|_| Cursor::Head),
}
}
pub fn ordered_iter(&self) -> OrderedIter<'_, T> {
OrderedIter {
map: self,
index: 0,
}
}
pub fn iter_mut(&mut self) -> IterMut<'_, T> {
IterMut {
entries: self.entries.as_mut_ptr(),
entries_len: self.entries.len(),
extra_values: self.extra_values.as_mut_ptr(),
entry: 0,
cursor: self.entries.first().map(|_| Cursor::Head),
lt: PhantomData,
}
}
pub fn keys(&self) -> Keys<'_, T> {
Keys {
inner: self.entries.iter(),
}
}
pub fn values(&self) -> Values<'_, T> {
Values { inner: self.iter() }
}
pub fn values_mut(&mut self) -> ValuesMut<'_, T> {
ValuesMut {
inner: self.iter_mut(),
}
}
pub fn drain(&mut self) -> Drain<'_, T> {
for i in self.indices.iter_mut() {
*i = Pos::none();
}
let entries = &mut self.entries[..] as *mut _;
let extra_values = &mut self.extra_values as *mut _;
let len = self.entries.len();
unsafe {
self.entries.set_len(0);
}
self.order.clear();
self.order_holes = 0;
Drain {
idx: 0,
len,
entries,
extra_values,
next: None,
lt: PhantomData,
}
}
fn value_iter(&self, idx: Option<usize>) -> ValueIter<'_, T> {
use self::Cursor::*;
if let Some(idx) = idx {
let back = {
let entry = &self.entries[idx];
entry.links.map(|l| Values(l.tail)).unwrap_or(Head)
};
ValueIter {
map: self,
index: idx,
front: Some(Head),
back: Some(back),
}
} else {
ValueIter {
map: self,
index: usize::MAX,
front: None,
back: None,
}
}
}
fn value_iter_mut(&mut self, idx: usize) -> ValueIterMut<'_, T> {
use self::Cursor::*;
let back = {
let entry = &self.entries[idx];
entry.links.map(|l| Values(l.tail)).unwrap_or(Head)
};
ValueIterMut {
entries: self.entries.as_mut_ptr(),
extra_values: self.extra_values.as_mut_ptr(),
index: idx,
front: Some(Head),
back: Some(back),
lt: PhantomData,
}
}
pub fn entry<K>(&mut self, key: K) -> Entry<'_, T>
where
K: IntoHeaderName,
{
key.try_entry(self).expect("size overflows MAX_SIZE")
}
pub fn try_entry<K>(&mut self, key: K) -> Result<Entry<'_, T>, InvalidHeaderName>
where
K: AsHeaderName,
{
key.try_entry(self).map_err(|err| match err {
as_header_name::TryEntryError::InvalidHeaderName(e) => e,
as_header_name::TryEntryError::MaxSizeReached(_e) => {
InvalidHeaderName::new()
}
})
}
fn try_entry2<K>(&mut self, key: K) -> Result<Entry<'_, T>, MaxSizeReached>
where
K: Into<HeaderName>,
{
self.try_reserve_one()?;
let key = key.into();
Ok(insert_phase_one!(
self,
key,
probe,
pos,
hash,
danger,
Entry::Vacant(VacantEntry {
map: self,
hash,
key: key.into(),
probe,
danger,
}),
Entry::Occupied(OccupiedEntry {
map: self,
index: pos,
probe,
}),
Entry::Vacant(VacantEntry {
map: self,
hash,
key: key.into(),
probe,
danger,
})
))
}
pub fn insert<K>(&mut self, key: K, val: T) -> Option<T>
where
K: IntoHeaderName,
{
self.try_insert(key, val).expect("size overflows MAX_SIZE")
}
pub fn try_insert<K>(&mut self, key: K, val: T) -> Result<Option<T>, MaxSizeReached>
where
K: IntoHeaderName,
{
key.try_insert(self, val)
}
#[inline]
fn try_insert2<K>(&mut self, key: K, value: T) -> Result<Option<T>, MaxSizeReached>
where
K: Into<HeaderName>,
{
self.try_reserve_one()?;
let key = key.into();
Ok(insert_phase_one!(
self,
key,
probe,
pos,
hash,
danger,
{
let _ = danger; let index = self.entries.len();
self.try_insert_entry(hash, key.into(), value)?;
self.indices[probe] = Pos::new(index, hash);
None
},
Some(self.insert_occupied(pos, key, value)),
{
self.try_insert_phase_two(key.into(), value, hash, probe, danger)?;
None
}
))
}
#[inline]
fn insert_occupied(&mut self, index: usize, key: HeaderName, value: T) -> T {
let old = self.insert_occupied_same_key(index, value);
self.entries[index].key = key;
old
}
#[inline]
fn insert_occupied_same_key(&mut self, index: usize, value: T) -> T {
if let Some(links) = self.entries[index].links {
self.remove_all_extra_values(links.next);
}
let entry = &mut self.entries[index];
mem::replace(&mut entry.value, value)
}
fn insert_occupied_mult(
&mut self,
index: usize,
key: HeaderName,
value: T,
) -> ValueDrain<'_, T> {
self.insert_occupied_mult_inner(index, Some(key), value)
}
fn insert_occupied_mult_same_key(&mut self, index: usize, value: T) -> ValueDrain<'_, T> {
self.insert_occupied_mult_inner(index, None, value)
}
fn insert_occupied_mult_inner(
&mut self,
index: usize,
key: Option<HeaderName>,
value: T,
) -> ValueDrain<'_, T> {
let old;
let links;
{
let entry = &mut self.entries[index];
old = mem::replace(&mut entry.value, value);
if let Some(key) = key {
entry.key = key;
}
links = entry.links;
}
let next = links.map(|l| self.drain_all_extra_values(l.next).into_iter());
ValueDrain {
first: Some(old),
next,
lt: PhantomData,
}
}
pub fn append<K>(&mut self, key: K, value: T) -> bool
where
K: IntoHeaderName,
{
self.try_append(key, value)
.expect("size overflows MAX_SIZE")
}
pub fn try_append<K>(&mut self, key: K, value: T) -> Result<bool, MaxSizeReached>
where
K: IntoHeaderName,
{
key.try_append(self, value)
}
#[inline]
fn try_append2<K>(&mut self, key: K, value: T) -> Result<bool, MaxSizeReached>
where
K: Into<HeaderName>,
{
self.try_reserve_one()?;
let key = key.into();
Ok(insert_phase_one!(
self,
key,
probe,
pos,
hash,
danger,
{
let _ = danger;
let index = self.entries.len();
self.try_insert_entry(hash, key.into(), value)?;
self.indices[probe] = Pos::new(index, hash);
false
},
{
self.append_value(pos, key, value);
true
},
{
self.try_insert_phase_two(key.into(), value, hash, probe, danger)?;
false
}
))
}
#[inline]
fn find<K>(&self, key: &K) -> Option<(usize, usize)>
where
K: Hash + Into<HeaderName> + ?Sized,
HeaderName: PartialEq<K>,
{
if self.entries.is_empty() {
return None;
}
let hash = hash_elem_using(&self.danger, key);
let mask = self.mask;
let mut probe = desired_pos(mask, hash);
let mut dist = 0;
probe_loop!(probe < self.indices.len(), {
if let Some((i, entry_hash)) = self.indices[probe].resolve() {
if dist > probe_distance(mask, entry_hash, probe) {
return None;
} else if entry_hash == hash && self.entries[i].key == *key {
return Some((probe, i));
}
} else {
return None;
}
dist += 1;
});
}
#[inline]
fn try_insert_phase_two(
&mut self,
key: HeaderName,
value: T,
hash: HashValue,
probe: usize,
danger: bool,
) -> Result<usize, MaxSizeReached> {
let index = self.entries.len();
self.try_insert_entry(hash, key, value)?;
let num_displaced = do_insert_phase_two(&mut self.indices, probe, Pos::new(index, hash));
if danger || num_displaced >= DISPLACEMENT_THRESHOLD {
self.danger.set_yellow();
}
Ok(index)
}
pub fn remove<K>(&mut self, key: K) -> Option<T>
where
K: AsHeaderName,
{
match key.find(self) {
Some((probe, idx)) => {
if let Some(links) = self.entries[idx].links {
self.remove_all_extra_values(links.next);
}
let entry = self.remove_found(probe, idx);
Some(entry.value)
}
None => None,
}
}
#[inline]
fn remove_found(&mut self, probe: usize, found: usize) -> Bucket<T> {
self.remove_order(self.entries[found].order);
self.indices[probe] = Pos::none();
let entry = self.entries.swap_remove(found);
if found < self.entries.len() {
let order = self.entries[found].order;
self.set_order_link(order, Link::Entry(found));
let entry = &self.entries[found];
let mut probe = desired_pos(self.mask, entry.hash);
probe_loop!(probe < self.indices.len(), {
if let Some((i, _)) = self.indices[probe].resolve() {
if i >= self.entries.len() {
self.indices[probe] = Pos::new(found, entry.hash);
break;
}
}
});
if let Some(links) = entry.links {
self.extra_values[links.next].prev = Link::Entry(found);
self.extra_values[links.tail].next = Link::Entry(found);
}
}
if !self.entries.is_empty() {
let mut last_probe = probe;
let mut probe = probe + 1;
probe_loop!(probe < self.indices.len(), {
if let Some((_, entry_hash)) = self.indices[probe].resolve() {
if probe_distance(self.mask, entry_hash, probe) > 0 {
self.indices[last_probe] = self.indices[probe];
self.indices[probe] = Pos::none();
} else {
break;
}
} else {
break;
}
last_probe = probe;
});
}
entry
}
#[inline]
fn remove_extra_value(&mut self, idx: usize) -> ExtraValue<T> {
self.remove_order(self.extra_values[idx].order);
let raw_links = self.raw_links();
let extra = remove_extra_value(raw_links, &mut self.extra_values, idx);
if let Some(moved) = self.extra_values.get(idx) {
let order = moved.order;
self.set_order_link(order, Link::Extra(idx));
}
extra
}
fn remove_all_extra_values(&mut self, mut head: usize) {
loop {
let extra = self.remove_extra_value(head);
if let Link::Extra(idx) = extra.next {
head = idx;
} else {
break;
}
}
}
fn drain_all_extra_values(&mut self, mut head: usize) -> Vec<T> {
let mut vec = Vec::new();
loop {
let extra = self.remove_extra_value(head);
vec.push(extra.value);
if let Link::Extra(idx) = extra.next {
head = idx;
} else {
break;
}
}
vec
}
#[inline]
fn try_insert_entry(
&mut self,
hash: HashValue,
key: HeaderName,
value: T,
) -> Result<(), MaxSizeReached> {
if self.entries.len() >= MAX_SIZE {
return Err(MaxSizeReached::new());
}
let index = self.entries.len();
let order = self.push_order(Link::Entry(index));
self.entries.push(Bucket {
hash,
key,
value,
links: None,
order,
});
Ok(())
}
fn rebuild(&mut self) {
'outer: for (index, entry) in self.entries.iter_mut().enumerate() {
let hash = hash_elem_using(&self.danger, &entry.key);
let mut probe = desired_pos(self.mask, hash);
let mut dist = 0;
entry.hash = hash;
probe_loop!(probe < self.indices.len(), {
if let Some((_, entry_hash)) = self.indices[probe].resolve() {
let their_dist = probe_distance(self.mask, entry_hash, probe);
if their_dist < dist {
break;
}
} else {
self.indices[probe] = Pos::new(index, hash);
continue 'outer;
}
dist += 1;
});
do_insert_phase_two(&mut self.indices, probe, Pos::new(index, hash));
}
}
fn reinsert_entry_in_order(&mut self, pos: Pos) {
if let Some((_, entry_hash)) = pos.resolve() {
let mut probe = desired_pos(self.mask, entry_hash);
probe_loop!(probe < self.indices.len(), {
if self.indices[probe].resolve().is_none() {
self.indices[probe] = pos;
return;
}
});
}
}
fn try_reserve_one(&mut self) -> Result<(), MaxSizeReached> {
let len = self.entries.len();
if self.danger.is_yellow() {
if self.entries.len() * LOAD_FACTOR_THRESHOLD >= self.indices.len() {
self.danger.set_green();
let new_cap = self.indices.len() * 2;
self.try_grow(new_cap)?;
} else {
self.danger.set_red();
for index in self.indices.iter_mut() {
*index = Pos::none();
}
self.rebuild();
}
} else if len == self.capacity() {
if len == 0 {
let new_raw_cap = 8;
self.mask = 8 - 1;
self.indices = vec![Pos::none(); new_raw_cap].into_boxed_slice();
let cap = usable_capacity(new_raw_cap);
self.entries = Vec::with_capacity(cap);
self.order.reserve_exact(cap);
} else {
let raw_cap = self.indices.len();
self.try_grow(raw_cap << 1)?;
}
}
Ok(())
}
#[inline]
fn try_grow(&mut self, new_raw_cap: usize) -> Result<(), MaxSizeReached> {
if new_raw_cap > MAX_SIZE {
return Err(MaxSizeReached::new());
}
let mut first_ideal = 0;
for (i, pos) in self.indices.iter().enumerate() {
if let Some((_, entry_hash)) = pos.resolve() {
if 0 == probe_distance(self.mask, entry_hash, i) {
first_ideal = i;
break;
}
}
}
let old_indices = mem::replace(
&mut self.indices,
vec![Pos::none(); new_raw_cap].into_boxed_slice(),
);
self.mask = new_raw_cap.wrapping_sub(1) as Size;
for &pos in &old_indices[first_ideal..] {
self.reinsert_entry_in_order(pos);
}
for &pos in &old_indices[..first_ideal] {
self.reinsert_entry_in_order(pos);
}
let more = self.capacity() - self.entries.len();
self.entries.reserve_exact(more);
self.order.reserve_exact(more);
Ok(())
}
#[inline]
fn raw_links(&mut self) -> RawLinks<T> {
RawLinks(&mut self.entries[..] as *mut _)
}
#[inline]
fn push_order(&mut self, link: Link) -> usize {
let index = self.order.len();
self.order.push(Some(link));
index
}
#[inline]
fn remove_order(&mut self, index: usize) {
if self.order[index].take().is_some() {
self.order_holes += 1;
}
}
#[inline]
fn set_order_link(&mut self, index: usize, link: Link) {
if let Some(slot) = self.order.get_mut(index)
&& slot.is_some()
{
*slot = Some(link);
}
}
#[inline]
fn append_value(&mut self, entry_idx: usize, key: HeaderName, value: T) {
let idx = self.extra_values.len();
let order = self.push_order(Link::Extra(idx));
let entry = &mut self.entries[entry_idx];
match entry.links {
Some(links) => {
self.extra_values.push(ExtraValue {
key,
value,
prev: Link::Extra(links.tail),
next: Link::Entry(entry_idx),
order,
});
self.extra_values[links.tail].next = Link::Extra(idx);
entry.links = Some(Links { tail: idx, ..links });
}
None => {
self.extra_values.push(ExtraValue {
key,
value,
prev: Link::Entry(entry_idx),
next: Link::Entry(entry_idx),
order,
});
entry.links = Some(Links {
next: idx,
tail: idx,
});
}
}
}
}
#[inline]
fn remove_extra_value<T>(
mut raw_links: RawLinks<T>,
extra_values: &mut Vec<ExtraValue<T>>,
idx: usize,
) -> ExtraValue<T> {
let prev;
let next;
{
debug_assert!(extra_values.len() > idx);
let extra = &extra_values[idx];
prev = extra.prev;
next = extra.next;
}
match (prev, next) {
(Link::Entry(prev), Link::Entry(next)) => {
debug_assert_eq!(prev, next);
raw_links[prev] = None;
}
(Link::Entry(prev), Link::Extra(next)) => {
debug_assert!(raw_links[prev].is_some());
raw_links[prev].as_mut().unwrap().next = next;
debug_assert!(extra_values.len() > next);
extra_values[next].prev = Link::Entry(prev);
}
(Link::Extra(prev), Link::Entry(next)) => {
debug_assert!(raw_links[next].is_some());
raw_links[next].as_mut().unwrap().tail = prev;
debug_assert!(extra_values.len() > prev);
extra_values[prev].next = Link::Entry(next);
}
(Link::Extra(prev), Link::Extra(next)) => {
debug_assert!(extra_values.len() > next);
debug_assert!(extra_values.len() > prev);
extra_values[prev].next = Link::Extra(next);
extra_values[next].prev = Link::Extra(prev);
}
}
let mut extra = extra_values.swap_remove(idx);
let old_idx = extra_values.len();
if extra.prev == Link::Extra(old_idx) {
extra.prev = Link::Extra(idx);
}
if extra.next == Link::Extra(old_idx) {
extra.next = Link::Extra(idx);
}
if idx != old_idx {
let next;
let prev;
{
debug_assert!(extra_values.len() > idx);
let moved = &extra_values[idx];
next = moved.next;
prev = moved.prev;
}
match prev {
Link::Entry(entry_idx) => {
debug_assert!(raw_links[entry_idx].is_some());
let links = raw_links[entry_idx].as_mut().unwrap();
links.next = idx;
}
Link::Extra(extra_idx) => {
debug_assert!(extra_values.len() > extra_idx);
extra_values[extra_idx].next = Link::Extra(idx);
}
}
match next {
Link::Entry(entry_idx) => {
debug_assert!(raw_links[entry_idx].is_some());
let links = raw_links[entry_idx].as_mut().unwrap();
links.tail = idx;
}
Link::Extra(extra_idx) => {
debug_assert!(extra_values.len() > extra_idx);
extra_values[extra_idx].prev = Link::Extra(idx);
}
}
}
debug_assert!({
for v in &*extra_values {
assert!(v.next != Link::Extra(old_idx));
assert!(v.prev != Link::Extra(old_idx));
}
true
});
extra
}
impl<'a, T> IntoIterator for &'a HeaderMap<T> {
type Item = (&'a HeaderName, &'a T);
type IntoIter = Iter<'a, T>;
fn into_iter(self) -> Iter<'a, T> {
self.iter()
}
}
impl<'a, T> IntoIterator for &'a mut HeaderMap<T> {
type Item = (&'a HeaderName, &'a mut T);
type IntoIter = IterMut<'a, T>;
fn into_iter(self) -> IterMut<'a, T> {
self.iter_mut()
}
}
impl<T> IntoIterator for HeaderMap<T> {
type Item = (Option<HeaderName>, T);
type IntoIter = IntoIter<T>;
fn into_iter(self) -> IntoIter<T> {
IntoIter {
next: None,
entries: self.entries.into_iter(),
extra_values: self.extra_values,
}
}
}
impl<T> HeaderMap<T> {
pub fn into_ordered_iter(self) -> IntoOrderedIter<T> {
IntoOrderedIter {
index: 0,
order: self.order,
entries: self.entries,
extra_values: self.extra_values,
}
}
}
impl<T> FromIterator<(HeaderName, T)> for HeaderMap<T> {
fn from_iter<I>(iter: I) -> Self
where
I: IntoIterator<Item = (HeaderName, T)>,
{
let mut map = HeaderMap::default();
map.extend(iter);
map
}
}
impl<'a, K, V, S, T> TryFrom<&'a HashMap<K, V, S>> for HeaderMap<T>
where
K: Eq + Hash,
HeaderName: TryFrom<&'a K>,
<HeaderName as TryFrom<&'a K>>::Error: Into<crate::Error>,
T: TryFrom<&'a V>,
T::Error: Into<crate::Error>,
{
type Error = Error;
fn try_from(c: &'a HashMap<K, V, S>) -> Result<Self, Self::Error> {
c.iter()
.map(|(k, v)| -> crate::Result<(HeaderName, T)> {
let name = TryFrom::try_from(k).map_err(Into::into)?;
let value = TryFrom::try_from(v).map_err(Into::into)?;
Ok((name, value))
})
.collect()
}
}
impl<T> Extend<(Option<HeaderName>, T)> for HeaderMap<T> {
fn extend<I: IntoIterator<Item = (Option<HeaderName>, T)>>(&mut self, iter: I) {
let mut iter = iter.into_iter();
let hint = if self.is_empty() {
iter.size_hint().0
} else {
(iter.size_hint().0 + 1) / 2
};
let max_reserve = usable_capacity(MAX_SIZE).saturating_sub(self.entries.len());
let reserve = hint.min(max_reserve);
self.reserve(reserve);
let (mut key, mut val) = match iter.next() {
Some((Some(key), val)) => (key, val),
Some((None, _)) => panic!("expected a header name, but got None"),
None => return,
};
'outer: loop {
let mut entry = match self.try_entry2(key).expect("size overflows MAX_SIZE") {
Entry::Occupied(mut e) => {
e.insert(val);
e
}
Entry::Vacant(e) => e.insert_entry(val),
};
loop {
match iter.next() {
Some((Some(k), v)) => {
key = k;
val = v;
continue 'outer;
}
Some((None, v)) => {
entry.append(v);
}
None => {
return;
}
}
}
}
}
}
impl<T> Extend<(HeaderName, T)> for HeaderMap<T> {
fn extend<I: IntoIterator<Item = (HeaderName, T)>>(&mut self, iter: I) {
let iter = iter.into_iter();
let hint = if self.is_empty() {
iter.size_hint().0
} else {
(iter.size_hint().0 + 1) / 2
};
let max_reserve = usable_capacity(MAX_SIZE).saturating_sub(self.entries.len());
let reserve = hint.min(max_reserve);
self.reserve(reserve);
for (k, v) in iter {
self.append(k, v);
}
}
}
impl<T: PartialEq> PartialEq for HeaderMap<T> {
fn eq(&self, other: &HeaderMap<T>) -> bool {
if self.len() != other.len() {
return false;
}
self.keys()
.all(|key| self.get_all(key) == other.get_all(key))
}
}
impl<T: Eq> Eq for HeaderMap<T> {}
impl<T: fmt::Debug> fmt::Debug for HeaderMap<T> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_map().entries(self.iter()).finish()
}
}
impl serde::Serialize for HeaderMap<HeaderValue> {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: serde::Serializer,
{
let headers: Result<Vec<_>, _> = self
.ordered_iter()
.map(|(name, value)| {
let value = value.to_str().map_err(serde::ser::Error::custom)?;
Ok::<_, S::Error>((name, value.to_owned()))
})
.collect();
headers?.serialize(serializer)
}
}
impl<'de> serde::Deserialize<'de> for HeaderMap<HeaderValue> {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: serde::Deserializer<'de>,
{
let headers = <Vec<(HeaderName, std::borrow::Cow<'de, str>)>>::deserialize(deserializer)?;
headers
.into_iter()
.map(|(name, value)| {
Ok::<_, D::Error>((
name,
HeaderValue::from_str(&value).map_err(serde::de::Error::custom)?,
))
})
.collect()
}
}
impl<K, T> ops::Index<K> for HeaderMap<T>
where
K: AsHeaderName,
{
type Output = T;
#[inline]
fn index(&self, index: K) -> &T {
match self.get2(&index) {
Some(val) => val,
None => panic!("no entry found for key {:?}", index.as_str()),
}
}
}
#[inline]
fn do_insert_phase_two(indices: &mut [Pos], mut probe: usize, mut old_pos: Pos) -> usize {
let mut num_displaced = 0;
probe_loop!(probe < indices.len(), {
let pos = &mut indices[probe];
if pos.is_none() {
*pos = old_pos;
break;
} else {
num_displaced += 1;
old_pos = mem::replace(pos, old_pos);
}
});
num_displaced
}
impl<'a, T> Iterator for Iter<'a, T> {
type Item = (&'a HeaderName, &'a T);
fn next(&mut self) -> Option<Self::Item> {
use self::Cursor::*;
if self.cursor.is_none() {
if (self.entry + 1) >= self.map.entries.len() {
return None;
}
self.entry += 1;
self.cursor = Some(Cursor::Head);
}
let entry = &self.map.entries[self.entry];
match self.cursor.unwrap() {
Head => {
self.cursor = entry.links.map(|l| Values(l.next));
Some((&entry.key, &entry.value))
}
Values(idx) => {
let extra = &self.map.extra_values[idx];
match extra.next {
Link::Entry(_) => self.cursor = None,
Link::Extra(i) => self.cursor = Some(Values(i)),
}
Some((&entry.key, &extra.value))
}
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
let map = self.map;
debug_assert!(map.entries.len() >= self.entry);
let lower = map.entries.len() - self.entry;
(lower, None)
}
}
impl<'a, T> FusedIterator for Iter<'a, T> {}
unsafe impl<'a, T: Sync> Sync for Iter<'a, T> {}
unsafe impl<'a, T: Sync> Send for Iter<'a, T> {}
impl<'a, T> Iterator for OrderedIter<'a, T> {
type Item = (&'a HeaderName, &'a T);
fn next(&mut self) -> Option<Self::Item> {
while self.index < self.map.order.len() {
let link = self.map.order[self.index];
self.index += 1;
if let Some(link) = link {
return match link {
Link::Entry(idx) => {
let entry = &self.map.entries[idx];
Some((&entry.key, &entry.value))
}
Link::Extra(idx) => {
let extra = &self.map.extra_values[idx];
Some((&extra.key, &extra.value))
}
};
}
}
None
}
fn size_hint(&self) -> (usize, Option<usize>) {
let remaining = self.map.order[self.index..]
.iter()
.filter(|link| link.is_some())
.count();
(remaining, Some(remaining))
}
}
impl<'a, T> FusedIterator for OrderedIter<'a, T> {}
unsafe impl<'a, T: Sync> Sync for OrderedIter<'a, T> {}
unsafe impl<'a, T: Sync> Send for OrderedIter<'a, T> {}
impl<'a, T> IterMut<'a, T> {
fn next_unsafe(&mut self) -> Option<(*const HeaderName, *mut T)> {
use self::Cursor::*;
if self.cursor.is_none() {
if (self.entry + 1) >= self.entries_len {
return None;
}
self.entry += 1;
self.cursor = Some(Cursor::Head);
}
let entry = unsafe { self.entries.add(self.entry) };
match self.cursor.unwrap() {
Head => {
self.cursor = unsafe { (*entry).links }.map(|l| Values(l.next));
Some(unsafe {
(
ptr::addr_of!((*entry).key),
ptr::addr_of_mut!((*entry).value),
)
})
}
Values(idx) => {
let extra = unsafe { self.extra_values.add(idx) };
match unsafe { (*extra).next } {
Link::Entry(_) => self.cursor = None,
Link::Extra(i) => self.cursor = Some(Values(i)),
}
Some(unsafe {
(
ptr::addr_of!((*entry).key),
ptr::addr_of_mut!((*extra).value),
)
})
}
}
}
}
impl<'a, T> Iterator for IterMut<'a, T> {
type Item = (&'a HeaderName, &'a mut T);
fn next(&mut self) -> Option<Self::Item> {
self.next_unsafe()
.map(|(key, ptr)| (unsafe { &*key }, unsafe { &mut *ptr }))
}
fn size_hint(&self) -> (usize, Option<usize>) {
debug_assert!(self.entries_len >= self.entry);
let lower = self.entries_len - self.entry;
(lower, None)
}
}
impl<'a, T> FusedIterator for IterMut<'a, T> {}
unsafe impl<'a, T: Sync> Sync for IterMut<'a, T> {}
unsafe impl<'a, T: Send> Send for IterMut<'a, T> {}
impl<'a, T> Iterator for Keys<'a, T> {
type Item = &'a HeaderName;
fn next(&mut self) -> Option<Self::Item> {
self.inner.next().map(|b| &b.key)
}
fn size_hint(&self) -> (usize, Option<usize>) {
self.inner.size_hint()
}
fn nth(&mut self, n: usize) -> Option<Self::Item> {
self.inner.nth(n).map(|b| &b.key)
}
fn count(self) -> usize {
self.inner.count()
}
fn last(self) -> Option<Self::Item> {
self.inner.last().map(|b| &b.key)
}
}
impl<'a, T> ExactSizeIterator for Keys<'a, T> {}
impl<'a, T> FusedIterator for Keys<'a, T> {}
impl<'a, T> Iterator for Values<'a, T> {
type Item = &'a T;
fn next(&mut self) -> Option<Self::Item> {
self.inner.next().map(|(_, v)| v)
}
fn size_hint(&self) -> (usize, Option<usize>) {
self.inner.size_hint()
}
}
impl<'a, T> FusedIterator for Values<'a, T> {}
impl<'a, T> Iterator for ValuesMut<'a, T> {
type Item = &'a mut T;
fn next(&mut self) -> Option<Self::Item> {
self.inner.next().map(|(_, v)| v)
}
fn size_hint(&self) -> (usize, Option<usize>) {
self.inner.size_hint()
}
}
impl<'a, T> FusedIterator for ValuesMut<'a, T> {}
impl<'a, T> Iterator for Drain<'a, T> {
type Item = (Option<HeaderName>, T);
fn next(&mut self) -> Option<Self::Item> {
if let Some(next) = self.next {
let raw_links = RawLinks(self.entries);
let extra = unsafe { remove_extra_value(raw_links, &mut *self.extra_values, next) };
match extra.next {
Link::Extra(idx) => self.next = Some(idx),
Link::Entry(_) => self.next = None,
}
return Some((None, extra.value));
}
let idx = self.idx;
if idx == self.len {
return None;
}
self.idx += 1;
unsafe {
let entry = &(*self.entries)[idx];
let key = ptr::read(&entry.key as *const _);
let value = ptr::read(&entry.value as *const _);
self.next = entry.links.map(|l| l.next);
Some((Some(key), value))
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
let lower = self.len - self.idx;
let upper = unsafe { (*self.extra_values).len() } + lower;
(lower, Some(upper))
}
}
impl<'a, T> FusedIterator for Drain<'a, T> {}
impl<'a, T> Drop for Drain<'a, T> {
fn drop(&mut self) {
struct Guard<'a, 'b, T>(&'a mut Drain<'b, T>);
struct ExtraValuesGuard<T>(*mut Vec<ExtraValue<T>>);
impl<T> Drop for ExtraValuesGuard<T> {
fn drop(&mut self) {
unsafe {
(*self.0).set_len(0);
}
}
}
impl<'a, 'b, T> Drop for Guard<'a, 'b, T> {
fn drop(&mut self) {
let _extra_values_guard = ExtraValuesGuard(self.0.extra_values);
for _ in self.0.by_ref() {}
}
}
let guard = Guard(self);
for _ in guard.0.by_ref() {}
}
}
unsafe impl<'a, T: Sync> Sync for Drain<'a, T> {}
unsafe impl<'a, T: Send> Send for Drain<'a, T> {}
impl<'a, T> Entry<'a, T> {
pub fn or_insert(self, default: T) -> &'a mut T {
self.or_try_insert(default)
.expect("size overflows MAX_SIZE")
}
pub fn or_try_insert(self, default: T) -> Result<&'a mut T, MaxSizeReached> {
use self::Entry::*;
match self {
Occupied(e) => Ok(e.into_mut()),
Vacant(e) => e.try_insert(default),
}
}
pub fn or_insert_with<F: FnOnce() -> T>(self, default: F) -> &'a mut T {
self.or_try_insert_with(default)
.expect("size overflows MAX_SIZE")
}
pub fn or_try_insert_with<F: FnOnce() -> T>(
self,
default: F,
) -> Result<&'a mut T, MaxSizeReached> {
use self::Entry::*;
match self {
Occupied(e) => Ok(e.into_mut()),
Vacant(e) => e.try_insert(default()),
}
}
pub fn key(&self) -> &HeaderName {
use self::Entry::*;
match *self {
Vacant(ref e) => e.key(),
Occupied(ref e) => e.key(),
}
}
}
impl<'a, T> VacantEntry<'a, T> {
pub fn key(&self) -> &HeaderName {
&self.key
}
pub fn into_key(self) -> HeaderName {
self.key
}
pub fn insert(self, value: T) -> &'a mut T {
self.try_insert(value).expect("size overflows MAX_SIZE")
}
pub fn try_insert(self, value: T) -> Result<&'a mut T, MaxSizeReached> {
let index =
self.map
.try_insert_phase_two(self.key, value, self.hash, self.probe, self.danger)?;
Ok(&mut self.map.entries[index].value)
}
pub fn insert_entry(self, value: T) -> OccupiedEntry<'a, T> {
self.try_insert_entry(value)
.expect("size overflows MAX_SIZE")
}
pub fn try_insert_entry(self, value: T) -> Result<OccupiedEntry<'a, T>, MaxSizeReached> {
let index =
self.map
.try_insert_phase_two(self.key, value, self.hash, self.probe, self.danger)?;
Ok(OccupiedEntry {
map: self.map,
index,
probe: self.probe,
})
}
}
impl<'a, T: 'a> GetAll<'a, T> {
pub fn iter(&self) -> ValueIter<'a, T> {
GetAll {
map: self.map,
index: self.index,
}
.into_iter()
}
}
impl<'a, T: PartialEq> PartialEq for GetAll<'a, T> {
fn eq(&self, other: &Self) -> bool {
self.iter().eq(other.iter())
}
}
impl<'a, T> IntoIterator for GetAll<'a, T> {
type Item = &'a T;
type IntoIter = ValueIter<'a, T>;
fn into_iter(self) -> ValueIter<'a, T> {
self.map.value_iter(self.index)
}
}
impl<'a, 'b: 'a, T> IntoIterator for &'b GetAll<'a, T> {
type Item = &'a T;
type IntoIter = ValueIter<'a, T>;
fn into_iter(self) -> ValueIter<'a, T> {
self.map.value_iter(self.index)
}
}
impl<'a, T: 'a> Iterator for ValueIter<'a, T> {
type Item = &'a T;
fn next(&mut self) -> Option<Self::Item> {
use self::Cursor::*;
match self.front {
Some(Head) => {
let entry = &self.map.entries[self.index];
if self.back == Some(Head) {
self.front = None;
self.back = None;
} else {
match entry.links {
Some(links) => {
self.front = Some(Values(links.next));
}
None => unreachable!(),
}
}
Some(&entry.value)
}
Some(Values(idx)) => {
let extra = &self.map.extra_values[idx];
if self.front == self.back {
self.front = None;
self.back = None;
} else {
match extra.next {
Link::Entry(_) => self.front = None,
Link::Extra(i) => self.front = Some(Values(i)),
}
}
Some(&extra.value)
}
None => None,
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
match (self.front, self.back) {
(Some(Cursor::Head), Some(Cursor::Head)) => (1, Some(1)),
(Some(_), _) => (1, None),
(None, _) => (0, Some(0)),
}
}
}
impl<'a, T: 'a> DoubleEndedIterator for ValueIter<'a, T> {
fn next_back(&mut self) -> Option<Self::Item> {
use self::Cursor::*;
match self.back {
Some(Head) => {
self.front = None;
self.back = None;
Some(&self.map.entries[self.index].value)
}
Some(Values(idx)) => {
let extra = &self.map.extra_values[idx];
if self.front == self.back {
self.front = None;
self.back = None;
} else {
match extra.prev {
Link::Entry(_) => self.back = Some(Head),
Link::Extra(idx) => self.back = Some(Values(idx)),
}
}
Some(&extra.value)
}
None => None,
}
}
}
impl<'a, T> FusedIterator for ValueIter<'a, T> {}
impl<'a, T: 'a> Iterator for ValueIterMut<'a, T> {
type Item = &'a mut T;
fn next(&mut self) -> Option<Self::Item> {
use self::Cursor::*;
let entry = unsafe { self.entries.add(self.index) };
match self.front {
Some(Head) => {
if self.back == Some(Head) {
self.front = None;
self.back = None;
} else {
match unsafe { (*entry).links } {
Some(links) => {
self.front = Some(Values(links.next));
}
None => unreachable!(),
}
}
Some(unsafe { &mut *ptr::addr_of_mut!((*entry).value) })
}
Some(Values(idx)) => {
let extra = unsafe { self.extra_values.add(idx) };
if self.front == self.back {
self.front = None;
self.back = None;
} else {
match unsafe { (*extra).next } {
Link::Entry(_) => self.front = None,
Link::Extra(i) => self.front = Some(Values(i)),
}
}
Some(unsafe { &mut *ptr::addr_of_mut!((*extra).value) })
}
None => None,
}
}
}
impl<'a, T: 'a> DoubleEndedIterator for ValueIterMut<'a, T> {
fn next_back(&mut self) -> Option<Self::Item> {
use self::Cursor::*;
let entry = unsafe { self.entries.add(self.index) };
match self.back {
Some(Head) => {
self.front = None;
self.back = None;
Some(unsafe { &mut *ptr::addr_of_mut!((*entry).value) })
}
Some(Values(idx)) => {
let extra = unsafe { self.extra_values.add(idx) };
if self.front == self.back {
self.front = None;
self.back = None;
} else {
match unsafe { (*extra).prev } {
Link::Entry(_) => self.back = Some(Head),
Link::Extra(idx) => self.back = Some(Values(idx)),
}
}
Some(unsafe { &mut *ptr::addr_of_mut!((*extra).value) })
}
None => None,
}
}
}
impl<'a, T> FusedIterator for ValueIterMut<'a, T> {}
unsafe impl<'a, T: Sync> Sync for ValueIterMut<'a, T> {}
unsafe impl<'a, T: Send> Send for ValueIterMut<'a, T> {}
impl<T> Iterator for IntoIter<T> {
type Item = (Option<HeaderName>, T);
fn next(&mut self) -> Option<Self::Item> {
if let Some(next) = self.next {
self.next = match self.extra_values[next].next {
Link::Entry(_) => None,
Link::Extra(v) => Some(v),
};
let key = ptr::addr_of_mut!(self.extra_values[next].key);
let value = unsafe {
ptr::drop_in_place(key);
ptr::read(&self.extra_values[next].value)
};
return Some((None, value));
}
if let Some(bucket) = self.entries.next() {
self.next = bucket.links.map(|l| l.next);
let name = Some(bucket.key);
let value = bucket.value;
return Some((name, value));
}
None
}
fn size_hint(&self) -> (usize, Option<usize>) {
let (lower, _) = self.entries.size_hint();
(lower, None)
}
}
impl<T> FusedIterator for IntoIter<T> {}
impl<T> Drop for IntoIter<T> {
fn drop(&mut self) {
struct Guard<'a, T>(&'a mut IntoIter<T>);
impl<'a, T> Drop for Guard<'a, T> {
fn drop(&mut self) {
unsafe {
self.0.extra_values.set_len(0);
}
}
}
let guard = Guard(self);
for _ in guard.0.by_ref() {}
}
}
impl<T> Iterator for IntoOrderedIter<T> {
type Item = (HeaderName, T);
fn next(&mut self) -> Option<Self::Item> {
while self.index < self.order.len() {
let link = self.order[self.index].take();
self.index += 1;
if let Some(link) = link {
return Some(match link {
Link::Entry(idx) => {
let entry = unsafe { self.entries.get_unchecked(idx) };
let name = unsafe { ptr::read(&entry.key) };
let value = unsafe { ptr::read(&entry.value) };
(name, value)
}
Link::Extra(idx) => {
let extra = unsafe { self.extra_values.get_unchecked(idx) };
let name = unsafe { ptr::read(&extra.key) };
let value = unsafe { ptr::read(&extra.value) };
(name, value)
}
});
}
}
None
}
fn size_hint(&self) -> (usize, Option<usize>) {
(0, Some(self.entries.len() + self.extra_values.len()))
}
}
impl<T> FusedIterator for IntoOrderedIter<T> {}
impl<T> Drop for IntoOrderedIter<T> {
fn drop(&mut self) {
struct Guard<'a, T>(&'a mut IntoOrderedIter<T>);
impl<'a, T> Drop for Guard<'a, T> {
fn drop(&mut self) {
unsafe {
self.0.entries.set_len(0);
self.0.extra_values.set_len(0);
}
}
}
let guard = Guard(self);
for _ in guard.0.by_ref() {}
}
}
impl<'a, T> OccupiedEntry<'a, T> {
pub fn key(&self) -> &HeaderName {
&self.map.entries[self.index].key
}
pub fn get(&self) -> &T {
&self.map.entries[self.index].value
}
pub fn get_mut(&mut self) -> &mut T {
&mut self.map.entries[self.index].value
}
pub fn into_mut(self) -> &'a mut T {
&mut self.map.entries[self.index].value
}
pub fn insert(&mut self, value: T) -> T {
self.map.insert_occupied_same_key(self.index, value)
}
pub fn insert_mult(&mut self, value: T) -> ValueDrain<'_, T> {
self.map.insert_occupied_mult_same_key(self.index, value)
}
pub fn append(&mut self, value: T) {
let idx = self.index;
let key = self.map.entries[idx].key.clone();
self.map.append_value(idx, key, value);
}
pub fn remove(self) -> T {
self.remove_entry().1
}
pub fn remove_entry(self) -> (HeaderName, T) {
if let Some(links) = self.map.entries[self.index].links {
self.map.remove_all_extra_values(links.next);
}
let entry = self.map.remove_found(self.probe, self.index);
(entry.key, entry.value)
}
pub fn remove_entry_mult(self) -> (HeaderName, ValueDrain<'a, T>) {
let next = self.map.entries[self.index]
.links
.map(|l| self.map.drain_all_extra_values(l.next).into_iter());
let entry = self.map.remove_found(self.probe, self.index);
let drain = ValueDrain {
first: Some(entry.value),
next,
lt: PhantomData,
};
(entry.key, drain)
}
pub fn iter(&self) -> ValueIter<'_, T> {
self.map.value_iter(Some(self.index))
}
pub fn iter_mut(&mut self) -> ValueIterMut<'_, T> {
self.map.value_iter_mut(self.index)
}
}
impl<'a, T> IntoIterator for OccupiedEntry<'a, T> {
type Item = &'a mut T;
type IntoIter = ValueIterMut<'a, T>;
fn into_iter(self) -> ValueIterMut<'a, T> {
self.map.value_iter_mut(self.index)
}
}
impl<'a, 'b: 'a, T> IntoIterator for &'b OccupiedEntry<'a, T> {
type Item = &'a T;
type IntoIter = ValueIter<'a, T>;
fn into_iter(self) -> ValueIter<'a, T> {
self.iter()
}
}
impl<'a, 'b: 'a, T> IntoIterator for &'b mut OccupiedEntry<'a, T> {
type Item = &'a mut T;
type IntoIter = ValueIterMut<'a, T>;
fn into_iter(self) -> ValueIterMut<'a, T> {
self.iter_mut()
}
}
impl<'a, T> Iterator for ValueDrain<'a, T> {
type Item = T;
fn next(&mut self) -> Option<T> {
if self.first.is_some() {
self.first.take()
} else if let Some(ref mut extras) = self.next {
extras.next()
} else {
None
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
match (&self.first, &self.next) {
(&Some(_), &None) => (1, Some(1)),
(&Some(_), Some(extras)) => {
let (l, u) = extras.size_hint();
(l + 1, u.map(|u| u + 1))
}
(&None, Some(extras)) => extras.size_hint(),
(&None, &None) => (0, Some(0)),
}
}
}
impl<'a, T> FusedIterator for ValueDrain<'a, T> {}
impl<'a, T> Drop for ValueDrain<'a, T> {
fn drop(&mut self) {
for _ in self.by_ref() {}
}
}
unsafe impl<'a, T: Sync> Sync for ValueDrain<'a, T> {}
unsafe impl<'a, T: Send> Send for ValueDrain<'a, T> {}
impl<T> Clone for RawLinks<T> {
fn clone(&self) -> RawLinks<T> {
*self
}
}
impl<T> Copy for RawLinks<T> {}
impl<T> ops::Index<usize> for RawLinks<T> {
type Output = Option<Links>;
fn index(&self, idx: usize) -> &Self::Output {
unsafe { &(*self.0)[idx].links }
}
}
impl<T> ops::IndexMut<usize> for RawLinks<T> {
fn index_mut(&mut self, idx: usize) -> &mut Self::Output {
unsafe { &mut (*self.0)[idx].links }
}
}
impl Pos {
#[inline]
fn new(index: usize, hash: HashValue) -> Self {
debug_assert!(index < MAX_SIZE);
Pos {
index: index as Size,
hash,
}
}
#[inline]
fn none() -> Self {
Pos {
index: !0,
hash: HashValue(0),
}
}
#[inline]
fn is_some(&self) -> bool {
!self.is_none()
}
#[inline]
fn is_none(&self) -> bool {
self.index == !0
}
#[inline]
fn resolve(&self) -> Option<(usize, HashValue)> {
if self.is_some() {
Some((self.index as usize, self.hash))
} else {
None
}
}
}
impl Danger {
fn is_red(&self) -> bool {
matches!(*self, Danger::Red(_))
}
fn set_red(&mut self) {
debug_assert!(self.is_yellow());
*self = Danger::Red(RandomState::new());
}
fn is_yellow(&self) -> bool {
matches!(*self, Danger::Yellow)
}
fn set_yellow(&mut self) {
if let Danger::Green = *self {
*self = Danger::Yellow;
}
}
fn set_green(&mut self) {
debug_assert!(self.is_yellow());
*self = Danger::Green;
}
}
impl MaxSizeReached {
fn new() -> Self {
MaxSizeReached { _priv: () }
}
}
impl fmt::Debug for MaxSizeReached {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("MaxSizeReached")
.finish()
}
}
impl fmt::Display for MaxSizeReached {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("max size reached")
}
}
impl std::error::Error for MaxSizeReached {}
#[inline]
fn usable_capacity(cap: usize) -> usize {
cap - cap / 4
}
#[inline]
fn to_raw_capacity(n: usize) -> Result<usize, MaxSizeReached> {
n.checked_add(n / 3).ok_or_else(MaxSizeReached::new)
}
#[inline]
fn desired_pos(mask: Size, hash: HashValue) -> usize {
(hash.0 & mask) as usize
}
#[inline]
fn probe_distance(mask: Size, hash: HashValue, current: usize) -> usize {
current.wrapping_sub(desired_pos(mask, hash)) & mask as usize
}
fn hash_elem_using<K>(danger: &Danger, k: &K) -> HashValue
where
K: Hash + ?Sized,
{
const MASK: u64 = (MAX_SIZE as u64) - 1;
let hash = match *danger {
Danger::Red(ref hasher) => {
let mut h = hasher.build_hasher();
k.hash(&mut h);
h.finish()
}
_ => {
let mut h = FnvHasher::new();
k.hash(&mut h);
h.finish()
}
};
HashValue((hash & MASK) as u16)
}
struct FnvHasher(u64);
impl FnvHasher {
#[inline]
fn new() -> Self {
FnvHasher(0xcbf29ce484222325)
}
}
impl std::hash::Hasher for FnvHasher {
#[inline]
fn finish(&self) -> u64 {
self.0
}
#[inline]
fn write(&mut self, bytes: &[u8]) {
let mut hash = self.0;
for &b in bytes {
hash ^= b as u64;
hash = hash.wrapping_mul(0x100000001b3);
}
self.0 = hash;
}
}
mod into_header_name {
use super::{Entry, HdrName, HeaderMap, HeaderName, MaxSizeReached};
pub trait IntoHeaderName: Sealed {}
pub trait Sealed {
#[doc(hidden)]
fn try_insert<T>(self, map: &mut HeaderMap<T>, val: T)
-> Result<Option<T>, MaxSizeReached>;
#[doc(hidden)]
fn try_append<T>(self, map: &mut HeaderMap<T>, val: T) -> Result<bool, MaxSizeReached>;
#[doc(hidden)]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, MaxSizeReached>;
}
impl Sealed for HeaderName {
#[inline]
fn try_insert<T>(
self,
map: &mut HeaderMap<T>,
val: T,
) -> Result<Option<T>, MaxSizeReached> {
map.try_insert2(self, val)
}
#[inline]
fn try_append<T>(self, map: &mut HeaderMap<T>, val: T) -> Result<bool, MaxSizeReached> {
map.try_append2(self, val)
}
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, MaxSizeReached> {
map.try_entry2(self)
}
}
impl IntoHeaderName for HeaderName {}
impl Sealed for &HeaderName {
#[inline]
fn try_insert<T>(
self,
map: &mut HeaderMap<T>,
val: T,
) -> Result<Option<T>, MaxSizeReached> {
map.try_insert2(self, val)
}
#[inline]
fn try_append<T>(self, map: &mut HeaderMap<T>, val: T) -> Result<bool, MaxSizeReached> {
map.try_append2(self, val)
}
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, MaxSizeReached> {
map.try_entry2(self)
}
}
impl IntoHeaderName for &HeaderName {}
impl Sealed for &'static str {
#[inline]
fn try_insert<T>(
self,
map: &mut HeaderMap<T>,
val: T,
) -> Result<Option<T>, MaxSizeReached> {
HdrName::from_static(self, move |hdr| map.try_insert2(hdr, val))
}
#[inline]
fn try_append<T>(self, map: &mut HeaderMap<T>, val: T) -> Result<bool, MaxSizeReached> {
HdrName::from_static(self, move |hdr| map.try_append2(hdr, val))
}
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, MaxSizeReached> {
HdrName::from_static(self, move |hdr| map.try_entry2(hdr))
}
}
impl IntoHeaderName for &'static str {}
}
mod as_header_name {
use super::{Entry, HdrName, HeaderMap, HeaderName, InvalidHeaderName, MaxSizeReached};
pub trait AsHeaderName: Sealed {}
#[allow(missing_debug_implementations)]
pub enum TryEntryError {
InvalidHeaderName(InvalidHeaderName),
MaxSizeReached(MaxSizeReached),
}
impl From<InvalidHeaderName> for TryEntryError {
fn from(e: InvalidHeaderName) -> TryEntryError {
TryEntryError::InvalidHeaderName(e)
}
}
impl From<MaxSizeReached> for TryEntryError {
fn from(e: MaxSizeReached) -> TryEntryError {
TryEntryError::MaxSizeReached(e)
}
}
pub trait Sealed {
#[doc(hidden)]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError>;
#[doc(hidden)]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)>;
#[doc(hidden)]
fn as_str(&self) -> &str;
}
impl Sealed for HeaderName {
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError> {
Ok(map.try_entry2(self)?)
}
#[inline]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)> {
map.find(self)
}
fn as_str(&self) -> &str {
<HeaderName>::as_str(self)
}
}
impl AsHeaderName for HeaderName {}
impl Sealed for &HeaderName {
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError> {
Ok(map.try_entry2(self)?)
}
#[inline]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)> {
map.find(*self)
}
fn as_str(&self) -> &str {
<HeaderName>::as_str(self)
}
}
impl AsHeaderName for &HeaderName {}
impl Sealed for &str {
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError> {
Ok(HdrName::from_bytes(self.as_bytes(), move |hdr| {
map.try_entry2(hdr)
})??)
}
#[inline]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)> {
HdrName::from_bytes(self.as_bytes(), move |hdr| map.find(&hdr)).unwrap_or(None)
}
fn as_str(&self) -> &str {
self
}
}
impl AsHeaderName for &str {}
impl Sealed for String {
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError> {
self.as_str().try_entry(map)
}
#[inline]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)> {
Sealed::find(&self.as_str(), map)
}
fn as_str(&self) -> &str {
self
}
}
impl AsHeaderName for String {}
impl Sealed for &String {
#[inline]
fn try_entry<T>(self, map: &mut HeaderMap<T>) -> Result<Entry<'_, T>, TryEntryError> {
self.as_str().try_entry(map)
}
#[inline]
fn find<T>(&self, map: &HeaderMap<T>) -> Option<(usize, usize)> {
Sealed::find(*self, map)
}
fn as_str(&self) -> &str {
self
}
}
impl AsHeaderName for &String {}
}
#[test]
fn test_bounds() {
fn check_bounds<T: Send + Send>() {}
check_bounds::<HeaderMap<()>>();
check_bounds::<Iter<'static, ()>>();
check_bounds::<IterMut<'static, ()>>();
check_bounds::<Keys<'static, ()>>();
check_bounds::<Values<'static, ()>>();
check_bounds::<ValuesMut<'static, ()>>();
check_bounds::<Drain<'static, ()>>();
check_bounds::<GetAll<'static, ()>>();
check_bounds::<Entry<'static, ()>>();
check_bounds::<VacantEntry<'static, ()>>();
check_bounds::<OccupiedEntry<'static, ()>>();
check_bounds::<ValueIter<'static, ()>>();
check_bounds::<ValueIterMut<'static, ()>>();
check_bounds::<ValueDrain<'static, ()>>();
}
#[test]
fn extend_size_hint_above_capacity() {
let name = HeaderName::from_static("h");
let value = HeaderValue::from_static("0");
let pairs: Vec<(HeaderName, HeaderValue)> =
std::iter::repeat_with(|| (name.clone(), value.clone()))
.take(24_577)
.collect();
let map = HeaderMap::from_iter(pairs);
assert_eq!(map.len(), 24_577);
assert_eq!(map.keys_len(), 1);
}
#[cfg(test)]
fn assert_header_map_invariants<T>(map: &HeaderMap<T>) {
assert_eq!(map.len(), map.entries.len() + map.extra_values.len());
let live_order_slots = map.order.iter().filter(|link| link.is_some()).count();
let order_holes = map.order.iter().filter(|link| link.is_none()).count();
assert_eq!(live_order_slots, map.len());
assert_eq!(order_holes, map.order_holes);
let mut indexed_entries = vec![false; map.entries.len()];
for pos in &*map.indices {
if let Some((idx, hash)) = pos.resolve() {
assert!(idx < map.entries.len(), "index points past entries: {idx}");
assert_eq!(map.entries[idx].hash, hash);
assert!(!indexed_entries[idx], "entry indexed more than once: {idx}");
indexed_entries[idx] = true;
}
}
assert!(
indexed_entries.iter().all(|seen| *seen),
"not every entry is reachable from indices"
);
let mut linked_extras = vec![false; map.extra_values.len()];
for (entry_idx, entry) in map.entries.iter().enumerate() {
assert_eq!(map.order[entry.order], Some(Link::Entry(entry_idx)));
let Some(links) = entry.links else {
continue;
};
assert!(links.next < map.extra_values.len());
assert!(links.tail < map.extra_values.len());
let mut prev = Link::Entry(entry_idx);
let mut current = links.next;
loop {
assert!(current < map.extra_values.len());
assert!(
!linked_extras[current],
"extra value linked more than once: {current}"
);
linked_extras[current] = true;
let extra = &map.extra_values[current];
assert_eq!(extra.prev, prev);
assert_eq!(extra.key, entry.key);
assert_eq!(map.order[extra.order], Some(Link::Extra(current)));
match extra.next {
Link::Entry(idx) => {
assert_eq!(idx, entry_idx);
assert_eq!(current, links.tail);
break;
}
Link::Extra(next) => {
prev = Link::Extra(current);
current = next;
}
}
}
}
assert!(
linked_extras.iter().all(|seen| *seen),
"not every extra value is linked from an entry"
);
for (order_idx, link) in map.order.iter().enumerate() {
match link {
Some(Link::Entry(idx)) => {
assert!(*idx < map.entries.len());
assert_eq!(map.entries[*idx].order, order_idx);
}
Some(Link::Extra(idx)) => {
assert!(*idx < map.extra_values.len());
assert_eq!(map.extra_values[*idx].order, order_idx);
}
None => {}
}
}
}
#[cfg(test)]
#[derive(Clone, Debug)]
struct HeaderMapOps(Vec<HeaderMapOp>);
#[cfg(test)]
#[derive(Clone, Debug)]
enum HeaderMapOp {
Append { name: u8, value: u8 },
Insert { name: u8, value: u8 },
Remove { name: u8 },
RemoveEntryMult { name: u8 },
EntryAppend { name: u8, value: u8 },
EntryInsertMult { name: u8, value: u8 },
}
#[cfg(test)]
#[derive(Clone, Debug, Eq, PartialEq)]
struct ModelHeader {
semantic_id: usize,
name: String,
value: String,
head: bool,
}
#[cfg(test)]
#[derive(Default)]
struct HeaderMapModel {
fields: Vec<Option<ModelHeader>>,
}
#[cfg(test)]
impl HeaderMapModel {
fn len(&self) -> usize {
self.fields.iter().filter(|field| field.is_some()).count()
}
fn ordered(&self) -> Vec<(String, String)> {
self.fields
.iter()
.filter_map(|field| {
field
.as_ref()
.map(|field| (field.name.clone(), field.value.clone()))
})
.collect()
}
fn values_for(&self, semantic_id: usize) -> Vec<String> {
self.fields
.iter()
.filter_map(|field| {
let field = field.as_ref()?;
(field.semantic_id == semantic_id).then(|| field.value.clone())
})
.collect()
}
fn head_index(&self, semantic_id: usize) -> Option<usize> {
self.fields.iter().position(|field| {
field
.as_ref()
.is_some_and(|field| field.semantic_id == semantic_id && field.head)
})
}
fn remove(&mut self, semantic_id: usize) {
for field in &mut self.fields {
if field
.as_ref()
.is_some_and(|field| field.semantic_id == semantic_id)
{
*field = None;
}
}
}
fn append(&mut self, semantic_id: usize, name: String, value: String) {
let head = self.head_index(semantic_id).is_none();
self.fields.push(Some(ModelHeader {
semantic_id,
name,
value,
head,
}));
}
fn insert(&mut self, semantic_id: usize, name: String, value: String) {
if let Some(head_idx) = self.head_index(semantic_id) {
for (idx, field) in self.fields.iter_mut().enumerate() {
if idx != head_idx
&& field
.as_ref()
.is_some_and(|field| field.semantic_id == semantic_id)
{
*field = None;
}
}
self.fields[head_idx] = Some(ModelHeader {
semantic_id,
name,
value,
head: true,
});
} else {
self.append(semantic_id, name, value);
}
}
fn entry_append(&mut self, semantic_id: usize, name: String, value: String) {
if let Some(head_idx) = self.head_index(semantic_id) {
let entry_name = self.fields[head_idx].as_ref().unwrap().name.clone();
self.append(semantic_id, entry_name, value);
} else {
self.append(semantic_id, name, value);
}
}
fn entry_insert_mult(&mut self, semantic_id: usize, name: String, value: String) {
if let Some(head_idx) = self.head_index(semantic_id) {
let entry_name = self.fields[head_idx].as_ref().unwrap().name.clone();
self.insert(semantic_id, entry_name, value);
} else {
self.append(semantic_id, name, value);
}
}
}
#[cfg(test)]
impl quickcheck::Arbitrary for HeaderMapOps {
fn arbitrary(g: &mut quickcheck::Gen) -> Self {
let len = usize::from(u8::arbitrary(g) % 80);
let ops = (0..len).map(|_| HeaderMapOp::arbitrary(g)).collect();
Self(ops)
}
fn shrink(&self) -> Box<dyn Iterator<Item = Self>> {
Box::new(self.0.shrink().map(Self))
}
}
#[cfg(test)]
impl quickcheck::Arbitrary for HeaderMapOp {
fn arbitrary(g: &mut quickcheck::Gen) -> Self {
let name = u8::arbitrary(g);
let value = u8::arbitrary(g);
match u8::arbitrary(g) % 6 {
0 => Self::Append { name, value },
1 => Self::Insert { name, value },
2 => Self::Remove { name },
3 => Self::RemoveEntryMult { name },
4 => Self::EntryAppend { name, value },
_ => Self::EntryInsertMult { name, value },
}
}
}
#[cfg(test)]
fn model_header_name(selector: u8) -> (&'static str, usize) {
match selector % 8 {
0 => ("x-a", 0),
1 => ("X-A", 0),
2 => ("x-b", 1),
3 => ("X-B", 1),
4 => ("cookie", 2),
5 => ("Cookie", 2),
6 => ("x-c", 3),
_ => ("X-C", 3),
}
}
#[cfg(test)]
fn model_header_value(value: u8) -> String {
format!("v{value}")
}
#[cfg(test)]
fn make_header_name(name: &str) -> HeaderName {
HeaderName::from_bytes(name.as_bytes()).unwrap()
}
#[cfg(test)]
fn model_original_name(name: &str) -> String {
make_header_name(name).to_string()
}
#[cfg(test)]
fn make_header_value(value: &str) -> HeaderValue {
HeaderValue::from_bytes(value.as_bytes()).unwrap()
}
#[cfg(test)]
fn assert_header_map_matches_model(map: &HeaderMap, model: &HeaderMapModel) {
assert_header_map_invariants(map);
assert_eq!(map.len(), model.len());
let ordered = map
.ordered_iter()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>();
assert_eq!(ordered, model.ordered());
let consumed = map
.clone()
.into_ordered_iter()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>();
assert_eq!(consumed, ordered);
for (selector, semantic_id) in [
("x-a", 0),
("X-A", 0),
("x-b", 1),
("X-B", 1),
("cookie", 2),
("Cookie", 2),
("x-c", 3),
("X-C", 3),
] {
let actual = map
.get_all(make_header_name(selector))
.iter()
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>();
assert_eq!(actual, model.values_for(semantic_id));
}
}
#[cfg(test)]
impl HeaderMapOps {
fn run(self) {
let mut map = HeaderMap::new();
let mut model = HeaderMapModel::default();
assert_header_map_matches_model(&map, &model);
for op in self.0 {
match op {
HeaderMapOp::Append { name, value } => {
let (name, semantic_id) = model_header_name(name);
let value = model_header_value(value);
map.append(make_header_name(name), make_header_value(&value));
model.append(semantic_id, model_original_name(name), value);
}
HeaderMapOp::Insert { name, value } => {
let (name, semantic_id) = model_header_name(name);
let value = model_header_value(value);
map.insert(make_header_name(name), make_header_value(&value));
model.insert(semantic_id, model_original_name(name), value);
}
HeaderMapOp::Remove { name } => {
let (name, semantic_id) = model_header_name(name);
map.remove(make_header_name(name));
model.remove(semantic_id);
}
HeaderMapOp::RemoveEntryMult { name } => {
let (name, semantic_id) = model_header_name(name);
if let Entry::Occupied(entry) = map.entry(make_header_name(name)) {
let _ = entry.remove_entry_mult();
}
model.remove(semantic_id);
}
HeaderMapOp::EntryAppend { name, value } => {
let (name, semantic_id) = model_header_name(name);
let value = model_header_value(value);
match map.entry(make_header_name(name)) {
Entry::Occupied(mut entry) => {
entry.append(make_header_value(&value));
}
Entry::Vacant(entry) => {
entry.insert_entry(make_header_value(&value));
}
}
model.entry_append(semantic_id, model_original_name(name), value);
}
HeaderMapOp::EntryInsertMult { name, value } => {
let (name, semantic_id) = model_header_name(name);
let value = model_header_value(value);
match map.entry(make_header_name(name)) {
Entry::Occupied(mut entry) => {
let _ = entry.insert_mult(make_header_value(&value));
}
Entry::Vacant(entry) => {
entry.insert_entry(make_header_value(&value));
}
}
model.entry_insert_mult(semantic_id, model_original_name(name), value);
}
}
assert_header_map_matches_model(&map, &model);
}
}
}
#[test]
fn quickcheck_header_map_order_and_multivalue_model() {
fn prop(ops: HeaderMapOps) -> bool {
ops.run();
true
}
quickcheck::QuickCheck::new()
.tests(500)
.quickcheck(prop as fn(HeaderMapOps) -> bool);
}
#[test]
fn ensure_miri_itermut_not_violated() {
let mut headers = HeaderMap::<u32>::default();
headers.insert(HeaderName::from_static("hello"), 1u32);
headers.insert(HeaderName::from_static("zomg"), 2u32);
let mut iter = headers.iter_mut();
let (_, first) = iter.next().unwrap();
let (_, second) = iter.next().unwrap();
*first += 10;
*second += 20;
}
#[test]
fn ensure_miri_valueitermut_not_violated() {
let mut headers = HeaderMap::<u32>::default();
headers.insert(HeaderName::from_static("hello"), 1u32);
headers.append(HeaderName::from_static("hello"), 2u32);
headers.append(HeaderName::from_static("hello"), 3u32);
let mut entry = match headers.entry(HeaderName::from_static("hello")) {
Entry::Occupied(entry) => entry,
Entry::Vacant(_) => panic!(),
};
let mut iter = entry.iter_mut();
let first = iter.next().unwrap();
let second = iter.next().unwrap();
*first += 10;
*second += 20;
}
struct ManuallyAllocated {
ptr: *mut u8,
panic_on_drop: bool,
}
impl ManuallyAllocated {
fn new(byte: u8, panic_on_drop: bool) -> Self {
Self {
ptr: Box::into_raw(Box::new(byte)),
panic_on_drop,
}
}
}
impl Drop for ManuallyAllocated {
fn drop(&mut self) {
unsafe {
drop(Box::from_raw(self.ptr));
}
if self.panic_on_drop {
panic!("intentional drop panic");
}
}
}
#[test]
fn into_iter_drop_panic_after_yielding_extra_value_double_drops() {
use std::panic::{AssertUnwindSafe, catch_unwind};
let mut map: HeaderMap<ManuallyAllocated> = HeaderMap::default();
map.append("x-first", ManuallyAllocated::new(1, false));
map.append("x-first", ManuallyAllocated::new(2, false));
map.insert("x-second", ManuallyAllocated::new(3, true));
let mut iter = map.into_iter();
drop(iter.next().unwrap());
drop(iter.next().unwrap());
let _ = catch_unwind(AssertUnwindSafe(|| drop(iter)));
}
#[test]
fn into_ordered_iter_drop_panic_after_yielding_values_does_not_double_drop() {
use std::panic::{AssertUnwindSafe, catch_unwind};
let mut map: HeaderMap<ManuallyAllocated> = HeaderMap::default();
map.append("x-first", ManuallyAllocated::new(1, false));
map.append("x-first", ManuallyAllocated::new(2, false));
map.insert("x-second", ManuallyAllocated::new(3, true));
let mut iter = map.into_ordered_iter();
drop(iter.next().unwrap());
drop(iter.next().unwrap());
let _ = catch_unwind(AssertUnwindSafe(|| drop(iter)));
}
#[test]
fn drain_drop_panic_leaves_map_empty_and_valid() {
use std::panic::{AssertUnwindSafe, catch_unwind};
let mut map: HeaderMap<ManuallyAllocated> = HeaderMap::default();
map.insert("x-panic", ManuallyAllocated::new(1, true));
map.append("x-rest", ManuallyAllocated::new(2, false));
map.append("x-rest", ManuallyAllocated::new(3, false));
let drain = map.drain();
let _ = catch_unwind(AssertUnwindSafe(|| drop(drain)));
assert_header_map_invariants(&map);
}
#[test]
fn skip_duplicates_during_key_iteration() {
let mut map = HeaderMap::new();
map.try_append("a", HeaderValue::from_static("a")).unwrap();
map.try_append("a", HeaderValue::from_static("b")).unwrap();
assert_eq!(map.keys().count(), map.keys_len());
}
#[test]
fn ordered_iter_size_hint_counts_remaining_live_order_slots() {
let mut map = HeaderMap::new();
map.try_insert("a", HeaderValue::from_static("a")).unwrap();
map.try_insert("b", HeaderValue::from_static("b")).unwrap();
map.try_insert("c", HeaderValue::from_static("c")).unwrap();
map.remove("b");
let mut iter = map.ordered_iter();
assert_eq!((2, Some(2)), iter.size_hint());
assert_eq!("a", iter.next().unwrap().0);
assert_eq!((1, Some(1)), iter.size_hint());
assert_eq!("c", iter.next().unwrap().0);
assert_eq!((0, Some(0)), iter.size_hint());
assert!(iter.next().is_none());
}
#[test]
fn drain_size_hint_counts_remaining_entries_and_extras() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.insert("x-second", HeaderValue::from_static("three"));
let mut drain = map.drain();
assert_eq!(drain.size_hint(), (2, Some(3)));
assert_eq!(
drain.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((Some("x-first".to_owned()), "one".to_owned()))
);
assert_eq!(drain.size_hint(), (1, Some(2)));
assert_eq!(
drain.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((None, "two".to_owned()))
);
assert_eq!(drain.size_hint(), (1, Some(1)));
assert_eq!(
drain.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((Some("x-second".to_owned()), "three".to_owned()))
);
assert_eq!(drain.size_hint(), (0, Some(0)));
assert!(drain.next().is_none());
}
#[test]
fn into_iter_size_hint_counts_remaining_entries_only() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.insert("x-second", HeaderValue::from_static("three"));
let mut iter = map.into_iter();
assert_eq!(iter.size_hint(), (2, None));
assert_eq!(
iter.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((Some("x-first".to_owned()), "one".to_owned()))
);
assert_eq!(iter.size_hint(), (1, None));
assert_eq!(
iter.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((None, "two".to_owned()))
);
assert_eq!(iter.size_hint(), (1, None));
assert_eq!(
iter.next().map(|(name, value)| (
name.map(|name| name.to_string()),
value.to_str().unwrap().to_owned()
)),
Some((Some("x-second".to_owned()), "three".to_owned()))
);
assert_eq!(iter.size_hint(), (0, None));
assert!(iter.next().is_none());
}
#[test]
fn into_ordered_iter_size_hint_upper_bound_counts_backing_values() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.insert("x-second", HeaderValue::from_static("three"));
let mut iter = map.into_ordered_iter();
assert_eq!(iter.size_hint(), (0, Some(3)));
assert_eq!(
iter.next()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned())),
Some(("x-first".to_owned(), "one".to_owned()))
);
assert_eq!(iter.size_hint(), (0, Some(3)));
assert_eq!(
iter.next()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned())),
Some(("x-first".to_owned(), "two".to_owned()))
);
assert_eq!(iter.size_hint(), (0, Some(3)));
assert_eq!(
iter.next()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned())),
Some(("x-second".to_owned(), "three".to_owned()))
);
assert_eq!(iter.size_hint(), (0, Some(3)));
assert!(iter.next().is_none());
}
#[test]
fn value_drain_size_hint_counts_remaining_values() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.append("x-first", HeaderValue::from_static("three"));
let Entry::Occupied(mut entry) = map.entry("x-first") else {
panic!("expected x-first header");
};
let mut drain = entry.insert_mult(HeaderValue::from_static("replacement"));
assert_eq!(drain.size_hint(), (3, Some(3)));
assert_eq!(
drain.next().map(|value| value.to_str().unwrap().to_owned()),
Some("one".to_owned())
);
assert_eq!(drain.size_hint(), (2, Some(2)));
assert_eq!(
drain.next().map(|value| value.to_str().unwrap().to_owned()),
Some("two".to_owned())
);
assert_eq!(drain.size_hint(), (1, Some(1)));
assert_eq!(
drain.next().map(|value| value.to_str().unwrap().to_owned()),
Some("three".to_owned())
);
assert_eq!(drain.size_hint(), (0, Some(0)));
assert!(drain.next().is_none());
}
#[test]
fn dropping_value_drain_drops_remaining_values() {
use std::cell::Cell;
use std::rc::Rc;
struct DropCounter(Rc<Cell<usize>>);
impl Drop for DropCounter {
fn drop(&mut self) {
self.0.set(self.0.get() + 1);
}
}
let drops = Rc::new(Cell::new(0));
let mut map: HeaderMap<DropCounter> = HeaderMap::default();
map.append("x-first", DropCounter(drops.clone()));
map.append("x-first", DropCounter(drops.clone()));
map.append("x-first", DropCounter(drops.clone()));
let Entry::Occupied(mut entry) = map.entry("x-first") else {
panic!("expected x-first header");
};
let mut drain = entry.insert_mult(DropCounter(drops.clone()));
drop(drain.next());
assert_eq!(drops.get(), 1);
drop(drain);
assert_eq!(drops.get(), 3);
}
#[test]
fn basic_map_api_contracts_cover_capacity_lookup_and_clear() {
let empty = HeaderMap::<HeaderValue>::try_with_capacity(0).unwrap();
assert!(empty.is_empty());
assert_eq!(empty.capacity(), 0);
let mut map = HeaderMap::<HeaderValue>::with_capacity(10);
assert!(map.is_empty());
assert_eq!(map.len(), 0);
assert_eq!(map.keys_len(), 0);
assert_eq!(map.capacity(), 12);
assert_eq!(map.mask as usize, 15);
let mut reserved = HeaderMap::<HeaderValue>::new();
reserved.reserve(10);
assert_eq!(reserved.capacity(), 12);
assert_eq!(reserved.mask as usize, 15);
let mut try_reserved = HeaderMap::<HeaderValue>::new();
try_reserved.try_reserve(10).unwrap();
assert_eq!(try_reserved.capacity(), 12);
assert_eq!(try_reserved.mask as usize, 15);
let max_capacity =
HeaderMap::<HeaderValue>::try_with_capacity(usable_capacity(MAX_SIZE)).unwrap();
assert_eq!(max_capacity.capacity(), usable_capacity(MAX_SIZE));
assert_eq!(max_capacity.mask as usize, MAX_SIZE - 1);
let max_size_err =
HeaderMap::<HeaderValue>::try_with_capacity(usable_capacity(MAX_SIZE) + 1).unwrap_err();
assert_eq!(max_size_err.to_string(), "max size reached");
assert_eq!(format!("{max_size_err:?}"), "MaxSizeReached");
let mut too_large_reserve = HeaderMap::<HeaderValue>::new();
assert!(
too_large_reserve
.try_reserve(usable_capacity(MAX_SIZE) + 1)
.is_err()
);
let key = HeaderName::from_static("x-basic");
let key_string = "x-basic".to_owned();
assert!(!map.contains_key(&key));
assert!(!map.contains_key(&key_string));
assert!(map.get(&key).is_none());
assert!(map.get("x-basic").is_none());
assert!(map.get(key_string.clone()).is_none());
assert!(map.get(&key_string).is_none());
assert!(
map.insert(key.clone(), HeaderValue::from_static("one"))
.is_none()
);
assert!(!map.is_empty());
assert_eq!(map.len(), 1);
assert_eq!(map.keys_len(), 1);
assert!(map.contains_key(&key));
assert!(map.contains_key("x-basic"));
assert!(map.contains_key(key_string.clone()));
assert!(map.contains_key(&key_string));
assert_eq!(map.get(&key).unwrap(), "one");
assert_eq!(map.get(key.clone()).unwrap(), "one");
assert_eq!(map.get("x-basic").unwrap(), "one");
assert_eq!(map.get(key_string.clone()).unwrap(), "one");
assert_eq!(map.get(&key_string).unwrap(), "one");
assert_eq!(map.get("missing"), None);
*map.get_mut("x-basic").unwrap() = HeaderValue::from_static("mutated");
assert_eq!(map.get("x-basic").unwrap(), "mutated");
assert!(map.append("x-basic", HeaderValue::from_static("two")));
assert_eq!(map.len(), 2);
assert_eq!(map.keys_len(), 1);
let borrowed = HeaderName::from_static("x-borrowed");
assert!(
map.insert(&borrowed, HeaderValue::from_static("borrowed"))
.is_none()
);
assert!(map.append(&borrowed, HeaderValue::from_static("borrowed-again")));
assert_eq!(
map.get_all(&borrowed)
.iter()
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>(),
["borrowed".to_owned(), "borrowed-again".to_owned()]
);
map.clear();
assert!(map.is_empty());
assert_eq!(map.len(), 0);
assert_eq!(map.keys_len(), 0);
assert!(map.capacity() >= 12);
assert!(!map.contains_key("x-basic"));
}
#[test]
fn extend_try_from_equality_debug_and_serde_contracts() {
use std::collections::HashMap;
use std::convert::TryFrom;
let mut raw = HashMap::new();
raw.insert("x-one".to_owned(), "one".to_owned());
raw.insert("x-two".to_owned(), "two".to_owned());
let from_hash = HeaderMap::<HeaderValue>::try_from(&raw).unwrap();
assert_eq!(from_hash.len(), 2);
assert_eq!(from_hash.get("x-one").unwrap(), "one");
assert_eq!(from_hash.get("x-two").unwrap(), "two");
let mut by_name = HeaderMap::new();
by_name.extend([
(
HeaderName::from_static("x-a"),
HeaderValue::from_static("a1"),
),
(
HeaderName::from_static("x-a"),
HeaderValue::from_static("a2"),
),
(
HeaderName::from_static("x-b"),
HeaderValue::from_static("b1"),
),
]);
assert_eq!(
by_name
.get_all("x-a")
.iter()
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>(),
["a1".to_owned(), "a2".to_owned()]
);
let mut by_optional_name = HeaderMap::new();
by_optional_name.extend([
(
Some(HeaderName::from_static("x-c")),
HeaderValue::from_static("c1"),
),
(None, HeaderValue::from_static("c2")),
(
Some(HeaderName::from_static("x-d")),
HeaderValue::from_static("d1"),
),
]);
assert_eq!(
by_optional_name
.get_all("x-c")
.iter()
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>(),
["c1".to_owned(), "c2".to_owned()]
);
let same = by_optional_name.clone();
assert_eq!(by_optional_name, same);
let mut different = same.clone();
different.append("x-c", HeaderValue::from_static("c3"));
assert_ne!(by_optional_name, different);
let debug = format!("{by_optional_name:?}");
assert!(debug.contains("x-c"));
assert!(debug.contains("c1"));
let json = serde_json::to_string(&by_optional_name).unwrap();
let roundtrip: HeaderMap = serde_json::from_str(&json).unwrap();
assert_eq!(
roundtrip
.ordered_iter()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>(),
by_optional_name
.ordered_iter()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>()
);
}
#[test]
fn unordered_iterators_and_key_value_views_have_stable_contracts() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.insert("x-second", HeaderValue::from_static("three"));
let iter = map.iter();
assert_eq!(iter.size_hint(), (2, None));
let mut iter = iter;
let first = iter
.next()
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.unwrap();
assert_eq!(iter.size_hint(), (2, None));
let mut pairs = iter
.map(|(name, value)| (name.to_string(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>();
pairs.push(first);
pairs.sort();
assert_eq!(
pairs,
[
("x-first".to_owned(), "one".to_owned()),
("x-first".to_owned(), "two".to_owned()),
("x-second".to_owned(), "three".to_owned()),
]
);
assert_eq!(map.keys().size_hint(), (2, Some(2)));
let mut keys = map.keys().map(|name| name.to_string()).collect::<Vec<_>>();
keys.sort();
assert_eq!(keys, ["x-first".to_owned(), "x-second".to_owned()]);
assert_eq!(map.keys().count(), 2);
assert!(map.keys().nth(1).is_some());
assert!(map.keys().last().is_some());
assert_eq!(map.values().size_hint(), (2, None));
let mut values = map
.values()
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>();
values.sort();
assert_eq!(
values,
["one".to_owned(), "three".to_owned(), "two".to_owned()]
);
let mut mutable: HeaderMap<String> = HeaderMap::default();
mutable.append("x-first", "one".to_owned());
mutable.append("x-first", "two".to_owned());
mutable.insert("x-second", "three".to_owned());
let mut iter_mut = mutable.iter_mut();
assert_eq!(iter_mut.size_hint(), (2, None));
iter_mut.next().unwrap().1.push('!');
assert_eq!(iter_mut.size_hint(), (2, None));
for (_, value) in iter_mut {
value.push('!');
}
assert_eq!(mutable.values_mut().size_hint(), (2, None));
for value in mutable.values_mut() {
value.push('?');
}
let mut values = mutable.values().cloned().collect::<Vec<_>>();
values.sort();
assert_eq!(
values,
["one!?".to_owned(), "three!?".to_owned(), "two!?".to_owned()]
);
}
#[test]
fn per_key_value_iterators_support_forward_and_reverse_traversal() {
let mut map = HeaderMap::new();
map.append("x-multi", HeaderValue::from_static("one"));
map.append("x-multi", HeaderValue::from_static("two"));
map.append("x-multi", HeaderValue::from_static("three"));
map.insert("x-other", HeaderValue::from_static("solo"));
assert_eq!(
map.get_all("x-multi"),
map.get_all(&HeaderName::from_static("x-multi"))
);
assert_ne!(map.get_all("x-multi"), map.get_all("x-other"));
let all = map.get_all("x-multi");
let mut iter = all.iter();
assert_eq!(iter.size_hint(), (1, None));
assert_eq!(iter.next().unwrap(), "one");
assert_eq!(iter.size_hint(), (1, None));
assert_eq!(iter.next_back().unwrap(), "three");
assert_eq!(iter.size_hint(), (1, None));
assert_eq!(iter.next().unwrap(), "two");
assert_eq!(iter.size_hint(), (0, Some(0)));
assert!(iter.next().is_none());
assert!(iter.next_back().is_none());
let mut mutable: HeaderMap<String> = HeaderMap::default();
mutable.append("x-multi", "one".to_owned());
mutable.append("x-multi", "two".to_owned());
mutable.append("x-multi", "three".to_owned());
let Entry::Occupied(mut entry) = mutable.entry("x-multi") else {
panic!("expected x-multi header");
};
let mut iter = entry.iter_mut();
assert_eq!(iter.size_hint(), (0, None));
iter.next().unwrap().push('1');
assert_eq!(iter.size_hint(), (0, None));
iter.next_back().unwrap().push('3');
assert_eq!(iter.size_hint(), (0, None));
iter.next().unwrap().push('2');
assert_eq!(iter.size_hint(), (0, None));
assert!(iter.next().is_none());
assert!(iter.next_back().is_none());
assert_eq!(
mutable
.get_all("x-multi")
.iter()
.cloned()
.collect::<Vec<_>>(),
["one1".to_owned(), "two2".to_owned(), "three3".to_owned()]
);
}
#[test]
fn growth_remove_and_lookup_stress_preserves_entries_and_order() {
let mut map: HeaderMap<String> = HeaderMap::with_capacity(1);
for i in 0..96 {
let name = HeaderName::from_bytes(format!("x-grow-{i}").as_bytes()).unwrap();
map.insert(name, format!("v{i}"));
}
assert_eq!(map.len(), 96);
assert_eq!(map.keys_len(), 96);
assert!(map.capacity() >= 96);
assert_header_map_invariants(&map);
for i in (0..96).step_by(3) {
let name = HeaderName::from_bytes(format!("x-grow-{i}").as_bytes()).unwrap();
assert_eq!(map.remove(name), Some(format!("v{i}")));
}
assert_eq!(map.len(), 64);
assert_eq!(map.keys_len(), 64);
assert_header_map_invariants(&map);
for i in 0..96 {
let name = format!("x-grow-{i}");
if i % 3 == 0 {
assert!(!map.contains_key(name.as_str()));
assert!(map.get(name.as_str()).is_none());
} else {
assert_eq!(map.get(name.as_str()).unwrap(), &format!("v{i}"));
}
}
for i in (0..96).step_by(3) {
let name = HeaderName::from_bytes(format!("x-grow-{i}").as_bytes()).unwrap();
map.insert(name, format!("reinserted-{i}"));
}
assert_eq!(map.len(), 96);
assert_header_map_invariants(&map);
assert_eq!(map.get("x-grow-0").unwrap(), "reinserted-0");
assert_eq!(map.get("x-grow-95").unwrap(), "v95");
}
#[test]
fn header_map_invariants_survive_ordered_multi_value_mutations() {
let mut map = HeaderMap::new();
assert_header_map_invariants(&map);
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.append("x-second", HeaderValue::from_static("alpha"));
map.append("x-second", HeaderValue::from_static("beta"));
map.insert("x-third", HeaderValue::from_static("single"));
assert_header_map_invariants(&map);
assert_eq!(map.remove("x-first").unwrap(), "one");
assert_header_map_invariants(&map);
map.append("x-second", HeaderValue::from_static("gamma"));
assert_header_map_invariants(&map);
assert_eq!(
map.insert("x-second", HeaderValue::from_static("replaced"))
.unwrap(),
"alpha"
);
assert_header_map_invariants(&map);
map.append("x-second", HeaderValue::from_static("delta"));
map.append("x-fourth", HeaderValue::from_static("uno"));
map.append("x-fourth", HeaderValue::from_static("dos"));
assert_header_map_invariants(&map);
let Entry::Occupied(fourth) = map.entry("x-fourth") else {
panic!("expected x-fourth header");
};
let (name, values) = fourth.remove_entry_mult();
assert_eq!(name, "x-fourth");
assert_eq!(
values
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>(),
["uno".to_owned(), "dos".to_owned()]
);
assert_header_map_invariants(&map);
map.append("x-fifth", HeaderValue::from_static("eins"));
map.append("x-fifth", HeaderValue::from_static("zwei"));
let Entry::Occupied(mut fifth) = map.entry("x-fifth") else {
panic!("expected x-fifth header");
};
assert_eq!(
fifth
.insert_mult(HeaderValue::from_static("new"))
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>(),
["eins".to_owned(), "zwei".to_owned()]
);
assert_header_map_invariants(&map);
let drained = map.drain().collect::<Vec<_>>();
assert!(!drained.is_empty());
assert_header_map_invariants(&map);
}
#[test]
fn remove_entry_mult_then_into_ordered_iter_skips_removed_extras() {
let mut map = HeaderMap::new();
map.append("cookie", HeaderValue::from_static("a=1"));
map.append("cookie", HeaderValue::from_static("b=2"));
map.append("cookie", HeaderValue::from_static("c=3"));
let Entry::Occupied(cookie_headers) = map.entry("cookie") else {
panic!("expected cookie header");
};
let (name, values) = cookie_headers.remove_entry_mult();
let merged = values
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>()
.join("; ");
map.insert(name, HeaderValue::from_str(&merged).unwrap());
let ordered = map
.into_ordered_iter()
.map(|(name, value)| (name.as_str().to_owned(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>();
assert_eq!(ordered, [("cookie".to_owned(), "a=1; b=2; c=3".to_owned())]);
}
#[test]
fn insert_mult_then_ordered_iter_updates_moved_extra_order_links() {
let mut map = HeaderMap::new();
map.append("x-first", HeaderValue::from_static("one"));
map.append("x-first", HeaderValue::from_static("two"));
map.append("x-second", HeaderValue::from_static("alpha"));
map.append("x-second", HeaderValue::from_static("beta"));
let Entry::Occupied(mut entry) = map.entry("x-first") else {
panic!("expected x-first header");
};
let old = entry
.insert_mult(HeaderValue::from_static("replaced"))
.map(|value| value.to_str().unwrap().to_owned())
.collect::<Vec<_>>();
assert_eq!(old, ["one".to_owned(), "two".to_owned()]);
let ordered = map
.ordered_iter()
.map(|(name, value)| (name.as_str().to_owned(), value.to_str().unwrap().to_owned()))
.collect::<Vec<_>>();
assert_eq!(
ordered,
[
("x-first".to_owned(), "replaced".to_owned()),
("x-second".to_owned(), "alpha".to_owned()),
("x-second".to_owned(), "beta".to_owned()),
]
);
}
#[test]
fn order_capacity_grows_with_entry_capacity() {
let mut map = HeaderMap::<HeaderValue>::new();
map.try_insert("a", HeaderValue::from_static("a")).unwrap();
assert!(
map.order.capacity() >= map.entries.capacity(),
"order capacity {} should track entries capacity {}",
map.order.capacity(),
map.entries.capacity()
);
let initial_order_cap = map.order.capacity();
let initial_entry_cap = map.entries.capacity();
for i in 0..initial_entry_cap {
let name = format!("x-rama-{i}");
let name = HeaderName::from_bytes(name.as_bytes()).unwrap();
map.try_insert(name, HeaderValue::from_static("x")).unwrap();
}
assert!(
map.order.capacity() >= map.entries.capacity(),
"order capacity {} should track grown entries capacity {}",
map.order.capacity(),
map.entries.capacity()
);
assert!(map.order.capacity() >= initial_order_cap);
}