summaryrefslogtreecommitdiff
path: root/runtime/ops/web_worker.rs
blob: 17561882ebb1f8ca093982491af05a6dc54c092d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.

use crate::web_worker::WebWorkerHandle;
use crate::web_worker::WorkerEvent;
use deno_core::futures::channel::mpsc;
use deno_core::serde_json::{json, Value};

pub fn init(
  rt: &mut deno_core::JsRuntime,
  sender: mpsc::Sender<WorkerEvent>,
  handle: WebWorkerHandle,
) {
  // Post message to host as guest worker.
  let sender_ = sender.clone();
  super::reg_json_sync(
    rt,
    "op_worker_post_message",
    move |_state, _args: Value, bufs| {
      assert_eq!(bufs.len(), 1, "Invalid number of arguments");
      let msg_buf: Box<[u8]> = (*bufs[0]).into();
      sender_
        .clone()
        .try_send(WorkerEvent::Message(msg_buf))
        .expect("Failed to post message to host");
      Ok(json!({}))
    },
  );

  // Notify host that guest worker closes.
  super::reg_json_sync(
    rt,
    "op_worker_close",
    move |_state, _args: Value, _bufs| {
      // Notify parent that we're finished
      sender.clone().close_channel();
      // Terminate execution of current worker
      handle.terminate();
      Ok(json!({}))
    },
  );
}