c7ed475f54
* Use bytes. Not yet zerocopy everywhere but compiles * Actually zerocopy * Actually zerocopy * Not actually storing the torrent on disk now
76 lines
2.1 KiB
Rust
76 lines
2.1 KiB
Rust
// These are helpers for objects that can be borrowed, but can be made owned while changing the type.
|
|
// The difference between e.g. Cow and CloneToOwned, is that we can implement it recursively for owned types.
|
|
//
|
|
// E.g. HashMap<&str, &str> can be converted to HashMap<String, String>.
|
|
//
|
|
// This lets us express types like TorrentMetaInfo<&[u8]> for zero-copy metadata about a bencode buffer in memory,
|
|
// but to have one-line conversion for it into TorrentMetaInfo<Vec<u8>> so that we can store it later somewhere.
|
|
|
|
use bytes::Bytes;
|
|
use std::collections::HashMap;
|
|
|
|
pub trait CloneToOwned {
|
|
type Target;
|
|
|
|
fn clone_to_owned(&self, within_buffer: Option<&Bytes>) -> Self::Target;
|
|
}
|
|
|
|
impl<T> CloneToOwned for Option<T>
|
|
where
|
|
T: CloneToOwned,
|
|
{
|
|
type Target = Option<<T as CloneToOwned>::Target>;
|
|
|
|
fn clone_to_owned(&self, within_buffer: Option<&Bytes>) -> Self::Target {
|
|
self.as_ref().map(|i| i.clone_to_owned(within_buffer))
|
|
}
|
|
}
|
|
|
|
impl<T> CloneToOwned for Vec<T>
|
|
where
|
|
T: CloneToOwned,
|
|
{
|
|
type Target = Vec<<T as CloneToOwned>::Target>;
|
|
|
|
fn clone_to_owned(&self, within_buffer: Option<&Bytes>) -> Self::Target {
|
|
self.iter()
|
|
.map(|i| i.clone_to_owned(within_buffer))
|
|
.collect()
|
|
}
|
|
}
|
|
|
|
impl CloneToOwned for u8 {
|
|
type Target = u8;
|
|
|
|
fn clone_to_owned(&self, _within_buffer: Option<&Bytes>) -> Self::Target {
|
|
*self
|
|
}
|
|
}
|
|
|
|
impl CloneToOwned for u32 {
|
|
type Target = u32;
|
|
|
|
fn clone_to_owned(&self, _within_buffer: Option<&Bytes>) -> Self::Target {
|
|
*self
|
|
}
|
|
}
|
|
|
|
impl<K, V> CloneToOwned for HashMap<K, V>
|
|
where
|
|
K: CloneToOwned,
|
|
<K as CloneToOwned>::Target: std::hash::Hash + Eq,
|
|
V: CloneToOwned,
|
|
{
|
|
type Target = HashMap<<K as CloneToOwned>::Target, <V as CloneToOwned>::Target>;
|
|
|
|
fn clone_to_owned(&self, within_buffer: Option<&Bytes>) -> Self::Target {
|
|
let mut result = HashMap::with_capacity(self.capacity());
|
|
for (k, v) in self {
|
|
result.insert(
|
|
k.clone_to_owned(within_buffer),
|
|
v.clone_to_owned(within_buffer),
|
|
);
|
|
}
|
|
result
|
|
}
|
|
}
|