124 lines
3.5 KiB
Rust
124 lines
3.5 KiB
Rust
use std::future::Future;
|
|
use std::marker::PhantomData;
|
|
use std::os::raw::{c_char, c_void};
|
|
use std::ptr;
|
|
|
|
use crate::{check_status, sys, JsError, Result};
|
|
|
|
pub struct FuturePromise<Data, Resolver: FnOnce(sys::napi_env, Data) -> Result<sys::napi_value>> {
|
|
deferred: sys::napi_deferred,
|
|
env: sys::napi_env,
|
|
tsfn: sys::napi_threadsafe_function,
|
|
async_resource_name: sys::napi_value,
|
|
resolver: Resolver,
|
|
_data: PhantomData<Data>,
|
|
_value: PhantomData<sys::napi_value>,
|
|
}
|
|
|
|
unsafe impl<T, F: FnOnce(sys::napi_env, T) -> Result<sys::napi_value>> Send
|
|
for FuturePromise<T, F>
|
|
{
|
|
}
|
|
|
|
impl<Data, Resolver: FnOnce(sys::napi_env, Data) -> Result<sys::napi_value>>
|
|
FuturePromise<Data, Resolver>
|
|
{
|
|
pub fn new(env: sys::napi_env, dererred: sys::napi_deferred, resolver: Resolver) -> Result<Self> {
|
|
let mut async_resource_name = ptr::null_mut();
|
|
let s = "napi_resolve_promise_from_future";
|
|
check_status!(unsafe {
|
|
sys::napi_create_string_utf8(
|
|
env,
|
|
s.as_ptr() as *const c_char,
|
|
s.len(),
|
|
&mut async_resource_name,
|
|
)
|
|
})?;
|
|
|
|
Ok(FuturePromise {
|
|
deferred: dererred,
|
|
resolver,
|
|
env,
|
|
tsfn: ptr::null_mut(),
|
|
async_resource_name,
|
|
_data: PhantomData,
|
|
_value: PhantomData,
|
|
})
|
|
}
|
|
|
|
pub(crate) fn start(self) -> Result<TSFNValue> {
|
|
let mut tsfn_value = ptr::null_mut();
|
|
let async_resource_name = self.async_resource_name;
|
|
let env = self.env;
|
|
let self_ref = Box::leak(Box::from(self));
|
|
check_status!(unsafe {
|
|
sys::napi_create_threadsafe_function(
|
|
env,
|
|
ptr::null_mut(),
|
|
ptr::null_mut(),
|
|
async_resource_name,
|
|
0,
|
|
1,
|
|
ptr::null_mut(),
|
|
None,
|
|
self_ref as *mut FuturePromise<Data, Resolver> as *mut c_void,
|
|
Some(call_js_cb::<Data, Resolver>),
|
|
&mut tsfn_value,
|
|
)
|
|
})?;
|
|
self_ref.tsfn = tsfn_value;
|
|
Ok(TSFNValue(tsfn_value))
|
|
}
|
|
}
|
|
|
|
pub(crate) struct TSFNValue(sys::napi_threadsafe_function);
|
|
|
|
unsafe impl Send for TSFNValue {}
|
|
|
|
pub(crate) async fn resolve_from_future<Data: Send, Fut: Future<Output = Result<Data>>>(
|
|
tsfn_value: TSFNValue,
|
|
fut: Fut,
|
|
) {
|
|
let val = fut.await;
|
|
check_status!(unsafe {
|
|
sys::napi_call_threadsafe_function(
|
|
tsfn_value.0,
|
|
Box::into_raw(Box::from(val)) as *mut Data as *mut c_void,
|
|
sys::napi_threadsafe_function_call_mode::napi_tsfn_nonblocking,
|
|
)
|
|
})
|
|
.expect("Failed to call thread safe function");
|
|
check_status!(unsafe {
|
|
sys::napi_release_threadsafe_function(
|
|
tsfn_value.0,
|
|
sys::napi_threadsafe_function_release_mode::napi_tsfn_release,
|
|
)
|
|
})
|
|
.expect("Failed to release thread safe function");
|
|
}
|
|
|
|
unsafe extern "C" fn call_js_cb<
|
|
Data,
|
|
Resolver: FnOnce(sys::napi_env, Data) -> Result<sys::napi_value>,
|
|
>(
|
|
env: sys::napi_env,
|
|
_js_callback: sys::napi_value,
|
|
context: *mut c_void,
|
|
data: *mut c_void,
|
|
) {
|
|
let future_promise = Box::from_raw(context as *mut FuturePromise<Data, Resolver>);
|
|
let value = Box::from_raw(data as *mut Result<Data>);
|
|
let resolver = future_promise.resolver;
|
|
let deferred = future_promise.deferred;
|
|
let js_value_to_resolve = value.and_then(move |v| (resolver)(env, v));
|
|
match js_value_to_resolve {
|
|
Ok(v) => {
|
|
let status = sys::napi_resolve_deferred(env, deferred, v);
|
|
debug_assert!(status == sys::Status::napi_ok, "Resolve promise failed");
|
|
}
|
|
Err(e) => {
|
|
let status = sys::napi_reject_deferred(env, deferred, JsError::from(e).into_value(env));
|
|
debug_assert!(status == sys::Status::napi_ok, "Reject promise failed");
|
|
}
|
|
};
|
|
}
|