1
0
Fork 0
mirror of https://github.com/denoland/deno.git synced 2024-12-22 15:24:46 -05:00
denoland-deno/runtime/js/40_signals.js
Bartek Iwańczuk c1fc7b2cd5
refactor: pull 'core', 'internals', 'primordials' from ES module (#21462)
This commit refactors how we access "core", "internals" and
"primordials" objects coming from `deno_core`, in our internal JavaScript code.

Instead of capturing them from "globalThis.__bootstrap" namespace, we
import them from recently added "ext:core/mod.js" file.
2023-12-07 14:21:01 +01:00

82 lines
2 KiB
JavaScript

// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
import { core, primordials } from "ext:core/mod.js";
const ops = core.ops;
const {
SafeSet,
SafeSetIterator,
SetPrototypeAdd,
SetPrototypeDelete,
TypeError,
} = primordials;
function bindSignal(signo) {
return ops.op_signal_bind(signo);
}
function pollSignal(rid) {
const promise = core.opAsync("op_signal_poll", rid);
core.unrefOpPromise(promise);
return promise;
}
function unbindSignal(rid) {
ops.op_signal_unbind(rid);
}
// Stores signal listeners and resource data. This has type of
// `Record<string, { rid: number | undefined, listeners: Set<() => void> }`
const signalData = {};
/** Gets the signal handlers and resource data of the given signal */
function getSignalData(signo) {
return signalData[signo] ??
(signalData[signo] = { rid: undefined, listeners: new SafeSet() });
}
function checkSignalListenerType(listener) {
if (typeof listener !== "function") {
throw new TypeError(
`Signal listener must be a function. "${typeof listener}" is given.`,
);
}
}
function addSignalListener(signo, listener) {
checkSignalListenerType(listener);
const sigData = getSignalData(signo);
SetPrototypeAdd(sigData.listeners, listener);
if (!sigData.rid) {
// If signal resource doesn't exist, create it.
// The program starts listening to the signal
sigData.rid = bindSignal(signo);
loop(sigData);
}
}
function removeSignalListener(signo, listener) {
checkSignalListenerType(listener);
const sigData = getSignalData(signo);
SetPrototypeDelete(sigData.listeners, listener);
if (sigData.listeners.size === 0 && sigData.rid) {
unbindSignal(sigData.rid);
sigData.rid = undefined;
}
}
async function loop(sigData) {
while (sigData.rid) {
if (await pollSignal(sigData.rid)) {
return;
}
for (const listener of new SafeSetIterator(sigData.listeners)) {
listener();
}
}
}
export { addSignalListener, removeSignalListener };