node-request-context/index.js at master · guyguyon/node-request-context · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
const asyncHooks = require('async_hooks');
const Namespace = require('./namespace');

const namespaces = {};

function createHooks(namespace) {
    function init(asyncId, type, triggerId, resource) {
        if (namespace.context[triggerId]) {
            namespace.context[asyncId] = namespace.context[triggerId];
        }
    }

    function destroy(asyncId) {
        delete namespace.context[asyncId];
    }

    const asyncHook = asyncHooks.createHook({ init, destroy });

    asyncHook.enable();
}

function createNamespace(name) {
    if (namespaces[name]) { throw new Error(`A namespace for ${name} already exists`); }

    const namespace = new Namespace();
    namespaces[name] = namespace;

    createHooks(namespace);

    return namespace;
}

function getNamespace(name) {
    // console.log('namespaces', namespaces);
    return namespaces[name];
}


module.exports = {
    createNamespace,
    getNamespace
};