mirror of https://github.com/nodejs/node.git
48 lines
918 B
JavaScript
48 lines
918 B
JavaScript
'use strict';
|
|
|
|
const {
|
|
ReflectApply,
|
|
} = primordials;
|
|
|
|
const AsyncContextFrame = require('internal/async_context_frame');
|
|
const { AsyncResource } = require('async_hooks');
|
|
|
|
class AsyncLocalStorage {
|
|
static bind(fn) {
|
|
return AsyncResource.bind(fn);
|
|
}
|
|
|
|
static snapshot() {
|
|
return AsyncLocalStorage.bind((cb, ...args) => cb(...args));
|
|
}
|
|
|
|
disable() {
|
|
AsyncContextFrame.disable(this);
|
|
}
|
|
|
|
enterWith(data) {
|
|
const frame = new AsyncContextFrame(this, data);
|
|
AsyncContextFrame.set(frame);
|
|
}
|
|
|
|
run(data, fn, ...args) {
|
|
const prior = AsyncContextFrame.current();
|
|
this.enterWith(data);
|
|
try {
|
|
return ReflectApply(fn, null, args);
|
|
} finally {
|
|
AsyncContextFrame.set(prior);
|
|
}
|
|
}
|
|
|
|
exit(fn, ...args) {
|
|
return this.run(undefined, fn, ...args);
|
|
}
|
|
|
|
getStore() {
|
|
return AsyncContextFrame.current()?.get(this);
|
|
}
|
|
}
|
|
|
|
module.exports = AsyncLocalStorage;
|