@azure/eventhubs-checkpointstore-table 1.0.0-alpha.20240112.1 → 1.0.0-alpha.20240126.2

package/dist/index.js

@@ -2,6 +2,7 @@
 
 Object.defineProperty(exports, '__esModule', { value: true });
 
+var tslib = require('tslib');
 var dataTables = require('@azure/data-tables');
 var logger$1 = require('@azure/logger');
 
@@ -54,33 +55,46 @@ class TableCheckpointStore {
      * @returns Partition ownership details of all the partitions that have had an owner.
      */
     async listOwnership(fullyQualifiedNamespace, eventHubName, consumerGroup) {
+        var _a, e_1, _b, _c;
         const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;
         const partitionOwnershipArray = [];
         const entitiesIter = this._tableClient.listEntities({
             queryOptions: { filter: dataTables.odata `PartitionKey eq ${partitionKey}` },
         });
         try {
-            for await (const entity of entitiesIter) {
-                if (!entity.timestamp) {
-                    throw new Error(`Unable to retrieve timestamp from partitionKey "${partitionKey}", rowKey "${entity.rowKey}"`);
+            try {
+                for (var _d = true, entitiesIter_1 = tslib.__asyncValues(entitiesIter), entitiesIter_1_1; entitiesIter_1_1 = await entitiesIter_1.next(), _a = entitiesIter_1_1.done, !_a; _d = true) {
+                    _c = entitiesIter_1_1.value;
+                    _d = false;
+                    const entity = _c;
+                    if (!entity.timestamp) {
+                        throw new Error(`Unable to retrieve timestamp from partitionKey "${partitionKey}", rowKey "${entity.rowKey}"`);
+                    }
+                    const partitionOwnership = {
+                        fullyQualifiedNamespace,
+                        eventHubName,
+                        consumerGroup,
+                        ownerId: entity.ownerid,
+                        partitionId: entity.rowKey,
+                        lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),
+                        etag: entity.etag,
+                    };
+                    partitionOwnershipArray.push(partitionOwnership);
                 }
-                const partitionOwnership = {
-                    fullyQualifiedNamespace,
-                    eventHubName,
-                    consumerGroup,
-                    ownerId: entity.ownerid,
-                    partitionId: entity.rowKey,
-                    lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),
-                    etag: entity.etag,
-                };
-                partitionOwnershipArray.push(partitionOwnership);
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = entitiesIter_1.return)) await _b.call(entitiesIter_1);
+                }
+                finally { if (e_1) throw e_1.error; }
             }
             return partitionOwnershipArray;
         }
         catch (err) {
             logger.warning(`Error occurred while fetching the list of entities`, err.message);
             logErrorStackTrace(err);
-            if (err?.name === "AbortError")
+            if ((err === null || err === void 0 ? void 0 : err.name) === "AbortError")
                 throw err;
             throw new Error(`Error occurred while fetching the list of entities. \n${err}`);
         }
@@ -98,7 +112,7 @@ class TableCheckpointStore {
     async claimOwnership(partitionOwnership) {
         const partitionOwnershipArray = [];
         for (const ownership of partitionOwnership) {
-            const updatedOwnership = { ...ownership };
+            const updatedOwnership = Object.assign({}, ownership);
             const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;
             const ownershipEntity = {
                 partitionKey: partitionKey,
@@ -163,20 +177,33 @@ class TableCheckpointStore {
      *  - `tracingOptions`: Options for configuring tracing.
      */
     async listCheckpoints(fullyQualifiedNamespace, eventHubName, consumerGroup) {
+        var _a, e_2, _b, _c;
         const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;
         const checkpoints = [];
         const entitiesIter = this._tableClient.listEntities({
             queryOptions: { filter: dataTables.odata `PartitionKey eq ${partitionKey}` },
         });
-        for await (const entity of entitiesIter) {
-            checkpoints.push({
-                consumerGroup,
-                eventHubName,
-                fullyQualifiedNamespace,
-                partitionId: entity.rowKey,
-                offset: parseInt(entity.offset, 10),
-                sequenceNumber: parseInt(entity.sequencenumber, 10),
-            });
+        try {
+            for (var _d = true, entitiesIter_2 = tslib.__asyncValues(entitiesIter), entitiesIter_2_1; entitiesIter_2_1 = await entitiesIter_2.next(), _a = entitiesIter_2_1.done, !_a; _d = true) {
+                _c = entitiesIter_2_1.value;
+                _d = false;
+                const entity = _c;
+                checkpoints.push({
+                    consumerGroup,
+                    eventHubName,
+                    fullyQualifiedNamespace,
+                    partitionId: entity.rowKey,
+                    offset: parseInt(entity.offset, 10),
+                    sequenceNumber: parseInt(entity.sequencenumber, 10),
+                });
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (!_d && !_a && (_b = entitiesIter_2.return)) await _b.call(entitiesIter_2);
+            }
+            finally { if (e_2) throw e_2.error; }
         }
         return checkpoints;
     }

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../src/log.ts","../src/tableCheckpointStore.ts"],"sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { createClientLogger } from \"@azure/logger\";\n\n/**\n * The `@azure/logger` configuration for this package.\n * This will output logs using the `azure:eventhubs-checkpointstore-table` namespace prefix.\n */\nexport const logger = createClientLogger(\"eventhubs-checkpointstore-table\");\n\n/**\n * Logs the error's stack trace to \"verbose\" if a stack trace is available.\n * @param error - Error containing a stack trace.\n * @internal\n */\nexport function logErrorStackTrace(error: unknown): void {\n  if (error && typeof error === \"object\" && \"stack\" in error) {\n    logger.verbose((error as any).stack);\n  }\n}\n","// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { Checkpoint, CheckpointStore, PartitionOwnership } from \"@azure/event-hubs\";\nimport { TableClient, TableInsertEntityHeaders, odata } from \"@azure/data-tables\";\nimport { logErrorStackTrace, logger } from \"./log\";\n\n/**\n *\n * Checks if the value contains a `Timestamp` field of type `string`.\n */\nfunction _hasTimestamp<T extends TableInsertEntityHeaders>(\n  value: T,\n): value is T & { Timestamp: string } {\n  return typeof (value as any).Timestamp === \"string\";\n}\n\n/**\n * A checkpoint entity of type CheckpointEntity to be stored in the table\n * @internal\n *\n */\nexport interface CheckpointEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  sequencenumber: string;\n  offset: string;\n}\n\n/**\n * An ownership entity of type PartitionOwnership to be stored in the table\n * @internal\n */\nexport interface PartitionOwnershipEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  ownerid: string;\n}\n\n/**\n * An implementation of CheckpointStore that uses Azure Table Storage to persist checkpoint data.\n */\nexport class TableCheckpointStore implements CheckpointStore {\n  private _tableClient: TableClient;\n\n  constructor(tableClient: TableClient) {\n    this._tableClient = tableClient;\n  }\n\n  /**\n   * Get the list of all existing partition ownership from the underlying data store. May return empty\n   * results if there are is no existing ownership information.\n   * Partition Ownership contains the information on which `EventHubConsumerClient` subscribe call is currently processing the partition.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns Partition ownership details of all the partitions that have had an owner.\n   */\n  async listOwnership(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<PartitionOwnership[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n    const entitiesIter = this._tableClient.listEntities<PartitionOwnershipEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    try {\n      for await (const entity of entitiesIter) {\n        if (!entity.timestamp) {\n          throw new Error(\n            `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entity.rowKey}\"`,\n          );\n        }\n\n        const partitionOwnership: PartitionOwnership = {\n          fullyQualifiedNamespace,\n          eventHubName,\n          consumerGroup,\n          ownerId: entity.ownerid,\n          partitionId: entity.rowKey,\n          lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),\n          etag: entity.etag,\n        };\n        partitionOwnershipArray.push(partitionOwnership);\n      }\n      return partitionOwnershipArray;\n    } catch (err: any) {\n      logger.warning(`Error occurred while fetching the list of entities`, err.message);\n      logErrorStackTrace(err);\n      if (err?.name === \"AbortError\") throw err;\n\n      throw new Error(`Error occurred while fetching the list of entities. \\n${err}`);\n    }\n  }\n\n  /**\n   * Claim ownership of a list of partitions. This will return the list of partitions that were\n   * successfully claimed.\n   *\n   * @param partitionOwnership - The list of partition ownership this instance is claiming to own.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A list partitions this instance successfully claimed ownership.\n   */\n  async claimOwnership(partitionOwnership: PartitionOwnership[]): Promise<PartitionOwnership[]> {\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n\n    for (const ownership of partitionOwnership) {\n      const updatedOwnership = { ...ownership };\n      const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;\n      const ownershipEntity: PartitionOwnershipEntity = {\n        partitionKey: partitionKey,\n        rowKey: ownership.partitionId,\n        ownerid: ownership.ownerId,\n      };\n\n      // When we have an etag, we know the entity existed.\n      // If we encounter an error we should fail.\n      try {\n        if (ownership.etag) {\n          const updatedMetadata = await this._tableClient.updateEntity(ownershipEntity, \"Replace\", {\n            etag: ownership.etag,\n          });\n          const entityRetrieved = await this._tableClient.getEntity(\n            ownershipEntity.partitionKey,\n            ownershipEntity.rowKey,\n          );\n          if (!entityRetrieved.timestamp) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entityRetrieved.rowKey}\"`,\n            );\n          }\n          updatedOwnership.lastModifiedTimeInMs = new Date(entityRetrieved.timestamp).getTime();\n          updatedOwnership.etag = updatedMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n          logger.info(\n            `[${ownership.ownerId}] Claimed ownership successfully for partition: ${ownership.partitionId}`,\n            `LastModifiedTime: ${ownership.lastModifiedTimeInMs}, ETag: ${ownership.etag}`,\n          );\n        } else {\n          const newOwnershipMetadata = await this._tableClient.createEntity(ownershipEntity, {\n            requestOptions: {\n              customHeaders: {\n                Prefer: \"return-content\",\n              },\n            },\n          });\n\n          if (!_hasTimestamp(newOwnershipMetadata)) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${ownershipEntity.rowKey}\"`,\n            );\n          }\n\n          updatedOwnership.lastModifiedTimeInMs = new Date(\n            newOwnershipMetadata.Timestamp,\n          ).getTime();\n          updatedOwnership.etag = newOwnershipMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n        }\n      } catch (err: any) {\n        if (err.statusCode === 412) {\n          // etag failures (precondition not met) aren't fatal errors. They happen\n          // as multiple consumers attempt to claim the same partition (first one wins)\n          // and losers get this error.\n          logger.verbose(\n            `[${ownership.ownerId}] Did not claim partition ${ownership.partitionId}. Another processor has already claimed it.`,\n          );\n          continue;\n        }\n        logger.warning(\n          `Error occurred while claiming ownership for partition: ${ownership.partitionId}`,\n          err.message,\n        );\n        logErrorStackTrace(err);\n      }\n    }\n    return partitionOwnershipArray;\n  }\n\n  /**\n   * Lists all the checkpoints in a data store for a given namespace, eventhub and consumer group.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   */\n  async listCheckpoints(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<Checkpoint[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;\n    const checkpoints: Checkpoint[] = [];\n    const entitiesIter = this._tableClient.listEntities<CheckpointEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    for await (const entity of entitiesIter) {\n      checkpoints.push({\n        consumerGroup,\n        eventHubName,\n        fullyQualifiedNamespace,\n        partitionId: entity.rowKey,\n        offset: parseInt(entity.offset, 10),\n        sequenceNumber: parseInt(entity.sequencenumber, 10),\n      });\n    }\n    return checkpoints;\n  }\n\n  /**\n   * Updates the checkpoint in the data store for a partition.\n   *\n   * @param checkpoint - The checkpoint.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A promise that resolves when the checkpoint has been updated.\n   */\n  async updateCheckpoint(checkpoint: Checkpoint): Promise<void> {\n    const partitionKey = `${checkpoint.fullyQualifiedNamespace} ${checkpoint.eventHubName} ${checkpoint.consumerGroup} Checkpoint`;\n    const checkpointEntity: CheckpointEntity = {\n      partitionKey: partitionKey,\n      rowKey: checkpoint.partitionId,\n      sequencenumber: checkpoint.sequenceNumber.toString(),\n      offset: checkpoint.offset.toString(),\n    };\n    try {\n      await this._tableClient.upsertEntity(checkpointEntity);\n      logger.verbose(`Updated checkpoint successfully for partition: ${checkpoint.partitionId}`);\n      return;\n    } catch (err: any) {\n      logger.verbose(\n        `Error occurred while upating the checkpoint for partition: ${checkpoint.partitionId}.`,\n        err.message,\n      );\n      throw err;\n    }\n  }\n}\n"],"names":["createClientLogger","odata"],"mappings":";;;;;;;AAAA;AACA;AAIA;;;AAGG;MACU,MAAM,GAAGA,2BAAkB,CAAC,iCAAiC,EAAE;AAE5E;;;;AAIG;AACG,SAAU,kBAAkB,CAAC,KAAc,EAAA;IAC/C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,EAAE;AAC1D,QAAA,MAAM,CAAC,OAAO,CAAE,KAAa,CAAC,KAAK,CAAC,CAAC;AACtC,KAAA;AACH;;ACpBA;AACA;AAMA;;;AAGG;AACH,SAAS,aAAa,CACpB,KAAQ,EAAA;AAER,IAAA,OAAO,OAAQ,KAAa,CAAC,SAAS,KAAK,QAAQ,CAAC;AACtD,CAAC;AAwCD;;AAEG;MACU,oBAAoB,CAAA;AAG/B,IAAA,WAAA,CAAY,WAAwB,EAAA;AAClC,QAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;KACjC;AAED;;;;;;;;;;;;;AAaG;AACH,IAAA,MAAM,aAAa,CACjB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB,EAAA;QAErB,MAAM,YAAY,GAAG,CAAG,EAAA,uBAAuB,IAAI,YAAY,CAAA,CAAA,EAAI,aAAa,CAAA,UAAA,CAAY,CAAC;QAC7F,MAAM,uBAAuB,GAAyB,EAAE,CAAC;AACzD,QAAA,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAA2B;YAC5E,YAAY,EAAE,EAAE,MAAM,EAAEC,gBAAK,CAAA,CAAA,gBAAA,EAAmB,YAAY,CAAA,CAAE,EAAE;AACjE,SAAA,CAAC,CAAC;QACH,IAAI;AACF,YAAA,WAAW,MAAM,MAAM,IAAI,YAAY,EAAE;AACvC,gBAAA,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;oBACrB,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,MAAM,CAAC,MAAM,CAAG,CAAA,CAAA,CAC9F,CAAC;AACH,iBAAA;AAED,gBAAA,MAAM,kBAAkB,GAAuB;oBAC7C,uBAAuB;oBACvB,YAAY;oBACZ,aAAa;oBACb,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,oBAAoB,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;oBAC1D,IAAI,EAAE,MAAM,CAAC,IAAI;iBAClB,CAAC;AACF,gBAAA,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAClD,aAAA;AACD,YAAA,OAAO,uBAAuB,CAAC;AAChC,SAAA;AAAC,QAAA,OAAO,GAAQ,EAAE;YACjB,MAAM,CAAC,OAAO,CAAC,CAAA,kDAAA,CAAoD,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAClF,kBAAkB,CAAC,GAAG,CAAC,CAAC;AACxB,YAAA,IAAI,GAAG,EAAE,IAAI,KAAK,YAAY;AAAE,gBAAA,MAAM,GAAG,CAAC;AAE1C,YAAA,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,CAAA,CAAE,CAAC,CAAC;AACjF,SAAA;KACF;AAED;;;;;;;;;AASG;IACH,MAAM,cAAc,CAAC,kBAAwC,EAAA;QAC3D,MAAM,uBAAuB,GAAyB,EAAE,CAAC;AAEzD,QAAA,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE;AAC1C,YAAA,MAAM,gBAAgB,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC;AAC1C,YAAA,MAAM,YAAY,GAAG,CAAG,EAAA,SAAS,CAAC,uBAAuB,CAAA,CAAA,EAAI,SAAS,CAAC,YAAY,CAAI,CAAA,EAAA,SAAS,CAAC,aAAa,YAAY,CAAC;AAC3H,YAAA,MAAM,eAAe,GAA6B;AAChD,gBAAA,YAAY,EAAE,YAAY;gBAC1B,MAAM,EAAE,SAAS,CAAC,WAAW;gBAC7B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC;;;YAIF,IAAI;gBACF,IAAI,SAAS,CAAC,IAAI,EAAE;AAClB,oBAAA,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,EAAE;wBACvF,IAAI,EAAE,SAAS,CAAC,IAAI;AACrB,qBAAA,CAAC,CAAC;AACH,oBAAA,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CACvD,eAAe,CAAC,YAAY,EAC5B,eAAe,CAAC,MAAM,CACvB,CAAC;AACF,oBAAA,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE;wBAC9B,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,eAAe,CAAC,MAAM,CAAG,CAAA,CAAA,CACvG,CAAC;AACH,qBAAA;AACD,oBAAA,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;AACtF,oBAAA,gBAAgB,CAAC,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;AAC7C,oBAAA,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/C,MAAM,CAAC,IAAI,CACT,CAAA,CAAA,EAAI,SAAS,CAAC,OAAO,CAAmD,gDAAA,EAAA,SAAS,CAAC,WAAW,EAAE,EAC/F,CAAA,kBAAA,EAAqB,SAAS,CAAC,oBAAoB,CAAA,QAAA,EAAW,SAAS,CAAC,IAAI,CAAE,CAAA,CAC/E,CAAC;AACH,iBAAA;AAAM,qBAAA;oBACL,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE;AACjF,wBAAA,cAAc,EAAE;AACd,4BAAA,aAAa,EAAE;AACb,gCAAA,MAAM,EAAE,gBAAgB;AACzB,6BAAA;AACF,yBAAA;AACF,qBAAA,CAAC,CAAC;AAEH,oBAAA,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE;wBACxC,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,eAAe,CAAC,MAAM,CAAG,CAAA,CAAA,CACvG,CAAC;AACH,qBAAA;AAED,oBAAA,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAC9C,oBAAoB,CAAC,SAAS,CAC/B,CAAC,OAAO,EAAE,CAAC;AACZ,oBAAA,gBAAgB,CAAC,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC;AAClD,oBAAA,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;AAChD,iBAAA;AACF,aAAA;AAAC,YAAA,OAAO,GAAQ,EAAE;AACjB,gBAAA,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE;;;;AAI1B,oBAAA,MAAM,CAAC,OAAO,CACZ,CAAA,CAAA,EAAI,SAAS,CAAC,OAAO,CAAA,0BAAA,EAA6B,SAAS,CAAC,WAAW,CAAA,2CAAA,CAA6C,CACrH,CAAC;oBACF,SAAS;AACV,iBAAA;AACD,gBAAA,MAAM,CAAC,OAAO,CACZ,CAAA,uDAAA,EAA0D,SAAS,CAAC,WAAW,CAAA,CAAE,EACjF,GAAG,CAAC,OAAO,CACZ,CAAC;gBACF,kBAAkB,CAAC,GAAG,CAAC,CAAC;AACzB,aAAA;AACF,SAAA;AACD,QAAA,OAAO,uBAAuB,CAAC;KAChC;AAED;;;;;;;;;;AAUG;AACH,IAAA,MAAM,eAAe,CACnB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB,EAAA;QAErB,MAAM,YAAY,GAAG,CAAG,EAAA,uBAAuB,IAAI,YAAY,CAAA,CAAA,EAAI,aAAa,CAAA,WAAA,CAAa,CAAC;QAC9F,MAAM,WAAW,GAAiB,EAAE,CAAC;AACrC,QAAA,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAmB;YACpE,YAAY,EAAE,EAAE,MAAM,EAAEA,gBAAK,CAAA,CAAA,gBAAA,EAAmB,YAAY,CAAA,CAAE,EAAE;AACjE,SAAA,CAAC,CAAC;AACH,QAAA,WAAW,MAAM,MAAM,IAAI,YAAY,EAAE;YACvC,WAAW,CAAC,IAAI,CAAC;gBACf,aAAa;gBACb,YAAY;gBACZ,uBAAuB;gBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;gBAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC;AACpD,aAAA,CAAC,CAAC;AACJ,SAAA;AACD,QAAA,OAAO,WAAW,CAAC;KACpB;AAED;;;;;;;;AAQG;IACH,MAAM,gBAAgB,CAAC,UAAsB,EAAA;AAC3C,QAAA,MAAM,YAAY,GAAG,CAAG,EAAA,UAAU,CAAC,uBAAuB,CAAA,CAAA,EAAI,UAAU,CAAC,YAAY,CAAI,CAAA,EAAA,UAAU,CAAC,aAAa,aAAa,CAAC;AAC/H,QAAA,MAAM,gBAAgB,GAAqB;AACzC,YAAA,YAAY,EAAE,YAAY;YAC1B,MAAM,EAAE,UAAU,CAAC,WAAW;AAC9B,YAAA,cAAc,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ,EAAE;AACpD,YAAA,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,EAAE;SACrC,CAAC;QACF,IAAI;YACF,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,CAAA,+CAAA,EAAkD,UAAU,CAAC,WAAW,CAAE,CAAA,CAAC,CAAC;YAC3F,OAAO;AACR,SAAA;AAAC,QAAA,OAAO,GAAQ,EAAE;AACjB,YAAA,MAAM,CAAC,OAAO,CACZ,CAAA,2DAAA,EAA8D,UAAU,CAAC,WAAW,CAAA,CAAA,CAAG,EACvF,GAAG,CAAC,OAAO,CACZ,CAAC;AACF,YAAA,MAAM,GAAG,CAAC;AACX,SAAA;KACF;AACF;;;;;"}
+{"version":3,"file":"index.js","sources":["../src/log.ts","../src/tableCheckpointStore.ts"],"sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { createClientLogger } from \"@azure/logger\";\n\n/**\n * The `@azure/logger` configuration for this package.\n * This will output logs using the `azure:eventhubs-checkpointstore-table` namespace prefix.\n */\nexport const logger = createClientLogger(\"eventhubs-checkpointstore-table\");\n\n/**\n * Logs the error's stack trace to \"verbose\" if a stack trace is available.\n * @param error - Error containing a stack trace.\n * @internal\n */\nexport function logErrorStackTrace(error: unknown): void {\n  if (error && typeof error === \"object\" && \"stack\" in error) {\n    logger.verbose((error as any).stack);\n  }\n}\n","// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { Checkpoint, CheckpointStore, PartitionOwnership } from \"@azure/event-hubs\";\nimport { TableClient, TableInsertEntityHeaders, odata } from \"@azure/data-tables\";\nimport { logErrorStackTrace, logger } from \"./log\";\n\n/**\n *\n * Checks if the value contains a `Timestamp` field of type `string`.\n */\nfunction _hasTimestamp<T extends TableInsertEntityHeaders>(\n  value: T,\n): value is T & { Timestamp: string } {\n  return typeof (value as any).Timestamp === \"string\";\n}\n\n/**\n * A checkpoint entity of type CheckpointEntity to be stored in the table\n * @internal\n *\n */\nexport interface CheckpointEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  sequencenumber: string;\n  offset: string;\n}\n\n/**\n * An ownership entity of type PartitionOwnership to be stored in the table\n * @internal\n */\nexport interface PartitionOwnershipEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  ownerid: string;\n}\n\n/**\n * An implementation of CheckpointStore that uses Azure Table Storage to persist checkpoint data.\n */\nexport class TableCheckpointStore implements CheckpointStore {\n  private _tableClient: TableClient;\n\n  constructor(tableClient: TableClient) {\n    this._tableClient = tableClient;\n  }\n\n  /**\n   * Get the list of all existing partition ownership from the underlying data store. May return empty\n   * results if there are is no existing ownership information.\n   * Partition Ownership contains the information on which `EventHubConsumerClient` subscribe call is currently processing the partition.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns Partition ownership details of all the partitions that have had an owner.\n   */\n  async listOwnership(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<PartitionOwnership[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n    const entitiesIter = this._tableClient.listEntities<PartitionOwnershipEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    try {\n      for await (const entity of entitiesIter) {\n        if (!entity.timestamp) {\n          throw new Error(\n            `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entity.rowKey}\"`,\n          );\n        }\n\n        const partitionOwnership: PartitionOwnership = {\n          fullyQualifiedNamespace,\n          eventHubName,\n          consumerGroup,\n          ownerId: entity.ownerid,\n          partitionId: entity.rowKey,\n          lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),\n          etag: entity.etag,\n        };\n        partitionOwnershipArray.push(partitionOwnership);\n      }\n      return partitionOwnershipArray;\n    } catch (err: any) {\n      logger.warning(`Error occurred while fetching the list of entities`, err.message);\n      logErrorStackTrace(err);\n      if (err?.name === \"AbortError\") throw err;\n\n      throw new Error(`Error occurred while fetching the list of entities. \\n${err}`);\n    }\n  }\n\n  /**\n   * Claim ownership of a list of partitions. This will return the list of partitions that were\n   * successfully claimed.\n   *\n   * @param partitionOwnership - The list of partition ownership this instance is claiming to own.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A list partitions this instance successfully claimed ownership.\n   */\n  async claimOwnership(partitionOwnership: PartitionOwnership[]): Promise<PartitionOwnership[]> {\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n\n    for (const ownership of partitionOwnership) {\n      const updatedOwnership = { ...ownership };\n      const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;\n      const ownershipEntity: PartitionOwnershipEntity = {\n        partitionKey: partitionKey,\n        rowKey: ownership.partitionId,\n        ownerid: ownership.ownerId,\n      };\n\n      // When we have an etag, we know the entity existed.\n      // If we encounter an error we should fail.\n      try {\n        if (ownership.etag) {\n          const updatedMetadata = await this._tableClient.updateEntity(ownershipEntity, \"Replace\", {\n            etag: ownership.etag,\n          });\n          const entityRetrieved = await this._tableClient.getEntity(\n            ownershipEntity.partitionKey,\n            ownershipEntity.rowKey,\n          );\n          if (!entityRetrieved.timestamp) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entityRetrieved.rowKey}\"`,\n            );\n          }\n          updatedOwnership.lastModifiedTimeInMs = new Date(entityRetrieved.timestamp).getTime();\n          updatedOwnership.etag = updatedMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n          logger.info(\n            `[${ownership.ownerId}] Claimed ownership successfully for partition: ${ownership.partitionId}`,\n            `LastModifiedTime: ${ownership.lastModifiedTimeInMs}, ETag: ${ownership.etag}`,\n          );\n        } else {\n          const newOwnershipMetadata = await this._tableClient.createEntity(ownershipEntity, {\n            requestOptions: {\n              customHeaders: {\n                Prefer: \"return-content\",\n              },\n            },\n          });\n\n          if (!_hasTimestamp(newOwnershipMetadata)) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${ownershipEntity.rowKey}\"`,\n            );\n          }\n\n          updatedOwnership.lastModifiedTimeInMs = new Date(\n            newOwnershipMetadata.Timestamp,\n          ).getTime();\n          updatedOwnership.etag = newOwnershipMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n        }\n      } catch (err: any) {\n        if (err.statusCode === 412) {\n          // etag failures (precondition not met) aren't fatal errors. They happen\n          // as multiple consumers attempt to claim the same partition (first one wins)\n          // and losers get this error.\n          logger.verbose(\n            `[${ownership.ownerId}] Did not claim partition ${ownership.partitionId}. Another processor has already claimed it.`,\n          );\n          continue;\n        }\n        logger.warning(\n          `Error occurred while claiming ownership for partition: ${ownership.partitionId}`,\n          err.message,\n        );\n        logErrorStackTrace(err);\n      }\n    }\n    return partitionOwnershipArray;\n  }\n\n  /**\n   * Lists all the checkpoints in a data store for a given namespace, eventhub and consumer group.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   */\n  async listCheckpoints(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<Checkpoint[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;\n    const checkpoints: Checkpoint[] = [];\n    const entitiesIter = this._tableClient.listEntities<CheckpointEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    for await (const entity of entitiesIter) {\n      checkpoints.push({\n        consumerGroup,\n        eventHubName,\n        fullyQualifiedNamespace,\n        partitionId: entity.rowKey,\n        offset: parseInt(entity.offset, 10),\n        sequenceNumber: parseInt(entity.sequencenumber, 10),\n      });\n    }\n    return checkpoints;\n  }\n\n  /**\n   * Updates the checkpoint in the data store for a partition.\n   *\n   * @param checkpoint - The checkpoint.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A promise that resolves when the checkpoint has been updated.\n   */\n  async updateCheckpoint(checkpoint: Checkpoint): Promise<void> {\n    const partitionKey = `${checkpoint.fullyQualifiedNamespace} ${checkpoint.eventHubName} ${checkpoint.consumerGroup} Checkpoint`;\n    const checkpointEntity: CheckpointEntity = {\n      partitionKey: partitionKey,\n      rowKey: checkpoint.partitionId,\n      sequencenumber: checkpoint.sequenceNumber.toString(),\n      offset: checkpoint.offset.toString(),\n    };\n    try {\n      await this._tableClient.upsertEntity(checkpointEntity);\n      logger.verbose(`Updated checkpoint successfully for partition: ${checkpoint.partitionId}`);\n      return;\n    } catch (err: any) {\n      logger.verbose(\n        `Error occurred while upating the checkpoint for partition: ${checkpoint.partitionId}.`,\n        err.message,\n      );\n      throw err;\n    }\n  }\n}\n"],"names":["createClientLogger","odata","__asyncValues"],"mappings":";;;;;;;;AAAA;AACA;AAIA;;;AAGG;MACU,MAAM,GAAGA,2BAAkB,CAAC,iCAAiC,EAAE;AAE5E;;;;AAIG;AACG,SAAU,kBAAkB,CAAC,KAAc,EAAA;IAC/C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,EAAE;AAC1D,QAAA,MAAM,CAAC,OAAO,CAAE,KAAa,CAAC,KAAK,CAAC,CAAC;KACtC;AACH;;ACpBA;AACA;AAMA;;;AAGG;AACH,SAAS,aAAa,CACpB,KAAQ,EAAA;AAER,IAAA,OAAO,OAAQ,KAAa,CAAC,SAAS,KAAK,QAAQ,CAAC;AACtD,CAAC;AAwCD;;AAEG;MACU,oBAAoB,CAAA;AAG/B,IAAA,WAAA,CAAY,WAAwB,EAAA;AAClC,QAAA,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;KACjC;AAED;;;;;;;;;;;;;AAaG;AACH,IAAA,MAAM,aAAa,CACjB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB,EAAA;;QAErB,MAAM,YAAY,GAAG,CAAG,EAAA,uBAAuB,IAAI,YAAY,CAAA,CAAA,EAAI,aAAa,CAAA,UAAA,CAAY,CAAC;QAC7F,MAAM,uBAAuB,GAAyB,EAAE,CAAC;AACzD,QAAA,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAA2B;YAC5E,YAAY,EAAE,EAAE,MAAM,EAAEC,gBAAK,CAAA,CAAA,gBAAA,EAAmB,YAAY,CAAA,CAAE,EAAE;AACjE,SAAA,CAAC,CAAC;AACH,QAAA,IAAI;;AACF,gBAAA,KAA2B,eAAA,cAAA,GAAAC,mBAAA,CAAA,YAAY,CAAA,EAAA,gBAAA,8FAAE;oBAAd,EAAY,GAAA,gBAAA,CAAA,KAAA,CAAA;oBAAZ,EAAY,GAAA,KAAA,CAAA;oBAA5B,MAAM,MAAM,KAAA,CAAA;AACrB,oBAAA,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;wBACrB,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,MAAM,CAAC,MAAM,CAAG,CAAA,CAAA,CAC9F,CAAC;qBACH;AAED,oBAAA,MAAM,kBAAkB,GAAuB;wBAC7C,uBAAuB;wBACvB,YAAY;wBACZ,aAAa;wBACb,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;wBAC1B,oBAAoB,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;wBAC1D,IAAI,EAAE,MAAM,CAAC,IAAI;qBAClB,CAAC;AACF,oBAAA,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;iBAClD;;;;;;;;;AACD,YAAA,OAAO,uBAAuB,CAAC;SAChC;QAAC,OAAO,GAAQ,EAAE;YACjB,MAAM,CAAC,OAAO,CAAC,CAAA,kDAAA,CAAoD,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAClF,kBAAkB,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAA,GAAG,KAAH,IAAA,IAAA,GAAG,uBAAH,GAAG,CAAE,IAAI,MAAK,YAAY;AAAE,gBAAA,MAAM,GAAG,CAAC;AAE1C,YAAA,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,CAAA,CAAE,CAAC,CAAC;SACjF;KACF;AAED;;;;;;;;;AASG;IACH,MAAM,cAAc,CAAC,kBAAwC,EAAA;QAC3D,MAAM,uBAAuB,GAAyB,EAAE,CAAC;AAEzD,QAAA,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE;AAC1C,YAAA,MAAM,gBAAgB,GAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EAAQ,SAAS,CAAE,CAAC;AAC1C,YAAA,MAAM,YAAY,GAAG,CAAG,EAAA,SAAS,CAAC,uBAAuB,CAAA,CAAA,EAAI,SAAS,CAAC,YAAY,CAAI,CAAA,EAAA,SAAS,CAAC,aAAa,YAAY,CAAC;AAC3H,YAAA,MAAM,eAAe,GAA6B;AAChD,gBAAA,YAAY,EAAE,YAAY;gBAC1B,MAAM,EAAE,SAAS,CAAC,WAAW;gBAC7B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC;;;AAIF,YAAA,IAAI;AACF,gBAAA,IAAI,SAAS,CAAC,IAAI,EAAE;AAClB,oBAAA,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,EAAE;wBACvF,IAAI,EAAE,SAAS,CAAC,IAAI;AACrB,qBAAA,CAAC,CAAC;AACH,oBAAA,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CACvD,eAAe,CAAC,YAAY,EAC5B,eAAe,CAAC,MAAM,CACvB,CAAC;AACF,oBAAA,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE;wBAC9B,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,eAAe,CAAC,MAAM,CAAG,CAAA,CAAA,CACvG,CAAC;qBACH;AACD,oBAAA,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;AACtF,oBAAA,gBAAgB,CAAC,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;AAC7C,oBAAA,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/C,MAAM,CAAC,IAAI,CACT,CAAA,CAAA,EAAI,SAAS,CAAC,OAAO,CAAmD,gDAAA,EAAA,SAAS,CAAC,WAAW,EAAE,EAC/F,CAAA,kBAAA,EAAqB,SAAS,CAAC,oBAAoB,CAAA,QAAA,EAAW,SAAS,CAAC,IAAI,CAAE,CAAA,CAC/E,CAAC;iBACH;qBAAM;oBACL,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE;AACjF,wBAAA,cAAc,EAAE;AACd,4BAAA,aAAa,EAAE;AACb,gCAAA,MAAM,EAAE,gBAAgB;AACzB,6BAAA;AACF,yBAAA;AACF,qBAAA,CAAC,CAAC;AAEH,oBAAA,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE;wBACxC,MAAM,IAAI,KAAK,CACb,CAAmD,gDAAA,EAAA,YAAY,CAAc,WAAA,EAAA,eAAe,CAAC,MAAM,CAAG,CAAA,CAAA,CACvG,CAAC;qBACH;AAED,oBAAA,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAC9C,oBAAoB,CAAC,SAAS,CAC/B,CAAC,OAAO,EAAE,CAAC;AACZ,oBAAA,gBAAgB,CAAC,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC;AAClD,oBAAA,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;iBAChD;aACF;YAAC,OAAO,GAAQ,EAAE;AACjB,gBAAA,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE;;;;AAI1B,oBAAA,MAAM,CAAC,OAAO,CACZ,CAAA,CAAA,EAAI,SAAS,CAAC,OAAO,CAAA,0BAAA,EAA6B,SAAS,CAAC,WAAW,CAAA,2CAAA,CAA6C,CACrH,CAAC;oBACF,SAAS;iBACV;AACD,gBAAA,MAAM,CAAC,OAAO,CACZ,CAAA,uDAAA,EAA0D,SAAS,CAAC,WAAW,CAAA,CAAE,EACjF,GAAG,CAAC,OAAO,CACZ,CAAC;gBACF,kBAAkB,CAAC,GAAG,CAAC,CAAC;aACzB;SACF;AACD,QAAA,OAAO,uBAAuB,CAAC;KAChC;AAED;;;;;;;;;;AAUG;AACH,IAAA,MAAM,eAAe,CACnB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB,EAAA;;QAErB,MAAM,YAAY,GAAG,CAAG,EAAA,uBAAuB,IAAI,YAAY,CAAA,CAAA,EAAI,aAAa,CAAA,WAAA,CAAa,CAAC;QAC9F,MAAM,WAAW,GAAiB,EAAE,CAAC;AACrC,QAAA,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAmB;YACpE,YAAY,EAAE,EAAE,MAAM,EAAED,gBAAK,CAAA,CAAA,gBAAA,EAAmB,YAAY,CAAA,CAAE,EAAE;AACjE,SAAA,CAAC,CAAC;;AACH,YAAA,KAA2B,eAAA,cAAA,GAAAC,mBAAA,CAAA,YAAY,CAAA,EAAA,gBAAA,8FAAE;gBAAd,EAAY,GAAA,gBAAA,CAAA,KAAA,CAAA;gBAAZ,EAAY,GAAA,KAAA,CAAA;gBAA5B,MAAM,MAAM,KAAA,CAAA;gBACrB,WAAW,CAAC,IAAI,CAAC;oBACf,aAAa;oBACb,YAAY;oBACZ,uBAAuB;oBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;oBACnC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC;AACpD,iBAAA,CAAC,CAAC;aACJ;;;;;;;;;AACD,QAAA,OAAO,WAAW,CAAC;KACpB;AAED;;;;;;;;AAQG;IACH,MAAM,gBAAgB,CAAC,UAAsB,EAAA;AAC3C,QAAA,MAAM,YAAY,GAAG,CAAG,EAAA,UAAU,CAAC,uBAAuB,CAAA,CAAA,EAAI,UAAU,CAAC,YAAY,CAAI,CAAA,EAAA,UAAU,CAAC,aAAa,aAAa,CAAC;AAC/H,QAAA,MAAM,gBAAgB,GAAqB;AACzC,YAAA,YAAY,EAAE,YAAY;YAC1B,MAAM,EAAE,UAAU,CAAC,WAAW;AAC9B,YAAA,cAAc,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ,EAAE;AACpD,YAAA,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,EAAE;SACrC,CAAC;AACF,QAAA,IAAI;YACF,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,CAAA,+CAAA,EAAkD,UAAU,CAAC,WAAW,CAAE,CAAA,CAAC,CAAC;YAC3F,OAAO;SACR;QAAC,OAAO,GAAQ,EAAE;AACjB,YAAA,MAAM,CAAC,OAAO,CACZ,CAAA,2DAAA,EAA8D,UAAU,CAAC,WAAW,CAAA,CAAA,CAAG,EACvF,GAAG,CAAC,OAAO,CACZ,CAAC;AACF,YAAA,MAAM,GAAG,CAAC;SACX;KACF;AACF;;;;;"}

package/dist-esm/src/log.js.map

@@ -1 +1 @@
-{"version":3,"file":"log.js","sourceRoot":"","sources":["../../src/log.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD;;;GAGG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG,kBAAkB,CAAC,iCAAiC,CAAC,CAAC;AAE5E;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,EAAE;QAC1D,MAAM,CAAC,OAAO,CAAE,KAAa,CAAC,KAAK,CAAC,CAAC;KACtC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { createClientLogger } from \"@azure/logger\";\n\n/**\n * The `@azure/logger` configuration for this package.\n * This will output logs using the `azure:eventhubs-checkpointstore-table` namespace prefix.\n */\nexport const logger = createClientLogger(\"eventhubs-checkpointstore-table\");\n\n/**\n * Logs the error's stack trace to \"verbose\" if a stack trace is available.\n * @param error - Error containing a stack trace.\n * @internal\n */\nexport function logErrorStackTrace(error: unknown): void {\n  if (error && typeof error === \"object\" && \"stack\" in error) {\n    logger.verbose((error as any).stack);\n  }\n}\n"]}
+{"version":3,"file":"log.js","sourceRoot":"","sources":["../../src/log.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEnD;;;GAGG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG,kBAAkB,CAAC,iCAAiC,CAAC,CAAC;AAE5E;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;QAC3D,MAAM,CAAC,OAAO,CAAE,KAAa,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { createClientLogger } from \"@azure/logger\";\n\n/**\n * The `@azure/logger` configuration for this package.\n * This will output logs using the `azure:eventhubs-checkpointstore-table` namespace prefix.\n */\nexport const logger = createClientLogger(\"eventhubs-checkpointstore-table\");\n\n/**\n * Logs the error's stack trace to \"verbose\" if a stack trace is available.\n * @param error - Error containing a stack trace.\n * @internal\n */\nexport function logErrorStackTrace(error: unknown): void {\n  if (error && typeof error === \"object\" && \"stack\" in error) {\n    logger.verbose((error as any).stack);\n  }\n}\n"]}

package/dist-esm/src/tableCheckpointStore.js

@@ -1,5 +1,6 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
+import { __asyncValues } from "tslib";
 import { odata } from "@azure/data-tables";
 import { logErrorStackTrace, logger } from "./log";
 /**
@@ -31,33 +32,46 @@ export class TableCheckpointStore {
      * @returns Partition ownership details of all the partitions that have had an owner.
      */
     async listOwnership(fullyQualifiedNamespace, eventHubName, consumerGroup) {
+        var _a, e_1, _b, _c;
         const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;
         const partitionOwnershipArray = [];
         const entitiesIter = this._tableClient.listEntities({
             queryOptions: { filter: odata `PartitionKey eq ${partitionKey}` },
         });
         try {
-            for await (const entity of entitiesIter) {
-                if (!entity.timestamp) {
-                    throw new Error(`Unable to retrieve timestamp from partitionKey "${partitionKey}", rowKey "${entity.rowKey}"`);
+            try {
+                for (var _d = true, entitiesIter_1 = __asyncValues(entitiesIter), entitiesIter_1_1; entitiesIter_1_1 = await entitiesIter_1.next(), _a = entitiesIter_1_1.done, !_a; _d = true) {
+                    _c = entitiesIter_1_1.value;
+                    _d = false;
+                    const entity = _c;
+                    if (!entity.timestamp) {
+                        throw new Error(`Unable to retrieve timestamp from partitionKey "${partitionKey}", rowKey "${entity.rowKey}"`);
+                    }
+                    const partitionOwnership = {
+                        fullyQualifiedNamespace,
+                        eventHubName,
+                        consumerGroup,
+                        ownerId: entity.ownerid,
+                        partitionId: entity.rowKey,
+                        lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),
+                        etag: entity.etag,
+                    };
+                    partitionOwnershipArray.push(partitionOwnership);
                 }
-                const partitionOwnership = {
-                    fullyQualifiedNamespace,
-                    eventHubName,
-                    consumerGroup,
-                    ownerId: entity.ownerid,
-                    partitionId: entity.rowKey,
-                    lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),
-                    etag: entity.etag,
-                };
-                partitionOwnershipArray.push(partitionOwnership);
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (!_d && !_a && (_b = entitiesIter_1.return)) await _b.call(entitiesIter_1);
+                }
+                finally { if (e_1) throw e_1.error; }
             }
             return partitionOwnershipArray;
         }
         catch (err) {
             logger.warning(`Error occurred while fetching the list of entities`, err.message);
             logErrorStackTrace(err);
-            if (err?.name === "AbortError")
+            if ((err === null || err === void 0 ? void 0 : err.name) === "AbortError")
                 throw err;
             throw new Error(`Error occurred while fetching the list of entities. \n${err}`);
         }
@@ -75,7 +89,7 @@ export class TableCheckpointStore {
     async claimOwnership(partitionOwnership) {
         const partitionOwnershipArray = [];
         for (const ownership of partitionOwnership) {
-            const updatedOwnership = { ...ownership };
+            const updatedOwnership = Object.assign({}, ownership);
             const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;
             const ownershipEntity = {
                 partitionKey: partitionKey,
@@ -140,20 +154,33 @@ export class TableCheckpointStore {
      *  - `tracingOptions`: Options for configuring tracing.
      */
     async listCheckpoints(fullyQualifiedNamespace, eventHubName, consumerGroup) {
+        var _a, e_2, _b, _c;
         const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;
         const checkpoints = [];
         const entitiesIter = this._tableClient.listEntities({
             queryOptions: { filter: odata `PartitionKey eq ${partitionKey}` },
         });
-        for await (const entity of entitiesIter) {
-            checkpoints.push({
-                consumerGroup,
-                eventHubName,
-                fullyQualifiedNamespace,
-                partitionId: entity.rowKey,
-                offset: parseInt(entity.offset, 10),
-                sequenceNumber: parseInt(entity.sequencenumber, 10),
-            });
+        try {
+            for (var _d = true, entitiesIter_2 = __asyncValues(entitiesIter), entitiesIter_2_1; entitiesIter_2_1 = await entitiesIter_2.next(), _a = entitiesIter_2_1.done, !_a; _d = true) {
+                _c = entitiesIter_2_1.value;
+                _d = false;
+                const entity = _c;
+                checkpoints.push({
+                    consumerGroup,
+                    eventHubName,
+                    fullyQualifiedNamespace,
+                    partitionId: entity.rowKey,
+                    offset: parseInt(entity.offset, 10),
+                    sequenceNumber: parseInt(entity.sequencenumber, 10),
+                });
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (!_d && !_a && (_b = entitiesIter_2.return)) await _b.call(entitiesIter_2);
+            }
+            finally { if (e_2) throw e_2.error; }
         }
         return checkpoints;
     }

package/dist-esm/src/tableCheckpointStore.js.map

@@ -1 +1 @@
-{"version":3,"file":"tableCheckpointStore.js","sourceRoot":"","sources":["../../src/tableCheckpointStore.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAyC,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAEnD;;;GAGG;AACH,SAAS,aAAa,CACpB,KAAQ;IAER,OAAO,OAAQ,KAAa,CAAC,SAAS,KAAK,QAAQ,CAAC;AACtD,CAAC;AAwCD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAG/B,YAAY,WAAwB;QAClC,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;IAClC,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,KAAK,CAAC,aAAa,CACjB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB;QAErB,MAAM,YAAY,GAAG,GAAG,uBAAuB,IAAI,YAAY,IAAI,aAAa,YAAY,CAAC;QAC7F,MAAM,uBAAuB,GAAyB,EAAE,CAAC;QACzD,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAA2B;YAC5E,YAAY,EAAE,EAAE,MAAM,EAAE,KAAK,CAAA,mBAAmB,YAAY,EAAE,EAAE;SACjE,CAAC,CAAC;QACH,IAAI;YACF,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,YAAY,EAAE;gBACvC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;oBACrB,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,MAAM,CAAC,MAAM,GAAG,CAC9F,CAAC;iBACH;gBAED,MAAM,kBAAkB,GAAuB;oBAC7C,uBAAuB;oBACvB,YAAY;oBACZ,aAAa;oBACb,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,oBAAoB,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;oBAC1D,IAAI,EAAE,MAAM,CAAC,IAAI;iBAClB,CAAC;gBACF,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;aAClD;YACD,OAAO,uBAAuB,CAAC;SAChC;QAAC,OAAO,GAAQ,EAAE;YACjB,MAAM,CAAC,OAAO,CAAC,oDAAoD,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAClF,kBAAkB,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,GAAG,EAAE,IAAI,KAAK,YAAY;gBAAE,MAAM,GAAG,CAAC;YAE1C,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,EAAE,CAAC,CAAC;SACjF;IACH,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,cAAc,CAAC,kBAAwC;QAC3D,MAAM,uBAAuB,GAAyB,EAAE,CAAC;QAEzD,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE;YAC1C,MAAM,gBAAgB,GAAG,EAAE,GAAG,SAAS,EAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,GAAG,SAAS,CAAC,uBAAuB,IAAI,SAAS,CAAC,YAAY,IAAI,SAAS,CAAC,aAAa,YAAY,CAAC;YAC3H,MAAM,eAAe,GAA6B;gBAChD,YAAY,EAAE,YAAY;gBAC1B,MAAM,EAAE,SAAS,CAAC,WAAW;gBAC7B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC;YAEF,oDAAoD;YACpD,2CAA2C;YAC3C,IAAI;gBACF,IAAI,SAAS,CAAC,IAAI,EAAE;oBAClB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,EAAE;wBACvF,IAAI,EAAE,SAAS,CAAC,IAAI;qBACrB,CAAC,CAAC;oBACH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CACvD,eAAe,CAAC,YAAY,EAC5B,eAAe,CAAC,MAAM,CACvB,CAAC;oBACF,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE;wBAC9B,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,eAAe,CAAC,MAAM,GAAG,CACvG,CAAC;qBACH;oBACD,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;oBACtF,gBAAgB,CAAC,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;oBAC7C,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/C,MAAM,CAAC,IAAI,CACT,IAAI,SAAS,CAAC,OAAO,mDAAmD,SAAS,CAAC,WAAW,EAAE,EAC/F,qBAAqB,SAAS,CAAC,oBAAoB,WAAW,SAAS,CAAC,IAAI,EAAE,CAC/E,CAAC;iBACH;qBAAM;oBACL,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE;wBACjF,cAAc,EAAE;4BACd,aAAa,EAAE;gCACb,MAAM,EAAE,gBAAgB;6BACzB;yBACF;qBACF,CAAC,CAAC;oBAEH,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE;wBACxC,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,eAAe,CAAC,MAAM,GAAG,CACvG,CAAC;qBACH;oBAED,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAC9C,oBAAoB,CAAC,SAAS,CAC/B,CAAC,OAAO,EAAE,CAAC;oBACZ,gBAAgB,CAAC,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC;oBAClD,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;iBAChD;aACF;YAAC,OAAO,GAAQ,EAAE;gBACjB,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE;oBAC1B,wEAAwE;oBACxE,6EAA6E;oBAC7E,6BAA6B;oBAC7B,MAAM,CAAC,OAAO,CACZ,IAAI,SAAS,CAAC,OAAO,6BAA6B,SAAS,CAAC,WAAW,6CAA6C,CACrH,CAAC;oBACF,SAAS;iBACV;gBACD,MAAM,CAAC,OAAO,CACZ,0DAA0D,SAAS,CAAC,WAAW,EAAE,EACjF,GAAG,CAAC,OAAO,CACZ,CAAC;gBACF,kBAAkB,CAAC,GAAG,CAAC,CAAC;aACzB;SACF;QACD,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,eAAe,CACnB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB;QAErB,MAAM,YAAY,GAAG,GAAG,uBAAuB,IAAI,YAAY,IAAI,aAAa,aAAa,CAAC;QAC9F,MAAM,WAAW,GAAiB,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAmB;YACpE,YAAY,EAAE,EAAE,MAAM,EAAE,KAAK,CAAA,mBAAmB,YAAY,EAAE,EAAE;SACjE,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,YAAY,EAAE;YACvC,WAAW,CAAC,IAAI,CAAC;gBACf,aAAa;gBACb,YAAY;gBACZ,uBAAuB;gBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;gBAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC;aACpD,CAAC,CAAC;SACJ;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,gBAAgB,CAAC,UAAsB;QAC3C,MAAM,YAAY,GAAG,GAAG,UAAU,CAAC,uBAAuB,IAAI,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,aAAa,aAAa,CAAC;QAC/H,MAAM,gBAAgB,GAAqB;YACzC,YAAY,EAAE,YAAY;YAC1B,MAAM,EAAE,UAAU,CAAC,WAAW;YAC9B,cAAc,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ,EAAE;YACpD,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,EAAE;SACrC,CAAC;QACF,IAAI;YACF,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,kDAAkD,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;YAC3F,OAAO;SACR;QAAC,OAAO,GAAQ,EAAE;YACjB,MAAM,CAAC,OAAO,CACZ,8DAA8D,UAAU,CAAC,WAAW,GAAG,EACvF,GAAG,CAAC,OAAO,CACZ,CAAC;YACF,MAAM,GAAG,CAAC;SACX;IACH,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { Checkpoint, CheckpointStore, PartitionOwnership } from \"@azure/event-hubs\";\nimport { TableClient, TableInsertEntityHeaders, odata } from \"@azure/data-tables\";\nimport { logErrorStackTrace, logger } from \"./log\";\n\n/**\n *\n * Checks if the value contains a `Timestamp` field of type `string`.\n */\nfunction _hasTimestamp<T extends TableInsertEntityHeaders>(\n  value: T,\n): value is T & { Timestamp: string } {\n  return typeof (value as any).Timestamp === \"string\";\n}\n\n/**\n * A checkpoint entity of type CheckpointEntity to be stored in the table\n * @internal\n *\n */\nexport interface CheckpointEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  sequencenumber: string;\n  offset: string;\n}\n\n/**\n * An ownership entity of type PartitionOwnership to be stored in the table\n * @internal\n */\nexport interface PartitionOwnershipEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  ownerid: string;\n}\n\n/**\n * An implementation of CheckpointStore that uses Azure Table Storage to persist checkpoint data.\n */\nexport class TableCheckpointStore implements CheckpointStore {\n  private _tableClient: TableClient;\n\n  constructor(tableClient: TableClient) {\n    this._tableClient = tableClient;\n  }\n\n  /**\n   * Get the list of all existing partition ownership from the underlying data store. May return empty\n   * results if there are is no existing ownership information.\n   * Partition Ownership contains the information on which `EventHubConsumerClient` subscribe call is currently processing the partition.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns Partition ownership details of all the partitions that have had an owner.\n   */\n  async listOwnership(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<PartitionOwnership[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n    const entitiesIter = this._tableClient.listEntities<PartitionOwnershipEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    try {\n      for await (const entity of entitiesIter) {\n        if (!entity.timestamp) {\n          throw new Error(\n            `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entity.rowKey}\"`,\n          );\n        }\n\n        const partitionOwnership: PartitionOwnership = {\n          fullyQualifiedNamespace,\n          eventHubName,\n          consumerGroup,\n          ownerId: entity.ownerid,\n          partitionId: entity.rowKey,\n          lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),\n          etag: entity.etag,\n        };\n        partitionOwnershipArray.push(partitionOwnership);\n      }\n      return partitionOwnershipArray;\n    } catch (err: any) {\n      logger.warning(`Error occurred while fetching the list of entities`, err.message);\n      logErrorStackTrace(err);\n      if (err?.name === \"AbortError\") throw err;\n\n      throw new Error(`Error occurred while fetching the list of entities. \\n${err}`);\n    }\n  }\n\n  /**\n   * Claim ownership of a list of partitions. This will return the list of partitions that were\n   * successfully claimed.\n   *\n   * @param partitionOwnership - The list of partition ownership this instance is claiming to own.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A list partitions this instance successfully claimed ownership.\n   */\n  async claimOwnership(partitionOwnership: PartitionOwnership[]): Promise<PartitionOwnership[]> {\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n\n    for (const ownership of partitionOwnership) {\n      const updatedOwnership = { ...ownership };\n      const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;\n      const ownershipEntity: PartitionOwnershipEntity = {\n        partitionKey: partitionKey,\n        rowKey: ownership.partitionId,\n        ownerid: ownership.ownerId,\n      };\n\n      // When we have an etag, we know the entity existed.\n      // If we encounter an error we should fail.\n      try {\n        if (ownership.etag) {\n          const updatedMetadata = await this._tableClient.updateEntity(ownershipEntity, \"Replace\", {\n            etag: ownership.etag,\n          });\n          const entityRetrieved = await this._tableClient.getEntity(\n            ownershipEntity.partitionKey,\n            ownershipEntity.rowKey,\n          );\n          if (!entityRetrieved.timestamp) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entityRetrieved.rowKey}\"`,\n            );\n          }\n          updatedOwnership.lastModifiedTimeInMs = new Date(entityRetrieved.timestamp).getTime();\n          updatedOwnership.etag = updatedMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n          logger.info(\n            `[${ownership.ownerId}] Claimed ownership successfully for partition: ${ownership.partitionId}`,\n            `LastModifiedTime: ${ownership.lastModifiedTimeInMs}, ETag: ${ownership.etag}`,\n          );\n        } else {\n          const newOwnershipMetadata = await this._tableClient.createEntity(ownershipEntity, {\n            requestOptions: {\n              customHeaders: {\n                Prefer: \"return-content\",\n              },\n            },\n          });\n\n          if (!_hasTimestamp(newOwnershipMetadata)) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${ownershipEntity.rowKey}\"`,\n            );\n          }\n\n          updatedOwnership.lastModifiedTimeInMs = new Date(\n            newOwnershipMetadata.Timestamp,\n          ).getTime();\n          updatedOwnership.etag = newOwnershipMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n        }\n      } catch (err: any) {\n        if (err.statusCode === 412) {\n          // etag failures (precondition not met) aren't fatal errors. They happen\n          // as multiple consumers attempt to claim the same partition (first one wins)\n          // and losers get this error.\n          logger.verbose(\n            `[${ownership.ownerId}] Did not claim partition ${ownership.partitionId}. Another processor has already claimed it.`,\n          );\n          continue;\n        }\n        logger.warning(\n          `Error occurred while claiming ownership for partition: ${ownership.partitionId}`,\n          err.message,\n        );\n        logErrorStackTrace(err);\n      }\n    }\n    return partitionOwnershipArray;\n  }\n\n  /**\n   * Lists all the checkpoints in a data store for a given namespace, eventhub and consumer group.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   */\n  async listCheckpoints(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<Checkpoint[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;\n    const checkpoints: Checkpoint[] = [];\n    const entitiesIter = this._tableClient.listEntities<CheckpointEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    for await (const entity of entitiesIter) {\n      checkpoints.push({\n        consumerGroup,\n        eventHubName,\n        fullyQualifiedNamespace,\n        partitionId: entity.rowKey,\n        offset: parseInt(entity.offset, 10),\n        sequenceNumber: parseInt(entity.sequencenumber, 10),\n      });\n    }\n    return checkpoints;\n  }\n\n  /**\n   * Updates the checkpoint in the data store for a partition.\n   *\n   * @param checkpoint - The checkpoint.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A promise that resolves when the checkpoint has been updated.\n   */\n  async updateCheckpoint(checkpoint: Checkpoint): Promise<void> {\n    const partitionKey = `${checkpoint.fullyQualifiedNamespace} ${checkpoint.eventHubName} ${checkpoint.consumerGroup} Checkpoint`;\n    const checkpointEntity: CheckpointEntity = {\n      partitionKey: partitionKey,\n      rowKey: checkpoint.partitionId,\n      sequencenumber: checkpoint.sequenceNumber.toString(),\n      offset: checkpoint.offset.toString(),\n    };\n    try {\n      await this._tableClient.upsertEntity(checkpointEntity);\n      logger.verbose(`Updated checkpoint successfully for partition: ${checkpoint.partitionId}`);\n      return;\n    } catch (err: any) {\n      logger.verbose(\n        `Error occurred while upating the checkpoint for partition: ${checkpoint.partitionId}.`,\n        err.message,\n      );\n      throw err;\n    }\n  }\n}\n"]}
+{"version":3,"file":"tableCheckpointStore.js","sourceRoot":"","sources":["../../src/tableCheckpointStore.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAGlC,OAAO,EAAyC,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAEnD;;;GAGG;AACH,SAAS,aAAa,CACpB,KAAQ;IAER,OAAO,OAAQ,KAAa,CAAC,SAAS,KAAK,QAAQ,CAAC;AACtD,CAAC;AAwCD;;GAEG;AACH,MAAM,OAAO,oBAAoB;IAG/B,YAAY,WAAwB;QAClC,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;IAClC,CAAC;IAED;;;;;;;;;;;;;OAaG;IACH,KAAK,CAAC,aAAa,CACjB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB;;QAErB,MAAM,YAAY,GAAG,GAAG,uBAAuB,IAAI,YAAY,IAAI,aAAa,YAAY,CAAC;QAC7F,MAAM,uBAAuB,GAAyB,EAAE,CAAC;QACzD,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAA2B;YAC5E,YAAY,EAAE,EAAE,MAAM,EAAE,KAAK,CAAA,mBAAmB,YAAY,EAAE,EAAE;SACjE,CAAC,CAAC;QACH,IAAI,CAAC;;gBACH,KAA2B,eAAA,iBAAA,cAAA,YAAY,CAAA,kBAAA,8FAAE,CAAC;oBAAf,4BAAY;oBAAZ,WAAY;oBAA5B,MAAM,MAAM,KAAA,CAAA;oBACrB,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;wBACtB,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,MAAM,CAAC,MAAM,GAAG,CAC9F,CAAC;oBACJ,CAAC;oBAED,MAAM,kBAAkB,GAAuB;wBAC7C,uBAAuB;wBACvB,YAAY;wBACZ,aAAa;wBACb,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;wBAC1B,oBAAoB,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;wBAC1D,IAAI,EAAE,MAAM,CAAC,IAAI;qBAClB,CAAC;oBACF,uBAAuB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;gBACnD,CAAC;;;;;;;;;YACD,OAAO,uBAAuB,CAAC;QACjC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,OAAO,CAAC,oDAAoD,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAClF,kBAAkB,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAA,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,IAAI,MAAK,YAAY;gBAAE,MAAM,GAAG,CAAC;YAE1C,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,EAAE,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,cAAc,CAAC,kBAAwC;QAC3D,MAAM,uBAAuB,GAAyB,EAAE,CAAC;QAEzD,KAAK,MAAM,SAAS,IAAI,kBAAkB,EAAE,CAAC;YAC3C,MAAM,gBAAgB,qBAAQ,SAAS,CAAE,CAAC;YAC1C,MAAM,YAAY,GAAG,GAAG,SAAS,CAAC,uBAAuB,IAAI,SAAS,CAAC,YAAY,IAAI,SAAS,CAAC,aAAa,YAAY,CAAC;YAC3H,MAAM,eAAe,GAA6B;gBAChD,YAAY,EAAE,YAAY;gBAC1B,MAAM,EAAE,SAAS,CAAC,WAAW;gBAC7B,OAAO,EAAE,SAAS,CAAC,OAAO;aAC3B,CAAC;YAEF,oDAAoD;YACpD,2CAA2C;YAC3C,IAAI,CAAC;gBACH,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC;oBACnB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE,SAAS,EAAE;wBACvF,IAAI,EAAE,SAAS,CAAC,IAAI;qBACrB,CAAC,CAAC;oBACH,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CACvD,eAAe,CAAC,YAAY,EAC5B,eAAe,CAAC,MAAM,CACvB,CAAC;oBACF,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC;wBAC/B,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,eAAe,CAAC,MAAM,GAAG,CACvG,CAAC;oBACJ,CAAC;oBACD,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;oBACtF,gBAAgB,CAAC,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC;oBAC7C,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/C,MAAM,CAAC,IAAI,CACT,IAAI,SAAS,CAAC,OAAO,mDAAmD,SAAS,CAAC,WAAW,EAAE,EAC/F,qBAAqB,SAAS,CAAC,oBAAoB,WAAW,SAAS,CAAC,IAAI,EAAE,CAC/E,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,eAAe,EAAE;wBACjF,cAAc,EAAE;4BACd,aAAa,EAAE;gCACb,MAAM,EAAE,gBAAgB;6BACzB;yBACF;qBACF,CAAC,CAAC;oBAEH,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE,CAAC;wBACzC,MAAM,IAAI,KAAK,CACb,mDAAmD,YAAY,cAAc,eAAe,CAAC,MAAM,GAAG,CACvG,CAAC;oBACJ,CAAC;oBAED,gBAAgB,CAAC,oBAAoB,GAAG,IAAI,IAAI,CAC9C,oBAAoB,CAAC,SAAS,CAC/B,CAAC,OAAO,EAAE,CAAC;oBACZ,gBAAgB,CAAC,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC;oBAClD,uBAAuB,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBACjD,CAAC;YACH,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;oBAC3B,wEAAwE;oBACxE,6EAA6E;oBAC7E,6BAA6B;oBAC7B,MAAM,CAAC,OAAO,CACZ,IAAI,SAAS,CAAC,OAAO,6BAA6B,SAAS,CAAC,WAAW,6CAA6C,CACrH,CAAC;oBACF,SAAS;gBACX,CAAC;gBACD,MAAM,CAAC,OAAO,CACZ,0DAA0D,SAAS,CAAC,WAAW,EAAE,EACjF,GAAG,CAAC,OAAO,CACZ,CAAC;gBACF,kBAAkB,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,OAAO,uBAAuB,CAAC;IACjC,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,eAAe,CACnB,uBAA+B,EAC/B,YAAoB,EACpB,aAAqB;;QAErB,MAAM,YAAY,GAAG,GAAG,uBAAuB,IAAI,YAAY,IAAI,aAAa,aAAa,CAAC;QAC9F,MAAM,WAAW,GAAiB,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,CAAmB;YACpE,YAAY,EAAE,EAAE,MAAM,EAAE,KAAK,CAAA,mBAAmB,YAAY,EAAE,EAAE;SACjE,CAAC,CAAC;;YACH,KAA2B,eAAA,iBAAA,cAAA,YAAY,CAAA,kBAAA,8FAAE,CAAC;gBAAf,4BAAY;gBAAZ,WAAY;gBAA5B,MAAM,MAAM,KAAA,CAAA;gBACrB,WAAW,CAAC,IAAI,CAAC;oBACf,aAAa;oBACb,YAAY;oBACZ,uBAAuB;oBACvB,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;oBACnC,cAAc,EAAE,QAAQ,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC;iBACpD,CAAC,CAAC;YACL,CAAC;;;;;;;;;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,gBAAgB,CAAC,UAAsB;QAC3C,MAAM,YAAY,GAAG,GAAG,UAAU,CAAC,uBAAuB,IAAI,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,aAAa,aAAa,CAAC;QAC/H,MAAM,gBAAgB,GAAqB;YACzC,YAAY,EAAE,YAAY;YAC1B,MAAM,EAAE,UAAU,CAAC,WAAW;YAC9B,cAAc,EAAE,UAAU,CAAC,cAAc,CAAC,QAAQ,EAAE;YACpD,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,QAAQ,EAAE;SACrC,CAAC;QACF,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,OAAO,CAAC,kDAAkD,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;YAC3F,OAAO;QACT,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,OAAO,CACZ,8DAA8D,UAAU,CAAC,WAAW,GAAG,EACvF,GAAG,CAAC,OAAO,CACZ,CAAC;YACF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { Checkpoint, CheckpointStore, PartitionOwnership } from \"@azure/event-hubs\";\nimport { TableClient, TableInsertEntityHeaders, odata } from \"@azure/data-tables\";\nimport { logErrorStackTrace, logger } from \"./log\";\n\n/**\n *\n * Checks if the value contains a `Timestamp` field of type `string`.\n */\nfunction _hasTimestamp<T extends TableInsertEntityHeaders>(\n  value: T,\n): value is T & { Timestamp: string } {\n  return typeof (value as any).Timestamp === \"string\";\n}\n\n/**\n * A checkpoint entity of type CheckpointEntity to be stored in the table\n * @internal\n *\n */\nexport interface CheckpointEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  sequencenumber: string;\n  offset: string;\n}\n\n/**\n * An ownership entity of type PartitionOwnership to be stored in the table\n * @internal\n */\nexport interface PartitionOwnershipEntity {\n  /**\n   * The partitionKey is a composite key assembled in the following format:\n   * `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`\n   */\n  partitionKey: string;\n  /**\n   * The rowKey is the partitionId\n   *\n   */\n  rowKey: string;\n  ownerid: string;\n}\n\n/**\n * An implementation of CheckpointStore that uses Azure Table Storage to persist checkpoint data.\n */\nexport class TableCheckpointStore implements CheckpointStore {\n  private _tableClient: TableClient;\n\n  constructor(tableClient: TableClient) {\n    this._tableClient = tableClient;\n  }\n\n  /**\n   * Get the list of all existing partition ownership from the underlying data store. May return empty\n   * results if there are is no existing ownership information.\n   * Partition Ownership contains the information on which `EventHubConsumerClient` subscribe call is currently processing the partition.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns Partition ownership details of all the partitions that have had an owner.\n   */\n  async listOwnership(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<PartitionOwnership[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Ownership`;\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n    const entitiesIter = this._tableClient.listEntities<PartitionOwnershipEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    try {\n      for await (const entity of entitiesIter) {\n        if (!entity.timestamp) {\n          throw new Error(\n            `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entity.rowKey}\"`,\n          );\n        }\n\n        const partitionOwnership: PartitionOwnership = {\n          fullyQualifiedNamespace,\n          eventHubName,\n          consumerGroup,\n          ownerId: entity.ownerid,\n          partitionId: entity.rowKey,\n          lastModifiedTimeInMs: new Date(entity.timestamp).getTime(),\n          etag: entity.etag,\n        };\n        partitionOwnershipArray.push(partitionOwnership);\n      }\n      return partitionOwnershipArray;\n    } catch (err: any) {\n      logger.warning(`Error occurred while fetching the list of entities`, err.message);\n      logErrorStackTrace(err);\n      if (err?.name === \"AbortError\") throw err;\n\n      throw new Error(`Error occurred while fetching the list of entities. \\n${err}`);\n    }\n  }\n\n  /**\n   * Claim ownership of a list of partitions. This will return the list of partitions that were\n   * successfully claimed.\n   *\n   * @param partitionOwnership - The list of partition ownership this instance is claiming to own.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A list partitions this instance successfully claimed ownership.\n   */\n  async claimOwnership(partitionOwnership: PartitionOwnership[]): Promise<PartitionOwnership[]> {\n    const partitionOwnershipArray: PartitionOwnership[] = [];\n\n    for (const ownership of partitionOwnership) {\n      const updatedOwnership = { ...ownership };\n      const partitionKey = `${ownership.fullyQualifiedNamespace} ${ownership.eventHubName} ${ownership.consumerGroup} Ownership`;\n      const ownershipEntity: PartitionOwnershipEntity = {\n        partitionKey: partitionKey,\n        rowKey: ownership.partitionId,\n        ownerid: ownership.ownerId,\n      };\n\n      // When we have an etag, we know the entity existed.\n      // If we encounter an error we should fail.\n      try {\n        if (ownership.etag) {\n          const updatedMetadata = await this._tableClient.updateEntity(ownershipEntity, \"Replace\", {\n            etag: ownership.etag,\n          });\n          const entityRetrieved = await this._tableClient.getEntity(\n            ownershipEntity.partitionKey,\n            ownershipEntity.rowKey,\n          );\n          if (!entityRetrieved.timestamp) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${entityRetrieved.rowKey}\"`,\n            );\n          }\n          updatedOwnership.lastModifiedTimeInMs = new Date(entityRetrieved.timestamp).getTime();\n          updatedOwnership.etag = updatedMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n          logger.info(\n            `[${ownership.ownerId}] Claimed ownership successfully for partition: ${ownership.partitionId}`,\n            `LastModifiedTime: ${ownership.lastModifiedTimeInMs}, ETag: ${ownership.etag}`,\n          );\n        } else {\n          const newOwnershipMetadata = await this._tableClient.createEntity(ownershipEntity, {\n            requestOptions: {\n              customHeaders: {\n                Prefer: \"return-content\",\n              },\n            },\n          });\n\n          if (!_hasTimestamp(newOwnershipMetadata)) {\n            throw new Error(\n              `Unable to retrieve timestamp from partitionKey \"${partitionKey}\", rowKey \"${ownershipEntity.rowKey}\"`,\n            );\n          }\n\n          updatedOwnership.lastModifiedTimeInMs = new Date(\n            newOwnershipMetadata.Timestamp,\n          ).getTime();\n          updatedOwnership.etag = newOwnershipMetadata.etag;\n          partitionOwnershipArray.push(updatedOwnership);\n        }\n      } catch (err: any) {\n        if (err.statusCode === 412) {\n          // etag failures (precondition not met) aren't fatal errors. They happen\n          // as multiple consumers attempt to claim the same partition (first one wins)\n          // and losers get this error.\n          logger.verbose(\n            `[${ownership.ownerId}] Did not claim partition ${ownership.partitionId}. Another processor has already claimed it.`,\n          );\n          continue;\n        }\n        logger.warning(\n          `Error occurred while claiming ownership for partition: ${ownership.partitionId}`,\n          err.message,\n        );\n        logErrorStackTrace(err);\n      }\n    }\n    return partitionOwnershipArray;\n  }\n\n  /**\n   * Lists all the checkpoints in a data store for a given namespace, eventhub and consumer group.\n   *\n   * @param fullyQualifiedNamespace - The fully qualified Event Hubs namespace. This is likely to be similar to\n   * <yournamespace>.servicebus.windows.net.\n   * @param eventHubName - The event hub name.\n   * @param consumerGroup - The consumer group name.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   */\n  async listCheckpoints(\n    fullyQualifiedNamespace: string,\n    eventHubName: string,\n    consumerGroup: string,\n  ): Promise<Checkpoint[]> {\n    const partitionKey = `${fullyQualifiedNamespace} ${eventHubName} ${consumerGroup} Checkpoint`;\n    const checkpoints: Checkpoint[] = [];\n    const entitiesIter = this._tableClient.listEntities<CheckpointEntity>({\n      queryOptions: { filter: odata`PartitionKey eq ${partitionKey}` },\n    });\n    for await (const entity of entitiesIter) {\n      checkpoints.push({\n        consumerGroup,\n        eventHubName,\n        fullyQualifiedNamespace,\n        partitionId: entity.rowKey,\n        offset: parseInt(entity.offset, 10),\n        sequenceNumber: parseInt(entity.sequencenumber, 10),\n      });\n    }\n    return checkpoints;\n  }\n\n  /**\n   * Updates the checkpoint in the data store for a partition.\n   *\n   * @param checkpoint - The checkpoint.\n   * @param options - A set of options that can be specified to influence the behavior of this method.\n   *  - `abortSignal`: A signal used to request operation cancellation.\n   *  - `tracingOptions`: Options for configuring tracing.\n   * @returns A promise that resolves when the checkpoint has been updated.\n   */\n  async updateCheckpoint(checkpoint: Checkpoint): Promise<void> {\n    const partitionKey = `${checkpoint.fullyQualifiedNamespace} ${checkpoint.eventHubName} ${checkpoint.consumerGroup} Checkpoint`;\n    const checkpointEntity: CheckpointEntity = {\n      partitionKey: partitionKey,\n      rowKey: checkpoint.partitionId,\n      sequencenumber: checkpoint.sequenceNumber.toString(),\n      offset: checkpoint.offset.toString(),\n    };\n    try {\n      await this._tableClient.upsertEntity(checkpointEntity);\n      logger.verbose(`Updated checkpoint successfully for partition: ${checkpoint.partitionId}`);\n      return;\n    } catch (err: any) {\n      logger.verbose(\n        `Error occurred while upating the checkpoint for partition: ${checkpoint.partitionId}.`,\n        err.message,\n      );\n      throw err;\n    }\n  }\n}\n"]}

package/dist-esm/src/util/error.js.map

@@ -1 +1 @@
-{"version":3,"file":"error.js","sourceRoot":"","sources":["../../../src/util/error.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEpD;;;;;;GAMG;AACH,MAAM,UAAU,gCAAgC,CAC9C,UAAkB,EAClB,aAAqB,EACrB,cAAuB;IAEvB,IAAI,cAAc,KAAK,SAAS,IAAI,cAAc,KAAK,IAAI,EAAE;QAC3D,MAAM,KAAK,GAAG,IAAI,SAAS,CACzB,GAAG,UAAU,sCAAsC,aAAa,GAAG,CACpE,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9B,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,KAAK,CAAC;KACb;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { logErrorStackTrace, logger } from \"../log\";\n\n/**\n * @internal\n * Logs and Throws TypeError if given parameter is undefined or null\n * @param methodName - Name of the method that was passed the parameter\n * @param parameterName - Name of the parameter to check\n * @param parameterValue - Value of the parameter to check\n */\nexport function throwTypeErrorIfParameterMissing(\n  methodName: string,\n  parameterName: string,\n  parameterValue: unknown,\n): void {\n  if (parameterValue === undefined || parameterValue === null) {\n    const error = new TypeError(\n      `${methodName} called without required argument \"${parameterName}\"`,\n    );\n    logger.warning(error.message);\n    logErrorStackTrace(error);\n    throw error;\n  }\n}\n"]}
+{"version":3,"file":"error.js","sourceRoot":"","sources":["../../../src/util/error.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEpD;;;;;;GAMG;AACH,MAAM,UAAU,gCAAgC,CAC9C,UAAkB,EAClB,aAAqB,EACrB,cAAuB;IAEvB,IAAI,cAAc,KAAK,SAAS,IAAI,cAAc,KAAK,IAAI,EAAE,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,SAAS,CACzB,GAAG,UAAU,sCAAsC,aAAa,GAAG,CACpE,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC9B,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\n\nimport { logErrorStackTrace, logger } from \"../log\";\n\n/**\n * @internal\n * Logs and Throws TypeError if given parameter is undefined or null\n * @param methodName - Name of the method that was passed the parameter\n * @param parameterName - Name of the parameter to check\n * @param parameterValue - Value of the parameter to check\n */\nexport function throwTypeErrorIfParameterMissing(\n  methodName: string,\n  parameterName: string,\n  parameterValue: unknown,\n): void {\n  if (parameterValue === undefined || parameterValue === null) {\n    const error = new TypeError(\n      `${methodName} called without required argument \"${parameterName}\"`,\n    );\n    logger.warning(error.message);\n    logErrorStackTrace(error);\n    throw error;\n  }\n}\n"]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@azure/eventhubs-checkpointstore-table",
   "sdk-type": "client",
-  "version": "1.0.0-alpha.20240112.1",
+  "version": "1.0.0-alpha.20240126.2",
   "description": "An Azure Storage Table solution to store checkpoints when using Event Hubs.",
   "author": "Microsoft Corporation",
   "license": "MIT",
@@ -97,7 +97,7 @@
     "c8": "^8.0.0",
     "rimraf": "^3.0.0",
     "ts-node": "^10.0.0",
-    "typescript": "~5.2.0",
+    "typescript": "~5.3.3",
     "util": "^0.12.1"
   },
   "//sampleConfiguration": {