pinpoint-node-agent/lib/instrumentation/context/async-context-local-storage.js

Pinpoint node agent provided by NAVER

/**
 * Pinpoint Node.js Agent
 * Copyright 2020-present NAVER Corp.
 * Apache License v2.0
 */

'use strict'

const { AsyncLocalStorage } = require('node:async_hooks')

class AsyncContextLocalStorage {
    constructor() {
        this.storage = new AsyncLocalStorage()
    }

    run(store, callback) {
        return this.storage.run(store, callback)
    }

    getStore() {
        return this.storage.getStore()
    }

    disable() {
        this.storage.disable()
    }

    enterWith(store) {
        this.storage.enterWith(store)
    }
}

module.exports = AsyncContextLocalStorage