routing/bedrock/
weak_instance_token_ext.rs

1// Copyright 2024 The Fuchsia Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5use crate::component_instance::{
6    ComponentInstanceInterface, ExtendedInstanceInterface, WeakComponentInstanceInterface,
7    WeakExtendedInstanceInterface,
8};
9use crate::error::ComponentInstanceError;
10use moniker::ExtendedMoniker;
11use sandbox::WeakInstanceToken;
12use std::sync::Arc;
13
14/// A trait to add functions WeakComponentInstancethat know about the component
15/// manager types.
16pub trait WeakInstanceTokenExt<C: ComponentInstanceInterface + 'static> {
17    /// Upgrade this token to the underlying instance.
18    fn to_instance(self) -> WeakExtendedInstanceInterface<C>;
19
20    /// Get a reference to the underlying instance.
21    fn as_ref(&self) -> &WeakExtendedInstanceInterface<C>;
22
23    /// Get a strong reference to the underlying instance.
24    fn upgrade(&self) -> Result<ExtendedInstanceInterface<C>, ComponentInstanceError>;
25
26    /// Get the moniker for this component.
27    fn moniker(&self) -> ExtendedMoniker;
28}
29
30impl<C: ComponentInstanceInterface + 'static> WeakInstanceTokenExt<C> for WeakInstanceToken {
31    fn to_instance(self) -> WeakExtendedInstanceInterface<C> {
32        self.as_ref().clone()
33    }
34
35    fn as_ref(&self) -> &WeakExtendedInstanceInterface<C> {
36        match self.inner.as_any().downcast_ref::<WeakExtendedInstanceInterface<C>>() {
37            Some(instance) => &instance,
38            None => panic!(),
39        }
40    }
41
42    fn upgrade(&self) -> Result<ExtendedInstanceInterface<C>, ComponentInstanceError> {
43        self.as_ref().upgrade()
44    }
45
46    fn moniker(&self) -> ExtendedMoniker {
47        WeakInstanceTokenExt::<C>::as_ref(self).extended_moniker()
48    }
49}
50
51/// Returns a token representing an invalid component. For testing.
52pub fn test_invalid_instance_token<C: ComponentInstanceInterface + 'static>() -> WeakInstanceToken {
53    WeakComponentInstanceInterface::<C>::invalid().into()
54}
55
56impl<C: ComponentInstanceInterface + 'static> From<WeakExtendedInstanceInterface<C>>
57    for WeakInstanceToken
58{
59    fn from(instance: WeakExtendedInstanceInterface<C>) -> Self {
60        Self { inner: Arc::new(instance) }
61    }
62}
63
64impl<C: ComponentInstanceInterface + 'static> From<WeakComponentInstanceInterface<C>>
65    for WeakInstanceToken
66{
67    fn from(instance: WeakComponentInstanceInterface<C>) -> Self {
68        Self::from(WeakExtendedInstanceInterface::<C>::Component(instance))
69    }
70}
routing/bedrock/weak_instance_token_ext.rs

routing/bedrock/
weak_instance_token_ext.rs
