74 lines
1.7 KiB
Rust
74 lines
1.7 KiB
Rust
|
#[macro_use]
|
||
|
|
extern crate napi_rs as napi;
|
||
|
|
|
||
|
|
use napi::{futures, Any, Env, Error, Object, Result, Status, Value};
|
||
|
|
|
||
|
|
register_module!(test_module, init);
|
||
|
|
|
||
|
|
fn init<'env>(
|
||
|
|
env: &'env Env,
|
||
|
|
exports: &'env mut Value<'env, Object>,
|
||
|
|
) -> Result<Option<Value<'env, Object>>> {
|
||
|
|
exports.set_named_property(
|
||
|
|
"testSpawn",
|
||
|
|
env.create_function("testSpawn", callback!(test_spawn)),
|
||
|
|
)?;
|
||
|
|
exports.set_named_property(
|
||
|
|
"testThrow",
|
||
|
|
env.create_function("testThrow", callback!(test_throw)),
|
||
|
|
)?;
|
||
|
|
Ok(None)
|
||
|
|
}
|
||
|
|
|
||
|
|
fn test_spawn<'a>(
|
||
|
|
env: &'a Env,
|
||
|
|
_this: Value<'a, Any>,
|
||
|
|
_args: &[Value<'a, Any>],
|
||
|
|
) -> Result<Option<Value<'a, Any>>> {
|
||
|
|
use futures::future::Executor;
|
||
|
|
use futures::{Future, Stream};
|
||
|
|
use std::{thread, time};
|
||
|
|
|
||
|
|
let async_context = env.async_init(None, "test_spawn");
|
||
|
|
let (promise, deferred) = env.create_promise();
|
||
|
|
let (tx, rx) = futures::sync::mpsc::unbounded();
|
||
|
|
|
||
|
|
let future = rx.for_each(|n: usize| {
|
||
|
|
println!("Received value {:?}", n);
|
||
|
|
futures::future::ok(())
|
||
|
|
}).and_then(move |_| {
|
||
|
|
async_context.enter(|env| {
|
||
|
|
env.resolve_deferred(deferred, env.get_undefined());
|
||
|
|
});
|
||
|
|
futures::future::ok(())
|
||
|
|
});
|
||
|
|
|
||
|
|
env.create_executor().execute(future).unwrap();
|
||
|
|
|
||
|
|
for _i in 0..10 {
|
||
|
|
let thread_tx = tx.clone();
|
||
|
|
thread::spawn(move || {
|
||
|
|
let mut n = 0;
|
||
|
|
loop {
|
||
|
|
println!("send {:?}", n);
|
||
|
|
thread_tx.unbounded_send(n).unwrap();
|
||
|
|
n += 1;
|
||
|
|
thread::sleep(time::Duration::from_millis(50));
|
||
|
|
if n == 10 {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(Some(promise.try_into().unwrap()))
|
||
|
|
}
|
||
|
|
|
||
|
|
fn test_throw<'a>(
|
||
|
|
_env: &'a Env,
|
||
|
|
_this: Value<'a, Any>,
|
||
|
|
_args: &[Value<'a, Any>],
|
||
|
|
) -> Result<Option<Value<'a, Any>>> {
|
||
|
|
Err(Error::new(Status::GenericFailure))
|
||
|
|
}
|