mirror of
https://github.com/denoland/deno.git
synced 2024-12-22 07:14:47 -05:00
79b3bc05d6
* establish basic event loop for workers * make "self.close()" inside worker * remove "runWorkerMessageLoop() - instead manually call global function in Rust when message arrives. This is done in preparation for structured clone * refactor "WorkerChannel" and use distinct structs for internal and external channels; "WorkerChannelsInternal" and "WorkerHandle" * move "State.worker_channels_internal" to "Worker.internal_channels" * add "WorkerEvent" enum for child->host communication; currently "Message(Buf)" and "Error(ErrBox)" variants are supported * add tests for nested workers * add tests for worker throwing error on startup
65 lines
1.7 KiB
Rust
65 lines
1.7 KiB
Rust
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
use super::dispatch_json::{JsonOp, Value};
|
|
use crate::ops::json_op;
|
|
use crate::state::State;
|
|
use crate::worker::WorkerEvent;
|
|
use deno_core::*;
|
|
use futures;
|
|
use futures::channel::mpsc;
|
|
use futures::sink::SinkExt;
|
|
use std;
|
|
use std::convert::From;
|
|
|
|
pub fn web_worker_op<D>(
|
|
sender: mpsc::Sender<WorkerEvent>,
|
|
dispatcher: D,
|
|
) -> impl Fn(Value, Option<ZeroCopyBuf>) -> Result<JsonOp, ErrBox>
|
|
where
|
|
D: Fn(
|
|
&mpsc::Sender<WorkerEvent>,
|
|
Value,
|
|
Option<ZeroCopyBuf>,
|
|
) -> Result<JsonOp, ErrBox>,
|
|
{
|
|
move |args: Value, zero_copy: Option<ZeroCopyBuf>| -> Result<JsonOp, ErrBox> {
|
|
dispatcher(&sender, args, zero_copy)
|
|
}
|
|
}
|
|
|
|
pub fn init(i: &mut Isolate, s: &State, sender: &mpsc::Sender<WorkerEvent>) {
|
|
i.register_op(
|
|
"worker_post_message",
|
|
s.core_op(json_op(web_worker_op(
|
|
sender.clone(),
|
|
op_worker_post_message,
|
|
))),
|
|
);
|
|
i.register_op(
|
|
"worker_close",
|
|
s.core_op(json_op(web_worker_op(sender.clone(), op_worker_close))),
|
|
);
|
|
}
|
|
|
|
/// Post message to host as guest worker
|
|
fn op_worker_post_message(
|
|
sender: &mpsc::Sender<WorkerEvent>,
|
|
_args: Value,
|
|
data: Option<ZeroCopyBuf>,
|
|
) -> Result<JsonOp, ErrBox> {
|
|
let d = Vec::from(data.unwrap().as_ref()).into_boxed_slice();
|
|
let mut sender = sender.clone();
|
|
let fut = sender.send(WorkerEvent::Message(d));
|
|
futures::executor::block_on(fut).expect("Failed to post message to host");
|
|
Ok(JsonOp::Sync(json!({})))
|
|
}
|
|
|
|
/// Notify host that guest worker closes
|
|
fn op_worker_close(
|
|
sender: &mpsc::Sender<WorkerEvent>,
|
|
_args: Value,
|
|
_data: Option<ZeroCopyBuf>,
|
|
) -> Result<JsonOp, ErrBox> {
|
|
let mut sender = sender.clone();
|
|
sender.close_channel();
|
|
Ok(JsonOp::Sync(json!({})))
|
|
}
|