@autofleet/matmon 2.0.8 → 2.1.1

package/lib/cache.d.ts

@@ -6,3 +6,10 @@ export declare const getWithCacheSupport: ({ cacheKey, cacheGet, cacheSet, fetch
     fetching: any;
     skipCache: any;
 }) => Promise<any>;
+export declare const getMultipleWithCache: ({ getFromCache, setInCache, getter, multiGetter, idField, }: {
+    getFromCache: any;
+    setInCache: any;
+    getter?: any;
+    multiGetter?: any;
+    idField?: string;
+}) => (queries: any[]) => Promise<any[]>;

package/lib/cache.js

@@ -22,7 +22,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getWithCacheSupport = exports.getNewLRU = void 0;
+exports.getMultipleWithCache = exports.getWithCacheSupport = exports.getNewLRU = void 0;
 const lru_cache_1 = __importDefault(require("lru-cache"));
 const locking = __importStar(require("./locking"));
 const dotenv_1 = __importDefault(require("dotenv"));
@@ -77,3 +77,39 @@ exports.getWithCacheSupport = async ({ cacheKey, cacheGet, cacheSet, fetching, s
     }
     return valueToReturn;
 };
+const getIdField = (query, idField) => {
+    if (typeof query === 'string') {
+        return query;
+    }
+    return query[idField];
+};
+exports.getMultipleWithCache = ({ getFromCache, setInCache, getter = undefined, multiGetter = undefined, idField = 'id', }) => async (queries) => {
+    const queriesMap = new Map(queries.filter(Boolean).map(query => [getIdField(query, idField), query]));
+    const resultMap = new Map();
+    const valuesToPullFromCache = [...queriesMap.values()];
+    const valuesFromCache = await Promise.all(valuesToPullFromCache.map(query => getFromCache(query)));
+    valuesFromCache.filter(Boolean).map((value) => {
+        queriesMap.delete(getIdField(value, idField));
+        resultMap.set(getIdField(value, idField), value);
+        return value;
+    });
+    let valuesFromGetter = [];
+    if (queriesMap.size > 0) {
+        if (multiGetter) {
+            valuesFromGetter = await multiGetter([...queriesMap.values()]);
+        }
+        else {
+            valuesFromGetter = await Promise.all([...queriesMap.values()].map(id => getter(id)));
+        }
+        valuesFromGetter.forEach((value) => {
+            setInCache(value[idField], value);
+            resultMap.set(getIdField(value, idField), value);
+        });
+    }
+    return queries.map((query) => {
+        if (!query) {
+            return null;
+        }
+        return resultMap.get(getIdField(query, idField));
+    });
+};

package/lib/redis/index.d.ts

@@ -9,6 +9,7 @@ declare class RedisCache {
     private useLock;
     constructor(options: any);
     get(key: any): Promise<any>;
+    getMultiple(keys: any): Promise<any>;
     set(key: any, value: any): Promise<void>;
     remove(key: any): Promise<void>;
     removeMultiple(keys: any): Promise<void>;

package/lib/redis/index.js

@@ -58,6 +58,18 @@ class RedisCache {
         }
         return JSON.parse(value);
     }
+    async getMultiple(keys) {
+        const keysWithPrefix = keys.map(key => KEY_PREFIX + key);
+        let values;
+        try {
+            // Try to get the value from redis.
+            values = await this.client.mgetAsync(keysWithPrefix);
+        }
+        catch (err) {
+            throw new errors_1.RedisCacheError('Failed to get a value', err);
+        }
+        return values.map(value => JSON.parse(value));
+    }
     async set(key, value) {
         const keyWithPrefix = KEY_PREFIX + key;
         const ttl = parseInt(String(this.baseTTL * (Math.random() + 1)), 10);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/matmon",
-  "version": "2.0.8",
+  "version": "2.1.1",
   "description": "manage cache",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",