mirror of
https://github.com/denoland/deno.git
synced 2024-10-31 09:14:20 -04:00
91 lines
2.7 KiB
TypeScript
91 lines
2.7 KiB
TypeScript
// Copyright 2018 the Deno authors. All rights reserved. MIT license.
|
|
import { libdeno } from "./libdeno";
|
|
import { flatbuffers } from "flatbuffers";
|
|
import * as msg from "gen/msg_generated";
|
|
import * as errors from "./errors";
|
|
import * as util from "./util";
|
|
import { maybePushTrace } from "./trace";
|
|
|
|
let nextCmdId = 0;
|
|
const promiseTable = new Map<number, util.Resolvable<msg.Base>>();
|
|
|
|
let fireTimers: () => void;
|
|
|
|
export function setFireTimersCallback(fn: () => void) {
|
|
fireTimers = fn;
|
|
}
|
|
|
|
export function handleAsyncMsgFromRust(ui8: Uint8Array) {
|
|
// If a the buffer is empty, recv() on the native side timed out and we
|
|
// did not receive a message.
|
|
if (ui8.length) {
|
|
const bb = new flatbuffers.ByteBuffer(ui8);
|
|
const base = msg.Base.getRootAsBase(bb);
|
|
const cmdId = base.cmdId();
|
|
const promise = promiseTable.get(cmdId);
|
|
util.assert(promise != null, `Expecting promise in table. ${cmdId}`);
|
|
promiseTable.delete(cmdId);
|
|
const err = errors.maybeError(base);
|
|
if (err != null) {
|
|
promise!.reject(err);
|
|
} else {
|
|
promise!.resolve(base);
|
|
}
|
|
}
|
|
// Fire timers that have become runnable.
|
|
fireTimers();
|
|
}
|
|
|
|
// @internal
|
|
export function sendAsync(
|
|
builder: flatbuffers.Builder,
|
|
innerType: msg.Any,
|
|
inner: flatbuffers.Offset,
|
|
data?: ArrayBufferView
|
|
): Promise<msg.Base> {
|
|
maybePushTrace(innerType, false); // add to trace if tracing
|
|
const [cmdId, resBuf] = sendInternal(builder, innerType, inner, data, false);
|
|
util.assert(resBuf == null);
|
|
const promise = util.createResolvable<msg.Base>();
|
|
promiseTable.set(cmdId, promise);
|
|
return promise;
|
|
}
|
|
|
|
// @internal
|
|
export function sendSync(
|
|
builder: flatbuffers.Builder,
|
|
innerType: msg.Any,
|
|
inner: flatbuffers.Offset,
|
|
data?: ArrayBufferView
|
|
): null | msg.Base {
|
|
maybePushTrace(innerType, true); // add to trace if tracing
|
|
const [cmdId, resBuf] = sendInternal(builder, innerType, inner, data, true);
|
|
util.assert(cmdId >= 0);
|
|
if (resBuf == null) {
|
|
return null;
|
|
} else {
|
|
const u8 = new Uint8Array(resBuf!);
|
|
const bb = new flatbuffers.ByteBuffer(u8);
|
|
const baseRes = msg.Base.getRootAsBase(bb);
|
|
errors.maybeThrowError(baseRes);
|
|
return baseRes;
|
|
}
|
|
}
|
|
|
|
function sendInternal(
|
|
builder: flatbuffers.Builder,
|
|
innerType: msg.Any,
|
|
inner: flatbuffers.Offset,
|
|
data: undefined | ArrayBufferView,
|
|
sync = true
|
|
): [number, null | Uint8Array] {
|
|
const cmdId = nextCmdId++;
|
|
msg.Base.startBase(builder);
|
|
msg.Base.addInner(builder, inner);
|
|
msg.Base.addInnerType(builder, innerType);
|
|
msg.Base.addSync(builder, sync);
|
|
msg.Base.addCmdId(builder, cmdId);
|
|
builder.finish(msg.Base.endBase(builder));
|
|
|
|
return [cmdId, libdeno.send(builder.asUint8Array(), data)];
|
|
}
|