From 613cd7129439d04f576325bcf55d276847efb136 Mon Sep 17 00:00:00 2001 From: HampusM Date: Sun, 24 Jul 2022 11:02:33 +0200 Subject: refactor: clean up intertrait lib --- src/libs/intertrait/mod.rs | 71 +++++++++++++++++++++------------------------- 1 file changed, 32 insertions(+), 39 deletions(-) (limited to 'src/libs/intertrait/mod.rs') diff --git a/src/libs/intertrait/mod.rs b/src/libs/intertrait/mod.rs index 1daca64..2cdc67d 100644 --- a/src/libs/intertrait/mod.rs +++ b/src/libs/intertrait/mod.rs @@ -12,64 +12,62 @@ * at your option. */ use std::any::{Any, TypeId}; -use std::collections::HashMap; use std::rc::Rc; -use std::sync::Arc; +use ahash::AHashMap; use linkme::distributed_slice; use once_cell::sync::Lazy; -mod hasher; - -use hasher::BuildFastHasher; - pub mod cast_box; pub mod cast_rc; pub type BoxedCaster = Box; +type CasterFn = fn() -> (TypeId, BoxedCaster); + #[distributed_slice] -pub static CASTERS: [fn() -> (TypeId, BoxedCaster)] = [..]; - -static CASTER_MAP: Lazy> = - Lazy::new(|| { - CASTERS - .iter() - .map(|f| { - let (type_id, caster) = f(); - ((type_id, (*caster).type_id()), caster) - }) - .collect() - }); - -pub struct Caster +pub static CASTERS: [CasterFn] = [..]; + +static CASTER_MAP: Lazy> = Lazy::new(|| { + CASTERS + .iter() + .map(|caster_fn| { + let (type_id, caster) = caster_fn(); + + ((type_id, (*caster).type_id()), caster) + }) + .collect() +}); + +pub struct Caster { /// Casts a `Box` holding a trait object for `Any` to another `Box` holding a trait object - /// for trait `T`. - pub cast_box: fn(from: Box) -> Box, + /// for `Trait`. + pub cast_box: fn(from: Box) -> Box, /// Casts an `Rc` holding a trait object for `Any` to another `Rc` holding a trait object - /// for trait `T`. - pub cast_rc: fn(from: Rc) -> Rc, + /// for `Trait`. + pub cast_rc: fn(from: Rc) -> Rc, } -impl Caster +impl Caster { pub fn new( - cast_box: fn(from: Box) -> Box, - cast_rc: fn(from: Rc) -> Rc, - ) -> Caster + cast_box: fn(from: Box) -> Box, + cast_rc: fn(from: Rc) -> Rc, + ) -> Caster { - Caster:: { cast_box, cast_rc } + Caster:: { cast_box, cast_rc } } } -/// Returns a `Caster` from a concrete type `S` to a trait `T` implemented by it. -fn caster(type_id: TypeId) -> Option<&'static Caster> +/// Returns a `Caster` from a concrete type `Implementation` +/// from inside `CASTER_MAP` to a `Trait` implemented by it. +fn caster(type_id: TypeId) -> Option<&'static Caster> { CASTER_MAP - .get(&(type_id, TypeId::of::>())) - .and_then(|caster| caster.downcast_ref::>()) + .get(&(type_id, TypeId::of::>())) + .and_then(|caster| caster.downcast_ref::>()) } /// `CastFrom` must be extended by a trait that wants to allow for casting into another trait. @@ -92,12 +90,7 @@ pub trait CastFrom: Any + 'static fn rc_any(self: Rc) -> Rc; } -pub trait CastFromSync: CastFrom + Sync + Send + 'static -{ - fn arc_any(self: Arc) -> Arc; -} - -impl CastFrom for T +impl CastFrom for Trait { fn box_any(self: Box) -> Box { -- cgit v1.2.3-18-g5258