trilium/src/services/cls.js

const clsHooked = require('cls-hooked');
const namespace = clsHooked.createNamespace("trilium");

function init(callback) {
    return namespace.runAndReturn(callback);
}

function wrap(callback) {
    return () => init(callback);
}

function get(key) {
    return namespace.get(key);
}

function set(key, value) {
    namespace.set(key, value);
}

function getSourceId() {
    return namespace.get('sourceId');
}

function getLocalNowDateTime() {
    return namespace.get('localNowDateTime');
}

function disableEntityEvents() {
    namespace.set('disableEntityEvents', true);
}

function isEntityEventsDisabled() {
    return !!namespace.get('disableEntityEvents');
}

function getAndClearSyncRows() {
    const syncRows = namespace.get('syncRows') || [];

    namespace.set('syncRows', []);

    return syncRows;
}

function addSyncRow(syncRow) {
    const syncRows = namespace.get('syncRows') || [];

    syncRows.push(syncRow);

    namespace.set('syncRows', syncRows);
}

function reset() {
    clsHooked.reset();
}

function getEntityFromCache(entityName, entityId) {
    return namespace.get(entityName + '-' + entityId);
}

function setEntityToCache(entityName, entityId, entity) {
    return namespace.set(entityName + '-' + entityId, entity);
}

module.exports = {
    init,
    wrap,
    get,
    set,
    namespace,
    getSourceId,
    getLocalNowDateTime,
    disableEntityEvents,
    isEntityEventsDisabled,
    reset,
    getAndClearSyncRows,
    addSyncRow,
    getEntityFromCache,
    setEntityToCache
};