use std::collections::{HashMap, HashSet};
use std::hash::{BuildHasherDefault, Hasher};
pub type FastMap<K, V> = HashMap<K, V, BuildHasherDefault<FastHasher>>;
pub type FastSet<T> = HashSet<T, BuildHasherDefault<FastHasher>>;
#[derive(Default)]
pub struct FastHasher(u64);
impl FastHasher {
#[inline]
fn mix(&mut self, value: u64) {
self.0 = (self.0 ^ value).wrapping_mul(0x9e37_79b9_7f4a_7c15);
self.0 ^= self.0 >> 29;
}
}
impl Hasher for FastHasher {
#[inline]
fn finish(&self) -> u64 {
self.0
}
#[inline]
fn write(&mut self, bytes: &[u8]) {
for &byte in bytes {
self.0 = (self.0 ^ u64::from(byte)).wrapping_mul(0x0000_0100_0000_01b3);
}
}
#[inline]
fn write_u8(&mut self, value: u8) {
self.mix(u64::from(value));
}
#[inline]
fn write_u16(&mut self, value: u16) {
self.mix(u64::from(value));
}
#[inline]
fn write_u32(&mut self, value: u32) {
self.mix(u64::from(value));
}
#[inline]
fn write_u64(&mut self, value: u64) {
self.mix(value);
}
#[inline]
fn write_usize(&mut self, value: usize) {
self.mix(value as u64);
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::hash::Hash;
fn hash_of<T: Hash>(value: T) -> u64 {
let mut hasher = FastHasher::default();
value.hash(&mut hasher);
hasher.finish()
}
#[test]
fn hashes_integers_deterministically() {
assert_eq!(hash_of(7usize), hash_of(7usize));
assert_ne!(hash_of(7usize), hash_of(8usize));
}
#[test]
fn hashes_tuples_and_byte_keys() {
assert_ne!(hash_of((1usize, 2usize)), hash_of((2usize, 1usize)));
assert_ne!(hash_of("alpha"), hash_of("beta"));
}
#[test]
fn fast_maps_use_the_fast_hasher() {
let mut map = FastMap::<usize, usize>::default();
map.insert(1, 2);
assert_eq!(map.get(&1), Some(&2));
let mut set = FastSet::<usize>::default();
set.insert(3);
assert!(set.contains(&3));
}
}