parent
5c784482f6
commit
9a7234806f
GPG Key ID:
284488987C31F630
@ -0,0 +1,9 @@
|
||||
//! Container for chain
|
||||
use super::*;
|
||||
use chain::{Chain, Chainable};
|
||||
|
||||
pub struct ChainContainer<T: Chainable>
|
||||
{
|
||||
chain: Chain<T>,
|
||||
inputs_exact: dedup::HashRefSet<[T]>,
|
||||
}
|
||||
@ -0,0 +1,94 @@
|
||||
//! De-duplicating inputs
|
||||
use super::*;
|
||||
use std::{
|
||||
hash::{Hash, Hasher},
|
||||
collections::{HashSet, hash_set},
|
||||
marker::PhantomData,
|
||||
borrow::Borrow,
|
||||
};
|
||||
use sha2::{Sha256, Digest};
|
||||
use cryptohelpers::sha256::Sha256Hash;
|
||||
|
||||
fn compute<T: Hash>(thing: &T) -> Sha256Hash
|
||||
{
|
||||
use std::mem::size_of;
|
||||
struct Sha256Hasher(Sha256);
|
||||
impl Hasher for Sha256Hasher
|
||||
{
|
||||
fn write(&mut self, bytes: &[u8])
|
||||
{
|
||||
self.0.update(bytes);
|
||||
}
|
||||
fn finish(&self) -> u64
|
||||
{
|
||||
let ar = self.0.clone().finalize();
|
||||
let mut rest = [0u8; size_of::<u64>()];
|
||||
bytes::copy_slice(&mut rest[..], &ar[..]);
|
||||
u64::from_le_bytes(rest)
|
||||
}
|
||||
}
|
||||
|
||||
let mut hasher = Sha256Hasher(Sha256::new());
|
||||
thing.hash(&mut hasher);
|
||||
hasher.0.into()
|
||||
}
|
||||
|
||||
/// A HashSet that doesn't own its items.
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
pub struct HashRefSet<T: Hash + ?Sized>(HashSet<Sha256Hash>, PhantomData<HashSet<*const T>>);
|
||||
|
||||
unsafe impl<T: Send + Hash + ?Sized> Send for HashRefSet<T>{}
|
||||
unsafe impl<T: Send + Sync + Hash + ?Sized> Sync for HashRefSet<T>{}
|
||||
|
||||
impl<T: Hash + ?Sized> HashRefSet<T>
|
||||
{
|
||||
pub fn new() -> Self
|
||||
{
|
||||
Self(HashSet::new(), PhantomData)
|
||||
}
|
||||
pub fn with_capacity(cap: usize) -> Self
|
||||
{
|
||||
Self(HashSet::with_capacity(cap), PhantomData)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: Hash + ?Sized> util::NewCapacity for HashRefSet<T>
|
||||
{
|
||||
fn new() -> Self
|
||||
{
|
||||
Self::new()
|
||||
}
|
||||
|
||||
fn with_capacity(cap: usize) -> Self
|
||||
{
|
||||
Self::with_capacity(cap)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: Hash> HashRefSet<T>
|
||||
{
|
||||
pub fn insert<U>(&mut self, value: &U) -> bool
|
||||
where U: Borrow<T>
|
||||
{
|
||||
self.0.insert(compute(value.borrow()))
|
||||
}
|
||||
pub fn remove<U>(&mut self, value: &U) -> bool
|
||||
where U: Borrow<T>
|
||||
{
|
||||
self.0.remove(&compute(value.borrow()))
|
||||
}
|
||||
pub fn contains<U>(&self, value: &U) -> bool
|
||||
where U: Borrow<T>
|
||||
{
|
||||
self.0.contains(&compute(value.borrow()))
|
||||
}
|
||||
pub fn len(&self) -> usize
|
||||
{
|
||||
self.0.len()
|
||||
}
|
||||
pub fn hashes(&self) -> hash_set::Iter<'_, Sha256Hash>
|
||||
{
|
||||
self.0.iter()
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue