@karmaniverous/entity-manager 4.0.0 → 4.2.0

package/dist/default/lib/EntityManager/EntityManager.js

@@ -271,24 +271,28 @@ class EntityManager {
     // Query every shard in pageKeys.
     const shardQueryResults = await Promise.all((0, _map2.default)(pageKeys, (pageKey, shardedKey) => new Promise(resolve => shardQuery(shardedKey, pageKey, shardLimit).then(_ref6 => {
       let {
+        count,
         items,
         pageKey
       } = _ref6;
       return resolve({
-        shardedKey,
+        count,
         items,
-        pageKey
+        pageKey,
+        shardedKey
       });
     }))));
 
     // Reduce shardQueryResults into a single result.
     const result = shardQueryResults.reduce((shardedQueryResult, _ref7) => {
       let {
-        shardedKey,
+        count,
         items,
-        pageKey
+        pageKey,
+        shardedKey
       } = _ref7;
       return {
+        count: shardedQueryResult.count + count,
         items: [...shardedQueryResult.items, ...items],
         pageKeys: {
           ...shardedQueryResult.pageKeys,
@@ -298,6 +302,7 @@ class EntityManager {
         }
       };
     }, {
+      count: 0,
       items: [],
       pageKeys: {}
     });
@@ -367,8 +372,8 @@ class EntityManager {
     delete newItem[shardKeyToken];
 
     // Remove keys.
-    (0, _forEach2.default)(keys, (getValue, key) => {
-      delete newItem[key];
+    (0, _forEach2.default)(keys, (value, key) => {
+      if (!value.retain) delete newItem[key];
     });
     _classPrivateFieldGet(this, _entityManager).logger.debug('done', newItem);
     return newItem;

package/dist/default/lib/EntityManager/PrivateEntityManager.js

@@ -108,7 +108,10 @@ const keySchema = {
   type: 'object',
   properties: {
     encode: {},
-    decode: {}
+    decode: {},
+    retain: {
+      type: 'boolean'
+    }
   },
   additionalProperties: false,
   required: ['encode']

package/lib/EntityManager/EntityManager.js

@@ -260,7 +260,8 @@ export class EntityManager {
         (pageKey, shardedKey) =>
           new Promise((resolve) =>
             shardQuery(shardedKey, pageKey, shardLimit).then(
-              ({ items, pageKey }) => resolve({ shardedKey, items, pageKey })
+              ({ count, items, pageKey }) =>
+                resolve({ count, items, pageKey, shardedKey })
             )
           )
       )
@@ -268,14 +269,15 @@ export class EntityManager {
 
     // Reduce shardQueryResults into a single result.
     const result = shardQueryResults.reduce(
-      (shardedQueryResult, { shardedKey, items, pageKey }) => ({
+      (shardedQueryResult, { count, items, pageKey, shardedKey }) => ({
+        count: shardedQueryResult.count + count,
         items: [...shardedQueryResult.items, ...items],
         pageKeys: {
           ...shardedQueryResult.pageKeys,
           ...(_.isUndefined(pageKey) ? {} : { [shardedKey]: pageKey }),
         },
       }),
-      { items: [], pageKeys: {} }
+      { count: 0, items: [], pageKeys: {} }
     );
 
     return result;
@@ -341,8 +343,8 @@ export class EntityManager {
     delete newItem[shardKeyToken];
 
     // Remove keys.
-    _.forEach(keys, (getValue, key) => {
-      delete newItem[key];
+    _.forEach(keys, (value, key) => {
+      if (!value.retain) delete newItem[key];
     });
     this.#entityManager.logger.debug('done', newItem);
 

package/lib/EntityManager/PrivateEntityManager.js

@@ -72,6 +72,9 @@ const keySchema = {
   properties: {
     encode: {},
     decode: {},
+    retain: {
+      type: 'boolean',
+    },
   },
   additionalProperties: false,
   required: ['encode'],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@karmaniverous/entity-manager",
-  "version": "4.0.0",
+  "version": "4.2.0",
   "publishConfig": {
     "access": "public"
   },