mirror of
https://github.com/denoland/deno
synced 2024-10-30 02:21:11 +00: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
109 lines
3.1 KiB
TypeScript
109 lines
3.1 KiB
TypeScript
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
|
|
// This module is the entry point for "worker" isolate, ie. the one
|
|
// that is created using `new Worker()` JS API.
|
|
//
|
|
// It provides a single function that should be called by Rust:
|
|
// - `bootstrapWorkerRuntime` - must be called once, when Isolate is created.
|
|
// It sets up runtime by providing globals for `DedicatedWorkerScope`.
|
|
|
|
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
import {
|
|
readOnly,
|
|
writable,
|
|
nonEnumerable,
|
|
windowOrWorkerGlobalScopeMethods,
|
|
windowOrWorkerGlobalScopeProperties,
|
|
eventTargetProperties
|
|
} from "./globals.ts";
|
|
import * as dispatch from "./dispatch.ts";
|
|
import { sendSync } from "./dispatch_json.ts";
|
|
import { log } from "./util.ts";
|
|
import { TextEncoder } from "./text_encoding.ts";
|
|
import * as runtime from "./runtime.ts";
|
|
|
|
const encoder = new TextEncoder();
|
|
|
|
// TODO(bartlomieju): remove these funtions
|
|
// Stuff for workers
|
|
export const onmessage: (e: { data: any }) => void = (): void => {};
|
|
export const onerror: (e: { data: any }) => void = (): void => {};
|
|
|
|
export function postMessage(data: any): void {
|
|
const dataJson = JSON.stringify(data);
|
|
const dataIntArray = encoder.encode(dataJson);
|
|
sendSync(dispatch.OP_WORKER_POST_MESSAGE, {}, dataIntArray);
|
|
}
|
|
|
|
let isClosing = false;
|
|
let hasBootstrapped = false;
|
|
|
|
export function close(): void {
|
|
if (isClosing) {
|
|
return;
|
|
}
|
|
|
|
isClosing = true;
|
|
sendSync(dispatch.OP_WORKER_CLOSE);
|
|
}
|
|
|
|
export async function workerMessageRecvCallback(data: string): Promise<void> {
|
|
let result: void | Promise<void>;
|
|
const event = { data };
|
|
|
|
try {
|
|
//
|
|
if (globalThis["onmessage"]) {
|
|
result = globalThis.onmessage!(event);
|
|
if (result && "then" in result) {
|
|
await result;
|
|
}
|
|
}
|
|
|
|
// TODO: run the rest of liteners
|
|
} catch (e) {
|
|
if (globalThis["onerror"]) {
|
|
const result = globalThis.onerror(
|
|
e.message,
|
|
e.fileName,
|
|
e.lineNumber,
|
|
e.columnNumber,
|
|
e
|
|
);
|
|
if (result === true) {
|
|
return;
|
|
}
|
|
}
|
|
throw e;
|
|
}
|
|
}
|
|
|
|
export const workerRuntimeGlobalProperties = {
|
|
self: readOnly(globalThis),
|
|
onmessage: writable(onmessage),
|
|
onerror: writable(onerror),
|
|
// TODO: should be readonly?
|
|
close: nonEnumerable(close),
|
|
postMessage: writable(postMessage),
|
|
workerMessageRecvCallback: nonEnumerable(workerMessageRecvCallback)
|
|
};
|
|
|
|
/**
|
|
* Main method to initialize worker runtime.
|
|
*
|
|
* It sets up global variables for DedicatedWorkerScope,
|
|
* and initializes ops.
|
|
*/
|
|
export function bootstrapWorkerRuntime(name: string): void {
|
|
if (hasBootstrapped) {
|
|
throw new Error("Worker runtime already bootstrapped");
|
|
}
|
|
log("bootstrapWorkerRuntime");
|
|
hasBootstrapped = true;
|
|
Object.defineProperties(globalThis, windowOrWorkerGlobalScopeMethods);
|
|
Object.defineProperties(globalThis, windowOrWorkerGlobalScopeProperties);
|
|
Object.defineProperties(globalThis, workerRuntimeGlobalProperties);
|
|
Object.defineProperties(globalThis, eventTargetProperties);
|
|
Object.defineProperties(globalThis, { name: readOnly(name) });
|
|
runtime.start(false, name);
|
|
}
|