Path: blob/main/crates/component-macro/tests/expanded/direct-import_tracing_async.rs
1692 views
/// Auto-generated bindings for a pre-instantiated version of a1/// component which implements the world `foo`.2///3/// This structure is created through [`FooPre::new`] which4/// takes a [`InstancePre`](wasmtime::component::InstancePre) that5/// has been created through a [`Linker`](wasmtime::component::Linker).6///7/// For more information see [`Foo`] as well.8pub struct FooPre<T: 'static> {9instance_pre: wasmtime::component::InstancePre<T>,10indices: FooIndices,11}12impl<T: 'static> Clone for FooPre<T> {13fn clone(&self) -> Self {14Self {15instance_pre: self.instance_pre.clone(),16indices: self.indices.clone(),17}18}19}20impl<_T: 'static> FooPre<_T> {21/// Creates a new copy of `FooPre` bindings which can then22/// be used to instantiate into a particular store.23///24/// This method may fail if the component behind `instance_pre`25/// does not have the required exports.26pub fn new(27instance_pre: wasmtime::component::InstancePre<_T>,28) -> wasmtime::Result<Self> {29let indices = FooIndices::new(&instance_pre)?;30Ok(Self { instance_pre, indices })31}32pub fn engine(&self) -> &wasmtime::Engine {33self.instance_pre.engine()34}35pub fn instance_pre(&self) -> &wasmtime::component::InstancePre<_T> {36&self.instance_pre37}38/// Instantiates a new instance of [`Foo`] within the39/// `store` provided.40///41/// This function will use `self` as the pre-instantiated42/// instance to perform instantiation. Afterwards the preloaded43/// indices in `self` are used to lookup all exports on the44/// resulting instance.45pub fn instantiate(46&self,47mut store: impl wasmtime::AsContextMut<Data = _T>,48) -> wasmtime::Result<Foo> {49let mut store = store.as_context_mut();50let instance = self.instance_pre.instantiate(&mut store)?;51self.indices.load(&mut store, &instance)52}53}54impl<_T: Send + 'static> FooPre<_T> {55/// Same as [`Self::instantiate`], except with `async`.56pub async fn instantiate_async(57&self,58mut store: impl wasmtime::AsContextMut<Data = _T>,59) -> wasmtime::Result<Foo> {60let mut store = store.as_context_mut();61let instance = self.instance_pre.instantiate_async(&mut store).await?;62self.indices.load(&mut store, &instance)63}64}65/// Auto-generated bindings for index of the exports of66/// `foo`.67///68/// This is an implementation detail of [`FooPre`] and can69/// be constructed if needed as well.70///71/// For more information see [`Foo`] as well.72#[derive(Clone)]73pub struct FooIndices {}74/// Auto-generated bindings for an instance a component which75/// implements the world `foo`.76///77/// This structure can be created through a number of means78/// depending on your requirements and what you have on hand:79///80/// * The most convenient way is to use81/// [`Foo::instantiate`] which only needs a82/// [`Store`], [`Component`], and [`Linker`].83///84/// * Alternatively you can create a [`FooPre`] ahead of85/// time with a [`Component`] to front-load string lookups86/// of exports once instead of per-instantiation. This87/// method then uses [`FooPre::instantiate`] to88/// create a [`Foo`].89///90/// * If you've instantiated the instance yourself already91/// then you can use [`Foo::new`].92///93/// These methods are all equivalent to one another and move94/// around the tradeoff of what work is performed when.95///96/// [`Store`]: wasmtime::Store97/// [`Component`]: wasmtime::component::Component98/// [`Linker`]: wasmtime::component::Linker99pub struct Foo {}100pub trait FooImportsWithStore: wasmtime::component::HasData + Send {}101impl<_T: ?Sized> FooImportsWithStore for _T102where103_T: wasmtime::component::HasData + Send,104{}105pub trait FooImports: Send {106fn foo(&mut self) -> impl ::core::future::Future<Output = ()> + Send;107}108impl<_T: FooImports + ?Sized + Send> FooImports for &mut _T {109fn foo(&mut self) -> impl ::core::future::Future<Output = ()> + Send {110async move { FooImports::foo(*self).await }111}112}113const _: () = {114#[allow(unused_imports)]115use wasmtime::component::__internal::anyhow;116impl FooIndices {117/// Creates a new copy of `FooIndices` bindings which can then118/// be used to instantiate into a particular store.119///120/// This method may fail if the component does not have the121/// required exports.122pub fn new<_T>(123_instance_pre: &wasmtime::component::InstancePre<_T>,124) -> wasmtime::Result<Self> {125let _component = _instance_pre.component();126let _instance_type = _instance_pre.instance_type();127Ok(FooIndices {})128}129/// Uses the indices stored in `self` to load an instance130/// of [`Foo`] from the instance provided.131///132/// Note that at this time this method will additionally133/// perform type-checks of all exports.134pub fn load(135&self,136mut store: impl wasmtime::AsContextMut,137instance: &wasmtime::component::Instance,138) -> wasmtime::Result<Foo> {139let _ = &mut store;140let _instance = instance;141Ok(Foo {})142}143}144impl Foo {145/// Convenience wrapper around [`FooPre::new`] and146/// [`FooPre::instantiate`].147pub fn instantiate<_T>(148store: impl wasmtime::AsContextMut<Data = _T>,149component: &wasmtime::component::Component,150linker: &wasmtime::component::Linker<_T>,151) -> wasmtime::Result<Foo> {152let pre = linker.instantiate_pre(component)?;153FooPre::new(pre)?.instantiate(store)154}155/// Convenience wrapper around [`FooIndices::new`] and156/// [`FooIndices::load`].157pub fn new(158mut store: impl wasmtime::AsContextMut,159instance: &wasmtime::component::Instance,160) -> wasmtime::Result<Foo> {161let indices = FooIndices::new(&instance.instance_pre(&store))?;162indices.load(&mut store, instance)163}164/// Convenience wrapper around [`FooPre::new`] and165/// [`FooPre::instantiate_async`].166pub async fn instantiate_async<_T>(167store: impl wasmtime::AsContextMut<Data = _T>,168component: &wasmtime::component::Component,169linker: &wasmtime::component::Linker<_T>,170) -> wasmtime::Result<Foo>171where172_T: Send,173{174let pre = linker.instantiate_pre(component)?;175FooPre::new(pre)?.instantiate_async(store).await176}177pub fn add_to_linker_imports<T, D>(178linker: &mut wasmtime::component::Linker<T>,179host_getter: fn(&mut T) -> D::Data<'_>,180) -> wasmtime::Result<()>181where182D: FooImportsWithStore,183for<'a> D::Data<'a>: FooImports,184T: 'static + Send,185{186let mut linker = linker.root();187linker188.func_wrap_async(189"foo",190move |mut caller: wasmtime::StoreContextMut<'_, T>, (): ()| {191use tracing::Instrument;192let span = tracing::span!(193tracing::Level::TRACE, "wit-bindgen import", module = "foo",194function = "foo",195);196wasmtime::component::__internal::Box::new(197async move {198tracing::event!(tracing::Level::TRACE, "call");199let host = &mut host_getter(caller.data_mut());200let r = FooImports::foo(host).await;201tracing::event!(202tracing::Level::TRACE, result = tracing::field::debug(& r),203"return"204);205Ok(r)206}207.instrument(span),208)209},210)?;211Ok(())212}213pub fn add_to_linker<T, D>(214linker: &mut wasmtime::component::Linker<T>,215host_getter: fn(&mut T) -> D::Data<'_>,216) -> wasmtime::Result<()>217where218D: FooImportsWithStore + Send,219for<'a> D::Data<'a>: FooImports + Send,220T: 'static + Send,221{222Self::add_to_linker_imports::<T, D>(linker, host_getter)?;223Ok(())224}225}226};227228229