90 lines
2 KiB
Rust
90 lines
2 KiB
Rust
use std::convert::TryInto;
|
|
|
|
use napi::{
|
|
CallContext, Env, Error, JsBuffer, JsBufferValue, JsNumber, JsObject, Ref, Result, Task,
|
|
};
|
|
|
|
struct ComputeFib {
|
|
n: u32,
|
|
}
|
|
|
|
impl ComputeFib {
|
|
pub fn new(n: u32) -> ComputeFib {
|
|
ComputeFib { n }
|
|
}
|
|
}
|
|
|
|
impl Task for ComputeFib {
|
|
type Output = u32;
|
|
type JsValue = JsNumber;
|
|
|
|
fn compute(&mut self) -> Result<Self::Output> {
|
|
Ok(fibonacci_native(self.n))
|
|
}
|
|
|
|
fn resolve(self, env: Env, output: Self::Output) -> Result<Self::JsValue> {
|
|
env.create_uint32(output)
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
fn fibonacci_native(n: u32) -> u32 {
|
|
match n {
|
|
1 | 2 => 1,
|
|
_ => fibonacci_native(n - 1) + fibonacci_native(n - 2),
|
|
}
|
|
}
|
|
|
|
#[js_function(1)]
|
|
fn test_spawn_thread(ctx: CallContext) -> Result<JsObject> {
|
|
let n = ctx.get::<JsNumber>(0)?;
|
|
let task = ComputeFib::new(n.try_into()?);
|
|
let async_promise = ctx.env.spawn(task)?;
|
|
Ok(async_promise.promise_object())
|
|
}
|
|
|
|
struct CountBufferLength {
|
|
data: Ref<JsBufferValue>,
|
|
}
|
|
|
|
impl CountBufferLength {
|
|
pub fn new(data: Ref<JsBufferValue>) -> Self {
|
|
Self { data }
|
|
}
|
|
}
|
|
|
|
impl Task for CountBufferLength {
|
|
type Output = usize;
|
|
type JsValue = JsNumber;
|
|
|
|
fn compute(&mut self) -> Result<Self::Output> {
|
|
if self.data.len() == 10 {
|
|
return Err(Error::from_reason("len can't be 5".to_string()));
|
|
}
|
|
Ok((&self.data).len())
|
|
}
|
|
|
|
fn resolve(self, env: Env, output: Self::Output) -> Result<Self::JsValue> {
|
|
self.data.unref(env)?;
|
|
env.create_uint32(output as _)
|
|
}
|
|
|
|
fn reject(self, env: Env, err: Error) -> Result<Self::JsValue> {
|
|
self.data.unref(env)?;
|
|
Err(err)
|
|
}
|
|
}
|
|
|
|
#[js_function(1)]
|
|
fn test_spawn_thread_with_ref(ctx: CallContext) -> Result<JsObject> {
|
|
let n = ctx.get::<JsBuffer>(0)?.into_ref()?;
|
|
let task = CountBufferLength::new(n);
|
|
let async_work_promise = ctx.env.spawn(task)?;
|
|
Ok(async_work_promise.promise_object())
|
|
}
|
|
|
|
pub fn register_js(exports: &mut JsObject) -> Result<()> {
|
|
exports.create_named_method("testSpawnThread", test_spawn_thread)?;
|
|
exports.create_named_method("testSpawnThreadWithRef", test_spawn_thread_with_ref)?;
|
|
Ok(())
|
|
}
|