2024-01-01 19:58:21 +00:00
|
|
|
// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license.
|
2023-03-17 17:53:57 +00:00
|
|
|
import { AsyncLocalStorage, AsyncResource } from "node:async_hooks";
|
2024-02-13 02:05:10 +00:00
|
|
|
import { assert, assertEquals } from "@std/assert/mod.ts";
|
2023-03-17 17:53:57 +00:00
|
|
|
|
|
|
|
Deno.test(async function foo() {
|
|
|
|
const asyncLocalStorage = new AsyncLocalStorage();
|
|
|
|
|
|
|
|
const out: string[] = [];
|
|
|
|
function logWithId(msg: string) {
|
|
|
|
const id = asyncLocalStorage.getStore();
|
|
|
|
out.push(`${id !== undefined ? id : "-"}: ${msg}`);
|
|
|
|
}
|
|
|
|
|
|
|
|
async function exec() {
|
|
|
|
logWithId("start");
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 100));
|
|
|
|
logWithId("finish");
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const foo of [1, 2, 3]) {
|
|
|
|
asyncLocalStorage.run(foo, exec);
|
|
|
|
}
|
|
|
|
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 500));
|
|
|
|
|
|
|
|
assertEquals(out, [
|
|
|
|
"1: start",
|
|
|
|
"2: start",
|
|
|
|
"3: start",
|
|
|
|
"1: finish",
|
|
|
|
"2: finish",
|
|
|
|
"3: finish",
|
|
|
|
]);
|
|
|
|
});
|
|
|
|
|
|
|
|
Deno.test(async function bar() {
|
|
|
|
let differentScopeDone = false;
|
|
|
|
const als = new AsyncLocalStorage();
|
|
|
|
const ac = new AbortController();
|
2023-04-26 12:54:03 +00:00
|
|
|
const server = Deno.serve({
|
|
|
|
signal: ac.signal,
|
|
|
|
port: 4000,
|
|
|
|
}, () => {
|
2023-03-17 17:53:57 +00:00
|
|
|
const differentScope = als.run(123, () =>
|
|
|
|
AsyncResource.bind(() => {
|
|
|
|
differentScopeDone = true;
|
|
|
|
}));
|
|
|
|
return als.run("Hello World", async () => {
|
|
|
|
// differentScope is attached to a different async context, so
|
|
|
|
// it will see a different value for als.getStore() (123)
|
|
|
|
setTimeout(differentScope, 5);
|
|
|
|
// Some simulated async delay.
|
|
|
|
await new Promise((res) => setTimeout(res, 10));
|
|
|
|
return new Response(als.getStore() as string); // "Hello World"
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
const res = await fetch("http://localhost:4000");
|
|
|
|
assertEquals(await res.text(), "Hello World");
|
|
|
|
ac.abort();
|
2023-06-13 18:05:23 +00:00
|
|
|
await server.finished;
|
2023-03-17 17:53:57 +00:00
|
|
|
assert(differentScopeDone);
|
|
|
|
});
|
|
|
|
|
|
|
|
Deno.test(async function nested() {
|
|
|
|
const als = new AsyncLocalStorage();
|
2023-11-22 11:11:20 +00:00
|
|
|
const deferred = Promise.withResolvers();
|
|
|
|
const deferred1 = Promise.withResolvers();
|
2023-03-17 17:53:57 +00:00
|
|
|
|
|
|
|
als.run(null, () => {
|
|
|
|
als.run({ x: 1 }, () => {
|
2023-11-22 11:11:20 +00:00
|
|
|
deferred.resolve(als.getStore());
|
2023-03-17 17:53:57 +00:00
|
|
|
});
|
2023-11-22 11:11:20 +00:00
|
|
|
deferred1.resolve(als.getStore());
|
2023-03-17 17:53:57 +00:00
|
|
|
});
|
|
|
|
|
2023-11-22 11:11:20 +00:00
|
|
|
assertEquals(await deferred.promise, { x: 1 });
|
|
|
|
assertEquals(await deferred1.promise, null);
|
2023-03-17 17:53:57 +00:00
|
|
|
});
|
2024-03-06 12:05:57 +00:00
|
|
|
|
|
|
|
Deno.test(async function enterWith() {
|
|
|
|
const als = new AsyncLocalStorage();
|
|
|
|
const deferred = Promise.withResolvers();
|
|
|
|
const deferred1 = Promise.withResolvers();
|
|
|
|
|
|
|
|
als.run(null, () => {
|
|
|
|
als.run({ x: 1 }, () => {
|
|
|
|
als.enterWith({ x: 2 });
|
|
|
|
deferred.resolve(als.getStore());
|
|
|
|
});
|
|
|
|
deferred1.resolve(als.getStore());
|
|
|
|
});
|
|
|
|
|
|
|
|
assertEquals(await deferred.promise, { x: 2 });
|
|
|
|
assertEquals(await deferred1.promise, { x: 1 });
|
|
|
|
});
|
2024-04-02 22:53:29 +00:00
|
|
|
|
|
|
|
Deno.test(async function snapshot() {
|
|
|
|
const als = new AsyncLocalStorage();
|
|
|
|
const deferred = Promise.withResolvers();
|
|
|
|
|
|
|
|
als.run(null, () => {
|
|
|
|
const snapshot = AsyncLocalStorage.snapshot();
|
|
|
|
als.run({ x: 1 }, () => {
|
|
|
|
deferred.resolve(snapshot(() => als.getStore()));
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
assertEquals(await deferred.promise, null);
|
|
|
|
});
|
|
|
|
|
|
|
|
Deno.test(async function bind() {
|
|
|
|
const als = new AsyncLocalStorage();
|
|
|
|
const deferred = Promise.withResolvers();
|
|
|
|
|
|
|
|
const bound = als.run(null, () => {
|
|
|
|
return AsyncLocalStorage.bind(() => {
|
|
|
|
deferred.resolve(als.getStore());
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
als.run({ x: 1 }, () => {
|
|
|
|
bound();
|
|
|
|
});
|
|
|
|
|
|
|
|
assertEquals(await deferred.promise, null);
|
|
|
|
});
|
2024-04-15 12:54:42 +00:00
|
|
|
|
|
|
|
Deno.test(function asyncResourceStub() {
|
|
|
|
const resource = new AsyncResource("dbquery");
|
|
|
|
assert(typeof resource.asyncId() === "number");
|
|
|
|
});
|
2024-05-14 22:42:43 +00:00
|
|
|
|
|
|
|
Deno.test(function emitDestroyStub() {
|
|
|
|
const resource = new AsyncResource("foo");
|
|
|
|
assert(typeof resource.emitDestroy === "function");
|
|
|
|
});
|