Add AsyncFunctionEnvHandleMut::as_function_env_mut

Author: Arshia001

lib/api/src/backend/sys/entities/function/env.rs

@@ -384,6 +384,14 @@ impl<T: 'static> AsyncFunctionEnvHandleMut<T> {
         let data = unsafe { &mut *data };
         (data, &mut self.write_lock)
     }
+
+    /// Borrows a new [`FunctionEnvMut`] from this [`AsyncFunctionEnvHandleMut`].
+    pub fn as_function_env_mut(&mut self) -> FunctionEnvMut<'_, T> {
+        FunctionEnvMut {
+            store_mut: self.write_lock.as_store_mut(),
+            func_env: self.func_env.clone(),
+        }
+    }
 }
 
 #[cfg(feature = "experimental-async")]

lib/api/src/entities/function/env/inner.rs

@@ -393,6 +393,17 @@ impl<T: 'static> BackendAsyncFunctionEnvHandleMut<T> {
             _ => unsupported_async_backend::<(&mut T, &mut crate::StoreMut)>(),
         }
     }
+
+    /// Borrows a new [`BackendFunctionEnvMut`] from this
+    /// [`BackendAsyncFunctionEnvHandleMut`].
+    pub fn as_function_env_mut(&mut self) -> BackendFunctionEnvMut<'_, T> {
+        match self {
+            #[cfg(feature = "sys")]
+            Self::Sys(f) => BackendFunctionEnvMut::Sys(f.as_function_env_mut()),
+            #[cfg(not(feature = "sys"))]
+            _ => unsupported_async_backend(),
+        }
+    }
 }
 
 #[cfg(feature = "experimental-async")]

lib/api/src/entities/function/env/mod.rs

@@ -193,6 +193,12 @@ impl<T: 'static> AsyncFunctionEnvHandleMut<T> {
     pub fn data_and_store_mut(&mut self) -> (&mut T, &mut impl AsStoreMut) {
         self.0.data_and_store_mut()
     }
+
+    /// Borrows a new [`FunctionEnvMut`] from this
+    /// [`AsyncFunctionEnvHandleMut`].
+    pub fn as_function_env_mut(&mut self) -> FunctionEnvMut<'_, T> {
+        FunctionEnvMut(self.0.as_function_env_mut())
+    }
 }
 
 #[cfg(feature = "experimental-async")]