mirror of
https://github.com/denoland/deno.git
synced 2024-10-31 09:14:20 -04:00
104 lines
2.5 KiB
Rust
104 lines
2.5 KiB
Rust
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
use super::dispatch_json::{JsonOp, Value};
|
|
use crate::op_error::OpError;
|
|
use crate::ops::json_op;
|
|
use crate::ops::JsonOpDispatcher;
|
|
use crate::state::State;
|
|
use crate::web_worker::WebWorkerHandle;
|
|
use crate::worker::WorkerEvent;
|
|
use deno_core::CoreIsolate;
|
|
use deno_core::CoreIsolateState;
|
|
use deno_core::ZeroCopyBuf;
|
|
use futures::channel::mpsc;
|
|
use std::convert::From;
|
|
|
|
pub fn web_worker_op<D>(
|
|
sender: mpsc::Sender<WorkerEvent>,
|
|
dispatcher: D,
|
|
) -> impl JsonOpDispatcher
|
|
where
|
|
D: Fn(
|
|
&mpsc::Sender<WorkerEvent>,
|
|
Value,
|
|
&mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, OpError>,
|
|
{
|
|
move |_isolate_state: &mut CoreIsolateState,
|
|
args: Value,
|
|
zero_copy: &mut [ZeroCopyBuf]|
|
|
-> Result<JsonOp, OpError> { dispatcher(&sender, args, zero_copy) }
|
|
}
|
|
|
|
pub fn web_worker_op2<D>(
|
|
handle: WebWorkerHandle,
|
|
sender: mpsc::Sender<WorkerEvent>,
|
|
dispatcher: D,
|
|
) -> impl JsonOpDispatcher
|
|
where
|
|
D: Fn(
|
|
WebWorkerHandle,
|
|
&mpsc::Sender<WorkerEvent>,
|
|
Value,
|
|
&mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, OpError>,
|
|
{
|
|
move |_isolate_state: &mut CoreIsolateState,
|
|
args: Value,
|
|
zero_copy: &mut [ZeroCopyBuf]|
|
|
-> Result<JsonOp, OpError> {
|
|
dispatcher(handle.clone(), &sender, args, zero_copy)
|
|
}
|
|
}
|
|
|
|
pub fn init(
|
|
i: &mut CoreIsolate,
|
|
s: &State,
|
|
sender: &mpsc::Sender<WorkerEvent>,
|
|
handle: WebWorkerHandle,
|
|
) {
|
|
i.register_op(
|
|
"op_worker_post_message",
|
|
s.core_op(json_op(web_worker_op(
|
|
sender.clone(),
|
|
op_worker_post_message,
|
|
))),
|
|
);
|
|
i.register_op(
|
|
"op_worker_close",
|
|
s.core_op(json_op(web_worker_op2(
|
|
handle,
|
|
sender.clone(),
|
|
op_worker_close,
|
|
))),
|
|
);
|
|
}
|
|
|
|
/// Post message to host as guest worker
|
|
fn op_worker_post_message(
|
|
sender: &mpsc::Sender<WorkerEvent>,
|
|
_args: Value,
|
|
data: &mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, OpError> {
|
|
assert_eq!(data.len(), 1, "Invalid number of arguments");
|
|
let d = Vec::from(&*data[0]).into_boxed_slice();
|
|
let mut sender = sender.clone();
|
|
sender
|
|
.try_send(WorkerEvent::Message(d))
|
|
.expect("Failed to post message to host");
|
|
Ok(JsonOp::Sync(json!({})))
|
|
}
|
|
|
|
/// Notify host that guest worker closes
|
|
fn op_worker_close(
|
|
handle: WebWorkerHandle,
|
|
sender: &mpsc::Sender<WorkerEvent>,
|
|
_args: Value,
|
|
_data: &mut [ZeroCopyBuf],
|
|
) -> Result<JsonOp, OpError> {
|
|
let mut sender = sender.clone();
|
|
// Notify parent that we're finished
|
|
sender.close_channel();
|
|
// Terminate execution of current worker
|
|
handle.terminate();
|
|
Ok(JsonOp::Sync(json!({})))
|
|
}
|