/** @module cacheManager/multiCaching */ var async = require('async'); var domain = require('domain'); var CallbackFiller = require('./callback_filler'); /** * Module that lets you specify a hierarchy of caches. * * @param {array} caches - Array of caching objects. * @param {object} [options] * @param {function} [options.isCacheableValue] - A callback function which is called * with every value returned from cache or from a wrapped function. This lets you specify * which values should and should not be cached. If the function returns true, it will be * stored in cache. By default it caches everything except undefined. * * If an underlying cache specifies its own isCacheableValue function, that function will * be used instead of the multiCaching's _isCacheableValue function. */ var multiCaching = function(caches, options) { var self = {}; options = options || {}; if (!Array.isArray(caches)) { throw new Error('multiCaching requires an array of caches'); } var callbackFiller = new CallbackFiller(); if (typeof options.isCacheableValue === 'function') { self._isCacheableValue = options.isCacheableValue; } else { self._isCacheableValue = function(value) { return value !== undefined; }; } /** * If the underlying cache specifies its own isCacheableValue function (such * as how node-cache-manager-redis does), use that function, otherwise use * self._isCacheableValue function. */ function getIsCacheableValueFunction(cache) { if (cache.store && typeof cache.store.isCacheableValue === 'function') { return cache.store.isCacheableValue; } else { return self._isCacheableValue; } } function getFromHighestPriorityCache(key, options, cb) { if (typeof options === 'function') { cb = options; options = {}; } var promised = false; if (!cb) { cb = Promise.defer(); promised = true; } var i = 0; async.eachSeries(caches, function(cache, next) { var callback = function(err, result) { if (err) { return next(err); } var _isCacheableValue = getIsCacheableValueFunction(cache); if (_isCacheableValue(result)) { // break out of async loop. if (!promised) { return cb(err, result, i); } return cb.resolve(result); } i += 1; next(); }; cache.store.get(key, options, callback); }, function(err, result) { if (!promised) { return cb(err, result); } return (err) ? cb.reject(err) : cb.resolve(result); }); if (promised) { return cb.promise; } } function setInMultipleCaches(caches, opts, cb) { opts.options = opts.options || {}; var promised = false; if (!cb) { promised = true; cb = Promise.defer(); } async.each(caches, function(cache, next) { var _isCacheableValue = getIsCacheableValueFunction(cache); if (_isCacheableValue(opts.value)) { cache.store.set(opts.key, opts.value, opts.options, next); } else { next(); } }, function(err, result) { if (promised) { return (err) ? cb.reject(err) : cb.resolve(result); } else { cb(err, result); } }); if (promised) { return cb.promise; } } /** * Looks for an item in cache tiers. * When a key is found in a lower cache, all higher levels are updated. * * @param {string} key * @param {function} cb */ self.getAndPassUp = function(key, cb) { return getFromHighestPriorityCache(key, function(err, result, index) { if (err) { if (cb) { return cb(err); } } if (cb) { cb(err, result); } if (index) { var cachesToUpdate = caches.slice(0, index); async.each(cachesToUpdate, function(cache, next) { var _isCacheableValue = getIsCacheableValueFunction(cache); if (_isCacheableValue(result)) { // We rely on the cache module's default TTL cache.set(key, result, next); } }); } }); }; /** * Wraps a function in one or more caches. * Has same API as regular caching module. * * If a key doesn't exist in any cache, it gets set in all caches. * If a key exists in a high-priority (e.g., first) cache, it gets returned immediately * without getting set in other lower-priority caches. * If a key doesn't exist in a higher-priority cache but exists in a lower-priority * cache, it gets set in all higher-priority caches. * * @param {string} key - The cache key to use in cache operations * @param {function} work - The function to wrap * @param {object} [options] - options passed to `set` function * @param {function} cb */ self.wrap = function(key, work, options, cb) { if (typeof options === 'function') { cb = options; options = {}; } function getOptsForSet(value) { return { key: key, value: value, options: options }; } if (!cb) { cb = Promise.defer(); var work2 = work; work = function(cb) { Promise.resolve().then(work2).then(function(res) { cb(null, res); }) .catch(cb); }; } var hasKey = callbackFiller.has(key); callbackFiller.add(key, {cb: cb, domain: process.domain}); if (hasKey) { return; } getFromHighestPriorityCache(key, function(err, result, index) { if (err) { return callbackFiller.fill(key, err); } else if (self._isCacheableValue(result)) { var cachesToUpdate = caches.slice(0, index); var opts = getOptsForSet(result); setInMultipleCaches(cachesToUpdate, opts, function(err) { callbackFiller.fill(key, err, result); }); } else { domain .create() .on('error', function(err) { if (callbackFiller.has(key)) { callbackFiller.fill(key, err); } }) .bind(work)(function(err, data) { if (err) { return callbackFiller.fill(key, err); } if (!self._isCacheableValue(data)) { if (typeof cb === 'object') { return cb.resolve(data); } return cb(); } var opts = getOptsForSet(data); setInMultipleCaches(caches, opts, function(err) { callbackFiller.fill(key, err, data); }); }); } }); if (typeof cb === 'object') { return cb.promise; } }; /** * Set value in all caches * * @function * @name set * * @param {string} key * @param {*} value * @param {object} [options] to pass to underlying set function. * @param {function} [cb] */ self.set = function(key, value, options, cb) { if (typeof options === 'function') { cb = options; options = {}; } var opts = { key: key, value: value, options: options }; return setInMultipleCaches(caches, opts, cb); }; /** * Get value from highest level cache that has stored it. * * @function * @name get * * @param {string} key * @param {object} [options] to pass to underlying get function. * @param {function} cb */ self.get = function(key, options, cb) { if (typeof options === 'function') { cb = options; options = {}; } return getFromHighestPriorityCache(key, options, cb); }; /** * Delete value from all caches. * * @function * @name del * * @param {string} key * @param {object} [options] to pass to underlying del function. * @param {function} cb */ self.del = function(key, options, cb) { if (typeof options === 'function') { cb = options; options = {}; } async.each(caches, function(cache, next) { cache.store.del(key, options, next); }, cb); }; return self; }; module.exports = multiCaching;