@karmaniverous/entity-manager 1.0.0 → 2.0.0

package/README.md

@@ -221,7 +221,7 @@ This change can be accomplished with no breaking changes to existing implementat
             * [new exports.EntityManager(options)](#new_module_entity-manager.EntityManager_new)
             * [.addKeys(entityToken, item, [overwrite])](#module_entity-manager.EntityManager+addKeys) ⇒ <code>object</code>
             * [.getKeySpace(entityToken, keyToken, item, timestamp)](#module_entity-manager.EntityManager+getKeySpace) ⇒ <code>Array.&lt;string&gt;</code>
-            * [.query(entityToken, keyToken, item, shardQuery, [pageKeys])](#module_entity-manager.EntityManager+query) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
+            * [.query(entityToken, keyToken, item, shardQuery, [options])](#module_entity-manager.EntityManager+query) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
             * [.removeKeys(entityToken, item)](#module_entity-manager.EntityManager+removeKeys) ⇒ <code>object</code>
     * _inner_
         * [~ShardQueryResult](#module_entity-manager..ShardQueryResult) : <code>object</code>
@@ -239,7 +239,7 @@ Manage DynamoDb entities.
     * [new exports.EntityManager(options)](#new_module_entity-manager.EntityManager_new)
     * [.addKeys(entityToken, item, [overwrite])](#module_entity-manager.EntityManager+addKeys) ⇒ <code>object</code>
     * [.getKeySpace(entityToken, keyToken, item, timestamp)](#module_entity-manager.EntityManager+getKeySpace) ⇒ <code>Array.&lt;string&gt;</code>
-    * [.query(entityToken, keyToken, item, shardQuery, [pageKeys])](#module_entity-manager.EntityManager+query) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
+    * [.query(entityToken, keyToken, item, shardQuery, [options])](#module_entity-manager.EntityManager+query) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
     * [.removeKeys(entityToken, item)](#module_entity-manager.EntityManager+removeKeys) ⇒ <code>object</code>
 
 <a name="new_module_entity-manager.EntityManager_new"></a>
@@ -303,7 +303,7 @@ Return an array of sharded keys valid for a given entity token & timestamp.
 
 <a name="module_entity-manager.EntityManager+query"></a>
 
-#### entityManager.query(entityToken, keyToken, item, shardQuery, [pageKeys]) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
+#### entityManager.query(entityToken, keyToken, item, shardQuery, [options]) ⇒ <code>Promise.&lt;ShardedQueryResult&gt;</code>
 Query an entity across shards.
 
 **Kind**: instance method of [<code>EntityManager</code>](#module_entity-manager.EntityManager)  
@@ -315,7 +315,9 @@ Query an entity across shards.
 | keyToken | <code>string</code> | Key token. |
 | item | <code>object</code> | Entity item sufficiently populated to generate property keyToken. |
 | shardQuery | <code>ShardQueryFunction</code> | Sharded query function. |
-| [pageKeys] | <code>object</code> | Map of shard page keys. |
+| [options] | <code>object</code> | Options object. |
+| [options.limit] | <code>number</code> | Request limit. |
+| [options.pageKeys] | <code>object</code> | Map of shard page keys. |
 
 <a name="module_entity-manager.EntityManager+removeKeys"></a>
 
@@ -357,7 +359,9 @@ Shard query function
 | Param | Type | Description |
 | --- | --- | --- |
 | shardedKey | <code>string</code> | Sharded key. |
-| [pageKey] | <code>\*</code> | Page key. |
+| [options] | <code>object</code> | Options object. |
+| [options.limit] | <code>number</code> | Request limit. |
+| [options.pageKey] | <code>\*</code> | Page key. |
 
 <a name="module_entity-manager..ShardedQueryResult"></a>
 

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

@@ -4,9 +4,14 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 exports.EntityManager = void 0;
-var _isUndefined2 = _interopRequireDefault(require("lodash/isUndefined"));
 var _map2 = _interopRequireDefault(require("lodash/map"));
+var _size2 = _interopRequireDefault(require("lodash/size"));
+var _isEmpty2 = _interopRequireDefault(require("lodash/isEmpty"));
 var _fromPairs2 = _interopRequireDefault(require("lodash/fromPairs"));
+var _isPlainObject2 = _interopRequireDefault(require("lodash/isPlainObject"));
+var _isInteger2 = _interopRequireDefault(require("lodash/isInteger"));
+var _isUndefined2 = _interopRequireDefault(require("lodash/isUndefined"));
+var _isFunction2 = _interopRequireDefault(require("lodash/isFunction"));
 var _sortedUniq2 = _interopRequireDefault(require("lodash/sortedUniq"));
 var _forEach2 = _interopRequireDefault(require("lodash/forEach"));
 var _isNil2 = _interopRequireDefault(require("lodash/isNil"));
@@ -135,7 +140,9 @@ class EntityManager {
    *
    * @callback ShardQueryFunction
    * @param {string} shardedKey - Sharded key.
-   * @param {*} [pageKey] - Page key.
+   * @param {object} [options] - Options object.
+   * @param {number} [options.limit] - Request limit.
+   * @param {*} [options.pageKey] - Page key.
    * @returns {Promise<ShardQueryResult>} Sharded query result.
    */
 
@@ -152,31 +159,50 @@ class EntityManager {
    * @param {string} keyToken - Key token.
    * @param {object} item - Entity item sufficiently populated to generate property keyToken.
    * @param {ShardQueryFunction} shardQuery - Sharded query function.
-   * @param {object} [pageKeys] - Map of shard page keys.
+   * @param {object} [options] - Options object.
+   * @param {number} [options.limit] - Request limit.
+   * @param {object} [options.pageKeys] - Map of shard page keys.
    * @returns {Promise<ShardedQueryResult>} Sharded query result.
    */
-  async query(entityToken, keyToken, item, shardQuery, pageKeys) {
+  async query(entityToken, keyToken, item, shardQuery) {
+    let {
+      limit,
+      pageKeys
+    } = arguments.length > 4 && arguments[4] !== undefined ? arguments[4] : {};
+    // Validate params.
+    _classPrivateFieldGet(this, _entityManager).validateKeyToken(entityToken, keyToken);
+    _classPrivateFieldGet(this, _entityManager).validateItem(item);
+    if (!(0, _isFunction2.default)(shardQuery)) throw new Error('shardQuery must be a function');
+    if (!(0, _isUndefined2.default)(limit) && !(0, _isInteger2.default)(limit)) throw new Error('limit must be an integer');
+    if (!(0, _isUndefined2.default)(pageKeys) && !(0, _isPlainObject2.default)(pageKeys)) throw new Error('pageKeys must be an object');
+
     // Generate default pageKeys if not provided
     pageKeys ??= (0, _fromPairs2.default)(this.getKeySpace(entityToken, keyToken, item).map(shardedKey => [shardedKey, undefined]));
 
+    // Return empty result if no pageKeys.
+    if ((0, _isEmpty2.default)(pageKeys)) return {
+      items: [],
+      pageKeys: {}
+    };
+
+    // Calculate shard limit.
+    const shardLimit = (0, _isUndefined2.default)(limit) ? undefined : Math.ceil(limit / (0, _size2.default)(pageKeys));
+
     // Query every shard in pageKeys.
-    const shardQueryResults = await Promise.all((0, _map2.default)(pageKeys, (pageKey, shardedKey) => new Promise(resolve => shardQuery(shardedKey, pageKey).then(_ref2 => {
+    const shardQueryResults = await Promise.all((0, _map2.default)(pageKeys, (pageKey, shardedKey) => new Promise(resolve => shardQuery(shardedKey, {
+      pageKey,
+      limit: shardLimit
+    }).then(_ref2 => {
       let {
         items,
         pageKey
       } = _ref2;
-      console.log('shardQueryResult', {
-        shardedKey,
-        items,
-        pageKey
-      });
       return resolve({
         shardedKey,
         items,
         pageKey
       });
     }))));
-    console.log('shardQueryResults', shardQueryResults);
 
     // Reduce shardQueryResults into a single result.
     const result = shardQueryResults.reduce((shardedQueryResult, _ref3) => {

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

@@ -379,6 +379,22 @@ class PrivateEntityManager {
     } else return true;
   }
 
+  /**
+   * Tests whether a key token is valid.
+   *
+   * @param {string} entityToken - Entity token.
+   * @param {string} keyToken - Key token.
+   * @returns {boolean} true if keyToken is valid.
+   */
+  validateKeyToken(entityToken, keyToken) {
+    this.validateEntityToken(entityToken);
+    if (!this.config.entities[entityToken].keys[keyToken]) {
+      const message = `Invalid entity ${entityToken} keyToken: ${keyToken}`;
+      this.logger.error(message);
+      throw new Error(message);
+    } else return true;
+  }
+
   /**
    * Tests whether an item is valid.
    *

package/lib/EntityManager/EntityManager.js

@@ -130,7 +130,9 @@ export class EntityManager {
    *
    * @callback ShardQueryFunction
    * @param {string} shardedKey - Sharded key.
-   * @param {*} [pageKey] - Page key.
+   * @param {object} [options] - Options object.
+   * @param {number} [options.limit] - Request limit.
+   * @param {*} [options.pageKey] - Page key.
    * @returns {Promise<ShardQueryResult>} Sharded query result.
    */
 
@@ -147,10 +149,28 @@ export class EntityManager {
    * @param {string} keyToken - Key token.
    * @param {object} item - Entity item sufficiently populated to generate property keyToken.
    * @param {ShardQueryFunction} shardQuery - Sharded query function.
-   * @param {object} [pageKeys] - Map of shard page keys.
+   * @param {object} [options] - Options object.
+   * @param {number} [options.limit] - Request limit.
+   * @param {object} [options.pageKeys] - Map of shard page keys.
    * @returns {Promise<ShardedQueryResult>} Sharded query result.
    */
-  async query(entityToken, keyToken, item, shardQuery, pageKeys) {
+  async query(
+    entityToken,
+    keyToken,
+    item,
+    shardQuery,
+    { limit, pageKeys } = {}
+  ) {
+    // Validate params.
+    this.#entityManager.validateKeyToken(entityToken, keyToken);
+    this.#entityManager.validateItem(item);
+    if (!_.isFunction(shardQuery))
+      throw new Error('shardQuery must be a function');
+    if (!_.isUndefined(limit) && !_.isInteger(limit))
+      throw new Error('limit must be an integer');
+    if (!_.isUndefined(pageKeys) && !_.isPlainObject(pageKeys))
+      throw new Error('pageKeys must be an object');
+
     // Generate default pageKeys if not provided
     pageKeys ??= _.fromPairs(
       this.getKeySpace(entityToken, keyToken, item).map((shardedKey) => [
@@ -159,22 +179,27 @@ export class EntityManager {
       ])
     );
 
+    // Return empty result if no pageKeys.
+    if (_.isEmpty(pageKeys)) return { items: [], pageKeys: {} };
+
+    // Calculate shard limit.
+    const shardLimit = _.isUndefined(limit)
+      ? undefined
+      : Math.ceil(limit / _.size(pageKeys));
+
     // Query every shard in pageKeys.
     const shardQueryResults = await Promise.all(
       _.map(
         pageKeys,
         (pageKey, shardedKey) =>
           new Promise((resolve) =>
-            shardQuery(shardedKey, pageKey).then(({ items, pageKey }) => {
-              console.log('shardQueryResult', { shardedKey, items, pageKey });
-              return resolve({ shardedKey, items, pageKey });
-            })
+            shardQuery(shardedKey, { pageKey, limit: shardLimit }).then(
+              ({ items, pageKey }) => resolve({ shardedKey, items, pageKey })
+            )
           )
       )
     );
 
-    console.log('shardQueryResults', shardQueryResults);
-
     // Reduce shardQueryResults into a single result.
     const result = shardQueryResults.reduce(
       (shardedQueryResult, { shardedKey, items, pageKey }) => ({

package/lib/EntityManager/PrivateEntityManager.js

@@ -348,6 +348,23 @@ export class PrivateEntityManager {
     } else return true;
   }
 
+  /**
+   * Tests whether a key token is valid.
+   *
+   * @param {string} entityToken - Entity token.
+   * @param {string} keyToken - Key token.
+   * @returns {boolean} true if keyToken is valid.
+   */
+  validateKeyToken(entityToken, keyToken) {
+    this.validateEntityToken(entityToken);
+
+    if (!this.config.entities[entityToken].keys[keyToken]) {
+      const message = `Invalid entity ${entityToken} keyToken: ${keyToken}`;
+      this.logger.error(message);
+      throw new Error(message);
+    } else return true;
+  }
+
   /**
    * Tests whether an item is valid.
    *

package/package.json

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