@azure/event-hubs 5.13.0-beta.1 → 6.0.0-alpha.20250128.2

package/dist/commonjs/util/timerLoop.js

@@ -0,0 +1,30 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTimerLoop = createTimerLoop;
+/**
+ * Creates a timer loop with the given timeout and task.
+ * @internal
+ */
+function createTimerLoop(timeoutInMs, createTask) {
+    let token;
+    const loop = {
+        start: () => {
+            clearTimeout(token);
+            token = setTimeout(() => createTask()
+                .catch(() => {
+                /** eats up any unhandled error */
+            })
+                .finally(loop.start), timeoutInMs);
+            loop.isRunning = true;
+        },
+        stop: () => {
+            clearTimeout(token);
+            loop.isRunning = false;
+        },
+        isRunning: false,
+    };
+    return loop;
+}
+//# sourceMappingURL=timerLoop.js.map

package/dist/commonjs/util/timerLoop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"timerLoop.js","sourceRoot":"","sources":["../../../src/util/timerLoop.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AA2BlC,0CA0BC;AA9BD;;;GAGG;AACH,SAAgB,eAAe,CAC7B,WAAmB,EACnB,UAAkC;IAElC,IAAI,KAAoC,CAAC;IACzC,MAAM,IAAI,GAAG;QACX,KAAK,EAAE,GAAG,EAAE;YACV,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,KAAK,GAAG,UAAU,CAChB,GAAG,EAAE,CACH,UAAU,EAAE;iBACT,KAAK,CAAC,GAAG,EAAE;gBACV,kCAAkC;YACpC,CAAC,CAAC;iBACD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EACxB,WAAW,CACZ,CAAC;YACF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,IAAI,EAAE,GAAG,EAAE;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC;QACD,SAAS,EAAE,KAAK;KACjB,CAAC;IACF,OAAO,IAAI,CAAC;AACd,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * A timer loop is a loop over multiple instances of a promise every specific\n * time interval. It is different from `setInterval` in that it waits until the\n * promise is settled before it goes to the next iteration.\n * */\nexport interface TimerLoop {\n  /**\n   * Starts the timer loop. If the loop is already running, the loop will be\n   * restarted.\n   */\n  start: () => void;\n  /**\n   * Stops the timer loop. If the loop is not running, it is a no-op.\n   */\n  stop: () => void;\n  /**\n   * Whether the loop is running.\n   */\n  isRunning: boolean;\n}\n\n/**\n * Creates a timer loop with the given timeout and task.\n * @internal\n */\nexport function createTimerLoop(\n  timeoutInMs: number,\n  createTask: () => Promise<unknown>,\n): TimerLoop {\n  let token: ReturnType<typeof setTimeout>;\n  const loop = {\n    start: () => {\n      clearTimeout(token);\n      token = setTimeout(\n        () =>\n          createTask()\n            .catch(() => {\n              /** eats up any unhandled error */\n            })\n            .finally(loop.start),\n        timeoutInMs,\n      );\n      loop.isRunning = true;\n    },\n    stop: () => {\n      clearTimeout(token);\n      loop.isRunning = false;\n    },\n    isRunning: false,\n  };\n  return loop;\n}\n"]}

package/dist/commonjs/util/typeGuards.d.ts

@@ -0,0 +1,8 @@
+import type { NamedKeyCredential, SASCredential, TokenCredential } from "@azure/core-auth";
+/**
+ * Typeguard that checks if the input is a credential type the clients accept.
+ * @param thing - Any object.
+ * @internal
+ */
+export declare function isCredential(thing: unknown): thing is TokenCredential | NamedKeyCredential | SASCredential;
+//# sourceMappingURL=typeGuards.d.ts.map

package/dist/commonjs/util/typeGuards.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeGuards.d.ts","sourceRoot":"","sources":["../../../src/util/typeGuards.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAG3F;;;;GAIG;AACH,wBAAgB,YAAY,CAC1B,KAAK,EAAE,OAAO,GACb,KAAK,IAAI,eAAe,GAAG,kBAAkB,GAAG,aAAa,CAE/D"}

package/dist/commonjs/util/typeGuards.js

@@ -0,0 +1,15 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isCredential = isCredential;
+const core_auth_1 = require("@azure/core-auth");
+/**
+ * Typeguard that checks if the input is a credential type the clients accept.
+ * @param thing - Any object.
+ * @internal
+ */
+function isCredential(thing) {
+    return (0, core_auth_1.isTokenCredential)(thing) || (0, core_auth_1.isNamedKeyCredential)(thing) || (0, core_auth_1.isSASCredential)(thing);
+}
+//# sourceMappingURL=typeGuards.js.map

package/dist/commonjs/util/typeGuards.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeGuards.js","sourceRoot":"","sources":["../../../src/util/typeGuards.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAUlC,oCAIC;AAXD,gDAA4F;AAE5F;;;;GAIG;AACH,SAAgB,YAAY,CAC1B,KAAc;IAEd,OAAO,IAAA,6BAAiB,EAAC,KAAK,CAAC,IAAI,IAAA,gCAAoB,EAAC,KAAK,CAAC,IAAI,IAAA,2BAAe,EAAC,KAAK,CAAC,CAAC;AAC3F,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { NamedKeyCredential, SASCredential, TokenCredential } from \"@azure/core-auth\";\nimport { isNamedKeyCredential, isSASCredential, isTokenCredential } from \"@azure/core-auth\";\n\n/**\n * Typeguard that checks if the input is a credential type the clients accept.\n * @param thing - Any object.\n * @internal\n */\nexport function isCredential(\n  thing: unknown,\n): thing is TokenCredential | NamedKeyCredential | SASCredential {\n  return isTokenCredential(thing) || isNamedKeyCredential(thing) || isSASCredential(thing);\n}\n"]}

package/dist/commonjs/util/utils.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Returns a random name by appending a guid to the input string as follows:
+ * `{name}-{uuid}`.
+ * @internal
+ */
+export declare function getRandomName(prefix?: string): string;
+//# sourceMappingURL=utils.d.ts.map

package/dist/commonjs/util/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/util/utils.ts"],"names":[],"mappings":"AAKA;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAGrD"}

package/dist/commonjs/util/utils.js

@@ -0,0 +1,16 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRandomName = getRandomName;
+const core_util_1 = require("@azure/core-util");
+/**
+ * Returns a random name by appending a guid to the input string as follows:
+ * `{name}-{uuid}`.
+ * @internal
+ */
+function getRandomName(prefix) {
+    const str = (0, core_util_1.randomUUID)();
+    return prefix ? `${prefix}-${str}` : str;
+}
+//# sourceMappingURL=utils.js.map

package/dist/commonjs/util/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/util/utils.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AASlC,sCAGC;AAVD,gDAA8C;AAE9C;;;;GAIG;AACH,SAAgB,aAAa,CAAC,MAAe;IAC3C,MAAM,GAAG,GAAG,IAAA,sBAAU,GAAE,CAAC;IACzB,OAAO,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;AAC3C,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { randomUUID } from \"@azure/core-util\";\n\n/**\n * Returns a random name by appending a guid to the input string as follows:\n * `{name}-{uuid}`.\n * @internal\n */\nexport function getRandomName(prefix?: string): string {\n  const str = randomUUID();\n  return prefix ? `${prefix}-${str}` : str;\n}\n"]}

package/dist/commonjs/withAuth.d.ts

@@ -0,0 +1,25 @@
+import type { CbsClient } from "@azure/core-amqp";
+import type { AbortSignalLike } from "@azure/abort-controller";
+import type { ConnectionContext } from "./connectionContext.js";
+import type { TimerLoop } from "./util/timerLoop.js";
+import type { SimpleLogger } from "./logger.js";
+/**
+ *
+ * @param callback - The callback to be executed after the token is obtained.
+ * @param context - The connection context.
+ * @param audience - The audience for which the token is needed.
+ * @param timeoutInMs - The timeout in milliseconds.
+ * @param logger - The logger to be used for logging.
+ * @returns  A TimerLoop that keeps the token refreshed.
+ * @internal
+ */
+export declare function withAuth(callback: () => Promise<void>, context: ConnectionContext, audience: string, timeoutInMs: number, logger: SimpleLogger, options: {
+    abortSignal?: AbortSignalLike;
+}): Promise<TimerLoop>;
+/**
+ * @internal
+ */
+export declare function openCbsSession(client: CbsClient, timeoutAfterStartTime: number, { abortSignal }?: {
+    abortSignal?: AbortSignalLike;
+}): Promise<void>;
+//# sourceMappingURL=withAuth.d.ts.map

package/dist/commonjs/withAuth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"withAuth.d.ts","sourceRoot":"","sources":["../../src/withAuth.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAoB,SAAS,EAAe,MAAM,kBAAkB,CAAC;AAEjF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAE/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAGhD;;;;;;;;;GASG;AACH,wBAAsB,QAAQ,CAC5B,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,EAC7B,OAAO,EAAE,iBAAiB,EAC1B,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,MAAM,EACnB,MAAM,EAAE,YAAY,EACpB,OAAO,EAAE;IACP,WAAW,CAAC,EAAE,eAAe,CAAC;CAC/B,GACA,OAAO,CAAC,SAAS,CAAC,CAoBpB;AAED;;GAEG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,SAAS,EACjB,qBAAqB,EAAE,MAAM,EAC7B,EAAE,WAAW,EAAE,GAAE;IAAE,WAAW,CAAC,EAAE,eAAe,CAAA;CAAO,GACtD,OAAO,CAAC,IAAI,CAAC,CAaf"}

package/dist/commonjs/withAuth.js

@@ -0,0 +1,88 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.withAuth = withAuth;
+exports.openCbsSession = openCbsSession;
+const core_amqp_1 = require("@azure/core-amqp");
+const timerLoop_js_1 = require("./util/timerLoop.js");
+const logger_js_1 = require("./logger.js");
+/**
+ *
+ * @param callback - The callback to be executed after the token is obtained.
+ * @param context - The connection context.
+ * @param audience - The audience for which the token is needed.
+ * @param timeoutInMs - The timeout in milliseconds.
+ * @param logger - The logger to be used for logging.
+ * @returns  A TimerLoop that keeps the token refreshed.
+ * @internal
+ */
+async function withAuth(callback, context, audience, timeoutInMs, logger, options) {
+    const info = await getTokenInfo(context.tokenCredential, audience);
+    await setupClaimNegotiation(context, audience, info, timeoutInMs, logger, options);
+    await callback();
+    async function createTask() {
+        try {
+            const info2 = await getTokenInfo(context.tokenCredential, audience);
+            await setupClaimNegotiation(context, audience, info2, timeoutInMs, logger, options);
+            logger.verbose(`next token renewal is in ${info2.timeToLiveInMs} milliseconds @(${new Date(Date.now() + info2.timeToLiveInMs).toString()}).`);
+        }
+        catch (err) {
+            logger.verbose(`an error occurred while renewing the token: ${(0, logger_js_1.logObj)(err)}`);
+        }
+    }
+    const loop = (0, timerLoop_js_1.createTimerLoop)(info.timeToLiveInMs, createTask);
+    loop.start();
+    return loop;
+}
+/**
+ * @internal
+ */
+async function openCbsSession(client, timeoutAfterStartTime, { abortSignal } = {}) {
+    return core_amqp_1.defaultCancellableLock.acquire(client.cbsLock, () => client.init({
+        abortSignal,
+        timeoutInMs: timeoutAfterStartTime - Date.now(),
+    }), {
+        abortSignal,
+        timeoutInMs: timeoutAfterStartTime - Date.now(),
+    });
+}
+async function getAadToken(cred) {
+    const token = await cred.getToken(core_amqp_1.Constants.aadEventHubsScope);
+    if (!token) {
+        throw new Error(`Failed to get token from the provided "TokenCredential" object`);
+    }
+    return {
+        token,
+        type: core_amqp_1.TokenType.CbsTokenTypeJwt,
+        timeToLiveInMs: token.expiresOnTimestamp - Date.now() - 2 * 60 * 1000,
+    };
+}
+async function getSharedKeyBasedToken(cred, audience) {
+    return {
+        token: await cred.getToken(audience),
+        type: core_amqp_1.TokenType.CbsTokenTypeSas,
+        timeToLiveInMs: 45 * 60 * 1000,
+    };
+}
+async function getTokenInfo(cred, audience) {
+    return (0, core_amqp_1.isSasTokenProvider)(cred) ? getSharedKeyBasedToken(cred, audience) : getAadToken(cred);
+}
+function negotiateClaim({ token, type }, audience, cbsSession, timeoutAfterStartTime, lock, abortSignal) {
+    return core_amqp_1.defaultCancellableLock.acquire(lock, () => cbsSession.negotiateClaim(audience, token.token, type, {
+        abortSignal,
+        timeoutInMs: timeoutAfterStartTime - Date.now(),
+    }), {
+        abortSignal,
+        timeoutInMs: timeoutAfterStartTime - Date.now(),
+    });
+}
+async function setupClaimNegotiation(context, audience, info, timeoutInMs, logger, { abortSignal, }) {
+    const startTime = Date.now();
+    logger.verbose(`acquiring cbs lock: '${context.cbsSession.cbsLock}' for creating the cbs session`);
+    await openCbsSession(context.cbsSession, timeoutInMs + startTime, { abortSignal });
+    logger.verbose(`acquiring cbs lock: '${context.negotiateClaimLock}' for cbs auth`);
+    await negotiateClaim(info, audience, context.cbsSession, timeoutInMs + startTime, context.negotiateClaimLock, abortSignal);
+    logger.verbose("claim negotiation succeeded");
+}
+//# sourceMappingURL=withAuth.js.map

package/dist/commonjs/withAuth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"withAuth.js","sourceRoot":"","sources":["../../src/withAuth.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAsBlC,4BA6BC;AAKD,wCAiBC;AAtED,gDAAoG;AAKpG,sDAAsD;AAEtD,2CAAqC;AAErC;;;;;;;;;GASG;AACI,KAAK,UAAU,QAAQ,CAC5B,QAA6B,EAC7B,OAA0B,EAC1B,QAAgB,EAChB,WAAmB,EACnB,MAAoB,EACpB,OAEC;IAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;IACnE,MAAM,qBAAqB,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACnF,MAAM,QAAQ,EAAE,CAAC;IACjB,KAAK,UAAU,UAAU;QACvB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,OAAO,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC;YACpE,MAAM,qBAAqB,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YACpF,MAAM,CAAC,OAAO,CACZ,4BAA4B,KAAK,CAAC,cAAc,mBAAmB,IAAI,IAAI,CACzE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,cAAc,CAClC,CAAC,QAAQ,EAAE,IAAI,CACjB,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,OAAO,CAAC,+CAA+C,IAAA,kBAAM,EAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IACD,MAAM,IAAI,GAAG,IAAA,8BAAe,EAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,KAAK,EAAE,CAAC;IACb,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACI,KAAK,UAAU,cAAc,CAClC,MAAiB,EACjB,qBAA6B,EAC7B,EAAE,WAAW,KAAwC,EAAE;IAEvD,OAAO,kCAAsB,CAAC,OAAO,CACnC,MAAM,CAAC,OAAO,EACd,GAAG,EAAE,CACH,MAAM,CAAC,IAAI,CAAC;QACV,WAAW;QACX,WAAW,EAAE,qBAAqB,GAAG,IAAI,CAAC,GAAG,EAAE;KAChD,CAAC,EACJ;QACE,WAAW;QACX,WAAW,EAAE,qBAAqB,GAAG,IAAI,CAAC,GAAG,EAAE;KAChD,CACF,CAAC;AACJ,CAAC;AAeD,KAAK,UAAU,WAAW,CAAC,IAAqB;IAC9C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,qBAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/D,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;IACpF,CAAC;IACD,OAAO;QACL,KAAK;QACL,IAAI,EAAE,qBAAS,CAAC,eAAe;QAC/B,cAAc,EAAE,KAAK,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI;KACtE,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,IAAsB,EACtB,QAAgB;IAEhB,OAAO;QACL,KAAK,EAAE,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACpC,IAAI,EAAE,qBAAS,CAAC,eAAe;QAC/B,cAAc,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI;KAC/B,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,IAAwC,EACxC,QAAgB;IAEhB,OAAO,IAAA,8BAAkB,EAAC,IAAI,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;AAC/F,CAAC;AAED,SAAS,cAAc,CACrB,EAAE,KAAK,EAAE,IAAI,EAAa,EAC1B,QAAgB,EAChB,UAAqB,EACrB,qBAA6B,EAC7B,IAAY,EACZ,WAA6B;IAE7B,OAAO,kCAAsB,CAAC,OAAO,CACnC,IAAI,EACJ,GAAG,EAAE,CACH,UAAU,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE;QACrD,WAAW;QACX,WAAW,EAAE,qBAAqB,GAAG,IAAI,CAAC,GAAG,EAAE;KAChD,CAAC,EACJ;QACE,WAAW;QACX,WAAW,EAAE,qBAAqB,GAAG,IAAI,CAAC,GAAG,EAAE;KAChD,CACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,OAA0B,EAC1B,QAAgB,EAChB,IAAe,EACf,WAAmB,EACnB,MAAoB,EACpB,EACE,WAAW,GAGZ;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,CAAC,OAAO,CACZ,wBAAwB,OAAO,CAAC,UAAU,CAAC,OAAO,gCAAgC,CACnF,CAAC;IAEF,MAAM,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,WAAW,GAAG,SAAS,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;IACnF,MAAM,CAAC,OAAO,CAAC,wBAAwB,OAAO,CAAC,kBAAkB,gBAAgB,CAAC,CAAC;IACnF,MAAM,cAAc,CAClB,IAAI,EACJ,QAAQ,EACR,OAAO,CAAC,UAAU,EAClB,WAAW,GAAG,SAAS,EACvB,OAAO,CAAC,kBAAkB,EAC1B,WAAW,CACZ,CAAC;IACF,MAAM,CAAC,OAAO,CAAC,6BAA6B,CAAC,CAAC;AAChD,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { SasTokenProvider, CbsClient, CbsResponse } from \"@azure/core-amqp\";\nimport { Constants, TokenType, defaultCancellableLock, isSasTokenProvider } from \"@azure/core-amqp\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { AccessToken, TokenCredential } from \"@azure/core-auth\";\nimport type { ConnectionContext } from \"./connectionContext.js\";\nimport type { TimerLoop } from \"./util/timerLoop.js\";\nimport { createTimerLoop } from \"./util/timerLoop.js\";\nimport type { SimpleLogger } from \"./logger.js\";\nimport { logObj } from \"./logger.js\";\n\n/**\n *\n * @param callback - The callback to be executed after the token is obtained.\n * @param context - The connection context.\n * @param audience - The audience for which the token is needed.\n * @param timeoutInMs - The timeout in milliseconds.\n * @param logger - The logger to be used for logging.\n * @returns  A TimerLoop that keeps the token refreshed.\n * @internal\n */\nexport async function withAuth(\n  callback: () => Promise<void>,\n  context: ConnectionContext,\n  audience: string,\n  timeoutInMs: number,\n  logger: SimpleLogger,\n  options: {\n    abortSignal?: AbortSignalLike;\n  },\n): Promise<TimerLoop> {\n  const info = await getTokenInfo(context.tokenCredential, audience);\n  await setupClaimNegotiation(context, audience, info, timeoutInMs, logger, options);\n  await callback();\n  async function createTask(): Promise<void> {\n    try {\n      const info2 = await getTokenInfo(context.tokenCredential, audience);\n      await setupClaimNegotiation(context, audience, info2, timeoutInMs, logger, options);\n      logger.verbose(\n        `next token renewal is in ${info2.timeToLiveInMs} milliseconds @(${new Date(\n          Date.now() + info2.timeToLiveInMs,\n        ).toString()}).`,\n      );\n    } catch (err) {\n      logger.verbose(`an error occurred while renewing the token: ${logObj(err)}`);\n    }\n  }\n  const loop = createTimerLoop(info.timeToLiveInMs, createTask);\n  loop.start();\n  return loop;\n}\n\n/**\n * @internal\n */\nexport async function openCbsSession(\n  client: CbsClient,\n  timeoutAfterStartTime: number,\n  { abortSignal }: { abortSignal?: AbortSignalLike } = {},\n): Promise<void> {\n  return defaultCancellableLock.acquire(\n    client.cbsLock,\n    () =>\n      client.init({\n        abortSignal,\n        timeoutInMs: timeoutAfterStartTime - Date.now(),\n      }),\n    {\n      abortSignal,\n      timeoutInMs: timeoutAfterStartTime - Date.now(),\n    },\n  );\n}\n\n/**\n * A representation of an access token.\n * @internal\n */\ninterface TokenInfo {\n  /** The access token */\n  token: AccessToken;\n  /** The type of the token */\n  type: TokenType;\n  /** The time duration after which the token should be refreshed */\n  timeToLiveInMs: number;\n}\n\nasync function getAadToken(cred: TokenCredential): Promise<TokenInfo> {\n  const token = await cred.getToken(Constants.aadEventHubsScope);\n  if (!token) {\n    throw new Error(`Failed to get token from the provided \"TokenCredential\" object`);\n  }\n  return {\n    token,\n    type: TokenType.CbsTokenTypeJwt,\n    timeToLiveInMs: token.expiresOnTimestamp - Date.now() - 2 * 60 * 1000,\n  };\n}\n\nasync function getSharedKeyBasedToken(\n  cred: SasTokenProvider,\n  audience: string,\n): Promise<TokenInfo> {\n  return {\n    token: await cred.getToken(audience),\n    type: TokenType.CbsTokenTypeSas,\n    timeToLiveInMs: 45 * 60 * 1000,\n  };\n}\n\nasync function getTokenInfo(\n  cred: SasTokenProvider | TokenCredential,\n  audience: string,\n): Promise<TokenInfo> {\n  return isSasTokenProvider(cred) ? getSharedKeyBasedToken(cred, audience) : getAadToken(cred);\n}\n\nfunction negotiateClaim(\n  { token, type }: TokenInfo,\n  audience: string,\n  cbsSession: CbsClient,\n  timeoutAfterStartTime: number,\n  lock: string,\n  abortSignal?: AbortSignalLike,\n): Promise<CbsResponse> {\n  return defaultCancellableLock.acquire(\n    lock,\n    () =>\n      cbsSession.negotiateClaim(audience, token.token, type, {\n        abortSignal,\n        timeoutInMs: timeoutAfterStartTime - Date.now(),\n      }),\n    {\n      abortSignal,\n      timeoutInMs: timeoutAfterStartTime - Date.now(),\n    },\n  );\n}\n\nasync function setupClaimNegotiation(\n  context: ConnectionContext,\n  audience: string,\n  info: TokenInfo,\n  timeoutInMs: number,\n  logger: SimpleLogger,\n  {\n    abortSignal,\n  }: {\n    abortSignal?: AbortSignalLike;\n  },\n): Promise<void> {\n  const startTime = Date.now();\n  logger.verbose(\n    `acquiring cbs lock: '${context.cbsSession.cbsLock}' for creating the cbs session`,\n  );\n\n  await openCbsSession(context.cbsSession, timeoutInMs + startTime, { abortSignal });\n  logger.verbose(`acquiring cbs lock: '${context.negotiateClaimLock}' for cbs auth`);\n  await negotiateClaim(\n    info,\n    audience,\n    context.cbsSession,\n    timeoutInMs + startTime,\n    context.negotiateClaimLock,\n    abortSignal,\n  );\n  logger.verbose(\"claim negotiation succeeded\");\n}\n"]}

package/dist/esm/batchingPartitionChannel.d.ts

@@ -0,0 +1,94 @@
+import type { AmqpAnnotatedMessage } from "@azure/core-amqp";
+import type { EventData, EventHubBufferedProducerClientOptions, EventHubProducerClient, OperationOptions } from "./index.js";
+import type { AbortSignalLike } from "@azure/abort-controller";
+export interface BatchingPartitionChannelProps {
+    loopAbortSignal: AbortSignalLike;
+    maxBufferSize: number;
+    maxWaitTimeInMs: number;
+    partitionId: string;
+    producer: EventHubProducerClient;
+    /**
+     * The handler to call once a batch has successfully published.
+     */
+    onSendEventsSuccessHandler?: EventHubBufferedProducerClientOptions["onSendEventsSuccessHandler"];
+    /**
+     * The handler to call when a batch fails to publish.
+     */
+    onSendEventsErrorHandler: EventHubBufferedProducerClientOptions["onSendEventsErrorHandler"];
+}
+/**
+ * The `BatchingPartitionChannel` is responsible for accepting enqueued events
+ * and optimally batching and sending them to an Event Hub.
+ * @internal
+ */
+export declare class BatchingPartitionChannel {
+    private _eventQueue;
+    private _batchedEvents;
+    private _bufferCount;
+    private _readyQueue;
+    private _flushState;
+    private _isRunning;
+    private _lastBatchCreationTime;
+    private _loopAbortSignal;
+    private _maxBufferSize;
+    private _maxWaitTimeInMs;
+    private _onSendEventsErrorHandler;
+    private _onSendEventsSuccessHandler?;
+    private _partitionId;
+    private _producer;
+    constructor({ loopAbortSignal, maxBufferSize, maxWaitTimeInMs, onSendEventsErrorHandler, onSendEventsSuccessHandler, partitionId, producer, }: BatchingPartitionChannelProps);
+    getCurrentBufferedCount(): number;
+    enqueueEvent(event: EventData | AmqpAnnotatedMessage): Promise<void>;
+    /**
+     * Sets the flush state so that no new events can be enqueued until
+     * all the currently buffered events are sent to the Event Hub.
+     *
+     * Returns a promise that resolves once flushing is complete.
+     */
+    flush(_options?: OperationOptions): Promise<void>;
+    /**
+     * Returns a promise that resolves once there is room for events to be added
+     * to the buffer.
+     */
+    private _ready;
+    /**
+     * Starts the loop that creates batches and sends them to the Event Hub.
+     *
+     * The loop will run until the `_loopAbortSignal` is aborted.
+     */
+    private _startPublishLoop;
+    /**
+     * Helper method that returns an `EventDataBatch`.
+     * This also has the side effects of
+     *  - keeping track of batch creation time: needed for maxWaitTime calculations.
+     *  - clearing reference to batched events.
+     *  - incrementing the readiness: creating a new batch indicates the buffer
+     *    should have room, so we can resolve some pending `ready()` calls.
+     */
+    private _createBatch;
+    /**
+     * This method will resolve as many pending `ready()` calls as it can
+     * based on how much space remains in the buffer.
+     *
+     * If the channel is currently flushing, this is a no-op. This prevents
+     * `enqueueEvent` calls from adding the event to the buffer until flushing
+     * completes.
+     */
+    private _incrementReadiness;
+    /**
+     * Calls the user-provided `onSendEventsSuccessHandler` with the events
+     * that were successfully sent.
+     */
+    private _reportSuccess;
+    /**
+     * Calls the user-provided `onSendEventsErrorHandler` with an error and the events
+     * that were not successfully sent.
+     */
+    private _reportFailure;
+    /**
+     * Updates the channel's flush state once the size of the
+     * event buffer has decreased to 0.
+     */
+    private _updateFlushState;
+}
+//# sourceMappingURL=batchingPartitionChannel.d.ts.map

package/dist/esm/batchingPartitionChannel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"batchingPartitionChannel.d.ts","sourceRoot":"","sources":["../../src/batchingPartitionChannel.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAE7D,OAAO,KAAK,EACV,SAAS,EAET,qCAAqC,EACrC,sBAAsB,EACtB,gBAAgB,EACjB,MAAM,YAAY,CAAC;AAGpB,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAM/D,MAAM,WAAW,6BAA6B;IAC5C,eAAe,EAAE,eAAe,CAAC;IACjC,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,sBAAsB,CAAC;IACjC;;OAEG;IACH,0BAA0B,CAAC,EAAE,qCAAqC,CAAC,4BAA4B,CAAC,CAAC;IACjG;;OAEG;IACH,wBAAwB,EAAE,qCAAqC,CAAC,0BAA0B,CAAC,CAAC;CAC7F;AAED;;;;GAIG;AACH,qBAAa,wBAAwB;IACnC,OAAO,CAAC,WAAW,CAA0D;IAC7E,OAAO,CAAC,cAAc,CAA+C;IACrE,OAAO,CAAC,YAAY,CAAa;IACjC,OAAO,CAAC,WAAW,CAGX;IACR,OAAO,CAAC,WAAW,CAIjB;IACF,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,sBAAsB,CAAa;IAC3C,OAAO,CAAC,gBAAgB,CAAkB;IAC1C,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,yBAAyB,CAAoE;IACrG,OAAO,CAAC,2BAA2B,CAAC,CAAsE;IAE1G,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAyB;gBAE9B,EACV,eAAe,EACf,aAAa,EACb,eAAe,EACf,wBAAwB,EACxB,0BAA0B,EAC1B,WAAW,EACX,QAAQ,GACT,EAAE,6BAA6B;IAUhC,uBAAuB,IAAI,MAAM;IAI3B,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,oBAAoB,GAAG,OAAO,CAAC,IAAI,CAAC;IAmB1E;;;;;OAKG;IACG,KAAK,CAAC,QAAQ,GAAE,gBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB3D;;;OAGG;IACH,OAAO,CAAC,MAAM;IAmBd;;;;OAIG;YACW,iBAAiB;IAuF/B;;;;;;;OAOG;YACW,YAAY;IAU1B;;;;;;;OAOG;IACH,OAAO,CAAC,mBAAmB;IAW3B;;;OAGG;IACH,OAAO,CAAC,cAAc;IAmBtB;;;OAGG;IACH,OAAO,CAAC,cAAc;IAoBtB;;;OAGG;IACH,OAAO,CAAC,iBAAiB;CAW1B"}

package/dist/esm/batchingPartitionChannel.js

@@ -0,0 +1,248 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { delay } from "@azure/core-amqp";
+import { isDefined, isObjectWithProperties } from "@azure/core-util";
+import { AwaitableQueue } from "./impl/awaitableQueue.js";
+import { getPromiseParts } from "./util/getPromiseParts.js";
+import { logger } from "./logger.js";
+import { cancelablePromiseRace } from "@azure/core-util";
+/**
+ * The `BatchingPartitionChannel` is responsible for accepting enqueued events
+ * and optimally batching and sending them to an Event Hub.
+ * @internal
+ */
+export class BatchingPartitionChannel {
+    constructor({ loopAbortSignal, maxBufferSize, maxWaitTimeInMs, onSendEventsErrorHandler, onSendEventsSuccessHandler, partitionId, producer, }) {
+        this._eventQueue = new AwaitableQueue();
+        this._batchedEvents = [];
+        this._bufferCount = 0;
+        this._readyQueue = [];
+        this._flushState = {
+            isFlushing: false,
+        };
+        this._isRunning = false;
+        this._lastBatchCreationTime = 0;
+        this._loopAbortSignal = loopAbortSignal;
+        this._maxBufferSize = maxBufferSize;
+        this._maxWaitTimeInMs = maxWaitTimeInMs;
+        this._onSendEventsErrorHandler = onSendEventsErrorHandler;
+        this._onSendEventsSuccessHandler = onSendEventsSuccessHandler;
+        this._partitionId = partitionId;
+        this._producer = producer;
+    }
+    getCurrentBufferedCount() {
+        return this._bufferCount;
+    }
+    async enqueueEvent(event) {
+        await this._ready();
+        this._eventQueue.push(event);
+        this._bufferCount++;
+        if (!this._isRunning) {
+            this._isRunning = true;
+            this._startPublishLoop().catch((e) => {
+                logger.error(`The following error occured during batch creation or sending: ${JSON.stringify(e, undefined, "  ")}`);
+            });
+        }
+    }
+    /**
+     * Sets the flush state so that no new events can be enqueued until
+     * all the currently buffered events are sent to the Event Hub.
+     *
+     * Returns a promise that resolves once flushing is complete.
+     */
+    async flush(_options = {}) {
+        const state = this._flushState;
+        if (state.isFlushing) {
+            return state.currentPromise;
+        }
+        if (this.getCurrentBufferedCount() === 0) {
+            return Promise.resolve();
+        }
+        const { promise, resolve } = getPromiseParts();
+        this._flushState = { isFlushing: true, currentPromise: promise, resolve };
+        return promise;
+    }
+    /**
+     * Returns a promise that resolves once there is room for events to be added
+     * to the buffer.
+     */
+    _ready() {
+        const currentBufferedCount = this.getCurrentBufferedCount();
+        // If the buffer isn't full and we don't have any pending `ready()` calls,
+        // then it's safe to return right away.
+        if (currentBufferedCount < this._maxBufferSize &&
+            !this._readyQueue.length &&
+            !this._flushState.isFlushing) {
+            return Promise.resolve();
+        }
+        const { promise: readyPromise, reject, resolve } = getPromiseParts();
+        this._readyQueue.push({ resolve, reject });
+        return readyPromise;
+    }
+    /**
+     * Starts the loop that creates batches and sends them to the Event Hub.
+     *
+     * The loop will run until the `_loopAbortSignal` is aborted.
+     */
+    async _startPublishLoop() {
+        let batch;
+        // `eventToAddToBatch` is used to keep track of an event that has been removed
+        // from the queue, but has not yet been added to a batch.
+        // This prevents losing an event if a `sendBatch` or `createBatch` call fails
+        // before the event is added to a batch.
+        let eventToAddToBatch;
+        while (!this._loopAbortSignal.aborted) {
+            try {
+                if (!isDefined(batch)) {
+                    batch = await this._createBatch();
+                }
+                const timeSinceLastBatchCreation = Date.now() - this._lastBatchCreationTime;
+                const maximumTimeToWaitForEvent = batch.count
+                    ? Math.max(this._maxWaitTimeInMs - timeSinceLastBatchCreation, 0)
+                    : this._maxWaitTimeInMs;
+                const event = eventToAddToBatch !== null && eventToAddToBatch !== void 0 ? eventToAddToBatch : (await cancelablePromiseRace([
+                    (abortOptions) => this._eventQueue.shift(abortOptions),
+                    (abortOptions) => delay(maximumTimeToWaitForEvent, abortOptions.abortSignal, abortOptions.abortErrorMsg),
+                ], { abortSignal: this._loopAbortSignal }));
+                if (!event) {
+                    // We didn't receive an event within the allotted time.
+                    // Send the existing batch if it has events in it.
+                    if (batch.count) {
+                        await this._producer.sendBatch(batch);
+                        this._reportSuccess();
+                        batch = await this._createBatch();
+                    }
+                    continue;
+                }
+                else if (!eventToAddToBatch) {
+                    eventToAddToBatch = event;
+                }
+                const didAdd = batch.tryAdd(event);
+                if (didAdd) {
+                    // This event will definitely make it to one of the user-provided handlers
+                    // since it was added to a batch.
+                    // Store it so we can return it in a handler later.
+                    this._batchedEvents.push(event);
+                    // Clear reference to existing event since it has been added to the batch.
+                    eventToAddToBatch = undefined;
+                }
+                if (didAdd && batch.count >= this._maxBufferSize) {
+                    // Whenever batch.count exceeds the max count of buffered events, send the batch.
+                    await this._producer.sendBatch(batch);
+                    this._reportSuccess();
+                    batch = await this._createBatch();
+                }
+                else if (!didAdd && batch.count) {
+                    // If the event wasn't able to be added and the current batch isn't empty,
+                    // attempt to send the current batch and add the event to a new batch.
+                    await this._producer.sendBatch(batch);
+                    this._reportSuccess();
+                    batch = await this._createBatch();
+                }
+                if (!didAdd && !batch.tryAdd(event)) {
+                    // TODO: Report MaxMesageSizeExceeded error. Mimic service's error.
+                    this._reportFailure(new Error("Placeholder for max message size exceeded"), event);
+                }
+                else if (!didAdd) {
+                    // Handles the case where the event _was_ successfull added to the new batch.
+                    this._batchedEvents.push(event);
+                }
+                // Clear reference to existing event since it has been added to the batch.
+                eventToAddToBatch = undefined;
+            }
+            catch (err) {
+                if (!isObjectWithProperties(err, ["name"]) || err.name !== "AbortError") {
+                    this._reportFailure(err);
+                    batch = undefined;
+                    this._batchedEvents = [];
+                }
+            }
+        }
+    }
+    /**
+     * Helper method that returns an `EventDataBatch`.
+     * This also has the side effects of
+     *  - keeping track of batch creation time: needed for maxWaitTime calculations.
+     *  - clearing reference to batched events.
+     *  - incrementing the readiness: creating a new batch indicates the buffer
+     *    should have room, so we can resolve some pending `ready()` calls.
+     */
+    async _createBatch() {
+        this._lastBatchCreationTime = Date.now();
+        this._batchedEvents = [];
+        const batch = await this._producer.createBatch({
+            partitionId: this._partitionId,
+        });
+        this._incrementReadiness();
+        return batch;
+    }
+    /**
+     * This method will resolve as many pending `ready()` calls as it can
+     * based on how much space remains in the buffer.
+     *
+     * If the channel is currently flushing, this is a no-op. This prevents
+     * `enqueueEvent` calls from adding the event to the buffer until flushing
+     * completes.
+     */
+    _incrementReadiness() {
+        var _a;
+        if (this._flushState.isFlushing) {
+            return;
+        }
+        const currentBufferedCount = this.getCurrentBufferedCount();
+        const num = Math.min(this._maxBufferSize - currentBufferedCount, this._readyQueue.length);
+        for (let i = 0; i < num; i++) {
+            (_a = this._readyQueue.shift()) === null || _a === void 0 ? void 0 : _a.resolve();
+        }
+    }
+    /**
+     * Calls the user-provided `onSendEventsSuccessHandler` with the events
+     * that were successfully sent.
+     */
+    _reportSuccess() {
+        var _a;
+        this._bufferCount = this._bufferCount - this._batchedEvents.length;
+        this._updateFlushState();
+        try {
+            (_a = this._onSendEventsSuccessHandler) === null || _a === void 0 ? void 0 : _a.call(this, {
+                events: this._batchedEvents,
+                partitionId: this._partitionId,
+            });
+        }
+        catch (e) {
+            logger.error(`The following error occurred in the onSendEventsSuccessHandler: ${JSON.stringify(e, undefined, "  ")}`);
+        }
+    }
+    /**
+     * Calls the user-provided `onSendEventsErrorHandler` with an error and the events
+     * that were not successfully sent.
+     */
+    _reportFailure(err, event) {
+        this._bufferCount = this._bufferCount - (event ? 1 : this._batchedEvents.length);
+        this._updateFlushState();
+        try {
+            this._onSendEventsErrorHandler({
+                error: err,
+                events: event ? [event] : this._batchedEvents,
+                partitionId: this._partitionId,
+            });
+        }
+        catch (e) {
+            logger.error(`The following error occurred in the onSendEventsErrorHandler: ${JSON.stringify(e, undefined, "  ")}`);
+        }
+    }
+    /**
+     * Updates the channel's flush state once the size of the
+     * event buffer has decreased to 0.
+     */
+    _updateFlushState() {
+        const state = this._flushState;
+        if (!state.isFlushing || this.getCurrentBufferedCount() !== 0) {
+            return;
+        }
+        state.resolve();
+        this._flushState = { isFlushing: false };
+        this._incrementReadiness();
+    }
+}
+//# sourceMappingURL=batchingPartitionChannel.js.map

package/dist/esm/batchingPartitionChannel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batchingPartitionChannel.js","sourceRoot":"","sources":["../../src/batchingPartitionChannel.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AASzC,OAAO,EAAE,SAAS,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAErE,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAkBzD;;;;GAIG;AACH,MAAM,OAAO,wBAAwB;IAwBnC,YAAY,EACV,eAAe,EACf,aAAa,EACb,eAAe,EACf,wBAAwB,EACxB,0BAA0B,EAC1B,WAAW,EACX,QAAQ,GACsB;QA/BxB,gBAAW,GAAG,IAAI,cAAc,EAAoC,CAAC;QACrE,mBAAc,GAA4C,EAAE,CAAC;QAC7D,iBAAY,GAAW,CAAC,CAAC;QACzB,gBAAW,GAGd,EAAE,CAAC;QACA,gBAAW,GAE4D;YAC7E,UAAU,EAAE,KAAK;SAClB,CAAC;QACM,eAAU,GAAY,KAAK,CAAC;QAC5B,2BAAsB,GAAW,CAAC,CAAC;QAmBzC,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,eAAe,CAAC;QACxC,IAAI,CAAC,yBAAyB,GAAG,wBAAwB,CAAC;QAC1D,IAAI,CAAC,2BAA2B,GAAG,0BAA0B,CAAC;QAC9D,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;IAC5B,CAAC;IAED,uBAAuB;QACrB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAuC;QACxD,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QACpB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;gBACnC,MAAM,CAAC,KAAK,CACV,iEAAiE,IAAI,CAAC,SAAS,CAC7E,CAAC,EACD,SAAS,EACT,IAAI,CACL,EAAE,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,WAA6B,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC;QAC/B,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC,cAAc,CAAC;QAC9B,CAAC;QAED,IAAI,IAAI,CAAC,uBAAuB,EAAE,KAAK,CAAC,EAAE,CAAC;YACzC,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,eAAe,EAAQ,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAE1E,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACK,MAAM;QACZ,MAAM,oBAAoB,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAE5D,0EAA0E;QAC1E,uCAAuC;QACvC,IACE,oBAAoB,GAAG,IAAI,CAAC,cAAc;YAC1C,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM;YACxB,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU,EAC5B,CAAC;YACD,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,EAAQ,CAAC;QAC3E,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAE3C,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,iBAAiB;QAC7B,IAAI,KAAiC,CAAC;QACtC,8EAA8E;QAC9E,yDAAyD;QACzD,6EAA6E;QAC7E,wCAAwC;QACxC,IAAI,iBAA+D,CAAC;QACpE,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;oBACtB,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpC,CAAC;gBACD,MAAM,0BAA0B,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,sBAAsB,CAAC;gBAC5E,MAAM,yBAAyB,GAAG,KAAK,CAAC,KAAK;oBAC3C,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,GAAG,0BAA0B,EAAE,CAAC,CAAC;oBACjE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAE1B,MAAM,KAAK,GACT,iBAAiB,aAAjB,iBAAiB,cAAjB,iBAAiB,GACjB,CAAC,MAAM,qBAAqB,CAC1B;oBACE,CAAC,YAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,YAAY,CAAC;oBACpE,CAAC,YAA0B,EAAE,EAAE,CAC7B,KAAK,CACH,yBAAyB,EACzB,YAAY,CAAC,WAAW,EACxB,YAAY,CAAC,aAAa,CAC3B;iBACJ,EACD,EAAE,WAAW,EAAE,IAAI,CAAC,gBAAgB,EAAE,CACvC,CAAC,CAAC;gBAEL,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,uDAAuD;oBACvD,kDAAkD;oBAClD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;wBAChB,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;wBACtC,IAAI,CAAC,cAAc,EAAE,CAAC;wBACtB,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;oBACpC,CAAC;oBACD,SAAS;gBACX,CAAC;qBAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBAC9B,iBAAiB,GAAG,KAAK,CAAC;gBAC5B,CAAC;gBAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,MAAM,EAAE,CAAC;oBACX,0EAA0E;oBAC1E,iCAAiC;oBACjC,mDAAmD;oBACnD,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBAChC,0EAA0E;oBAC1E,iBAAiB,GAAG,SAAS,CAAC;gBAChC,CAAC;gBAED,IAAI,MAAM,IAAI,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;oBACjD,iFAAiF;oBACjF,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;oBACtC,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpC,CAAC;qBAAM,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAClC,0EAA0E;oBAC1E,sEAAsE;oBACtE,MAAM,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;oBACtC,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpC,CAAC;gBAED,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;oBACpC,mEAAmE;oBACnE,IAAI,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC,2CAA2C,CAAC,EAAE,KAAK,CAAC,CAAC;gBACrF,CAAC;qBAAM,IAAI,CAAC,MAAM,EAAE,CAAC;oBACnB,6EAA6E;oBAC7E,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAClC,CAAC;gBACD,0EAA0E;gBAC1E,iBAAiB,GAAG,SAAS,CAAC;YAChC,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,sBAAsB,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACxE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;oBACzB,KAAK,GAAG,SAAS,CAAC;oBAClB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,YAAY;QACxB,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;YAC7C,WAAW,EAAE,IAAI,CAAC,YAAY;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;;;OAOG;IACK,mBAAmB;;QACzB,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QACD,MAAM,oBAAoB,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC5D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,GAAG,oBAAoB,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC1F,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,MAAA,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,0CAAE,OAAO,EAAE,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,cAAc;;QACpB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;QACnE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,MAAA,IAAI,CAAC,2BAA2B,qDAAG;gBACjC,MAAM,EAAE,IAAI,CAAC,cAAc;gBAC3B,WAAW,EAAE,IAAI,CAAC,YAAY;aAC/B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CACV,mEAAmE,IAAI,CAAC,SAAS,CAC/E,CAAC,EACD,SAAS,EACT,IAAI,CACL,EAAE,CACJ,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,cAAc,CAAC,GAAQ,EAAE,KAAwC;QACvE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACjF,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,IAAI,CAAC,yBAAyB,CAAC;gBAC7B,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc;gBAC7C,WAAW,EAAE,IAAI,CAAC,YAAY;aAC/B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAU,EAAE,CAAC;YACpB,MAAM,CAAC,KAAK,CACV,iEAAiE,IAAI,CAAC,SAAS,CAC7E,CAAC,EACD,SAAS,EACT,IAAI,CACL,EAAE,CACJ,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,iBAAiB;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC,uBAAuB,EAAE,KAAK,CAAC,EAAE,CAAC;YAC9D,OAAO;QACT,CAAC;QAED,KAAK,CAAC,OAAO,EAAE,CAAC;QAEhB,IAAI,CAAC,WAAW,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;QACzC,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC7B,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { AmqpAnnotatedMessage } from \"@azure/core-amqp\";\nimport { delay } from \"@azure/core-amqp\";\nimport type {\n  EventData,\n  EventDataBatch,\n  EventHubBufferedProducerClientOptions,\n  EventHubProducerClient,\n  OperationOptions,\n} from \"./index.js\";\nimport type { AbortOptions } from \"@azure/core-util\";\nimport { isDefined, isObjectWithProperties } from \"@azure/core-util\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { AwaitableQueue } from \"./impl/awaitableQueue.js\";\nimport { getPromiseParts } from \"./util/getPromiseParts.js\";\nimport { logger } from \"./logger.js\";\nimport { cancelablePromiseRace } from \"@azure/core-util\";\n\nexport interface BatchingPartitionChannelProps {\n  loopAbortSignal: AbortSignalLike;\n  maxBufferSize: number;\n  maxWaitTimeInMs: number;\n  partitionId: string;\n  producer: EventHubProducerClient;\n  /**\n   * The handler to call once a batch has successfully published.\n   */\n  onSendEventsSuccessHandler?: EventHubBufferedProducerClientOptions[\"onSendEventsSuccessHandler\"];\n  /**\n   * The handler to call when a batch fails to publish.\n   */\n  onSendEventsErrorHandler: EventHubBufferedProducerClientOptions[\"onSendEventsErrorHandler\"];\n}\n\n/**\n * The `BatchingPartitionChannel` is responsible for accepting enqueued events\n * and optimally batching and sending them to an Event Hub.\n * @internal\n */\nexport class BatchingPartitionChannel {\n  private _eventQueue = new AwaitableQueue<EventData | AmqpAnnotatedMessage>();\n  private _batchedEvents: Array<EventData | AmqpAnnotatedMessage> = [];\n  private _bufferCount: number = 0;\n  private _readyQueue: Array<{\n    resolve: (value: void) => void;\n    reject: (reason?: any) => void;\n  }> = [];\n  private _flushState:\n    | { isFlushing: false }\n    | { isFlushing: true; currentPromise: Promise<void>; resolve: () => void } = {\n    isFlushing: false,\n  };\n  private _isRunning: boolean = false;\n  private _lastBatchCreationTime: number = 0;\n  private _loopAbortSignal: AbortSignalLike;\n  private _maxBufferSize: number;\n  private _maxWaitTimeInMs: number;\n  private _onSendEventsErrorHandler: EventHubBufferedProducerClientOptions[\"onSendEventsErrorHandler\"];\n  private _onSendEventsSuccessHandler?: EventHubBufferedProducerClientOptions[\"onSendEventsSuccessHandler\"];\n\n  private _partitionId: string;\n  private _producer: EventHubProducerClient;\n\n  constructor({\n    loopAbortSignal,\n    maxBufferSize,\n    maxWaitTimeInMs,\n    onSendEventsErrorHandler,\n    onSendEventsSuccessHandler,\n    partitionId,\n    producer,\n  }: BatchingPartitionChannelProps) {\n    this._loopAbortSignal = loopAbortSignal;\n    this._maxBufferSize = maxBufferSize;\n    this._maxWaitTimeInMs = maxWaitTimeInMs;\n    this._onSendEventsErrorHandler = onSendEventsErrorHandler;\n    this._onSendEventsSuccessHandler = onSendEventsSuccessHandler;\n    this._partitionId = partitionId;\n    this._producer = producer;\n  }\n\n  getCurrentBufferedCount(): number {\n    return this._bufferCount;\n  }\n\n  async enqueueEvent(event: EventData | AmqpAnnotatedMessage): Promise<void> {\n    await this._ready();\n    this._eventQueue.push(event);\n    this._bufferCount++;\n\n    if (!this._isRunning) {\n      this._isRunning = true;\n      this._startPublishLoop().catch((e) => {\n        logger.error(\n          `The following error occured during batch creation or sending: ${JSON.stringify(\n            e,\n            undefined,\n            \"  \",\n          )}`,\n        );\n      });\n    }\n  }\n\n  /**\n   * Sets the flush state so that no new events can be enqueued until\n   * all the currently buffered events are sent to the Event Hub.\n   *\n   * Returns a promise that resolves once flushing is complete.\n   */\n  async flush(_options: OperationOptions = {}): Promise<void> {\n    const state = this._flushState;\n    if (state.isFlushing) {\n      return state.currentPromise;\n    }\n\n    if (this.getCurrentBufferedCount() === 0) {\n      return Promise.resolve();\n    }\n\n    const { promise, resolve } = getPromiseParts<void>();\n    this._flushState = { isFlushing: true, currentPromise: promise, resolve };\n\n    return promise;\n  }\n\n  /**\n   * Returns a promise that resolves once there is room for events to be added\n   * to the buffer.\n   */\n  private _ready(): Promise<void> {\n    const currentBufferedCount = this.getCurrentBufferedCount();\n\n    // If the buffer isn't full and we don't have any pending `ready()` calls,\n    // then it's safe to return right away.\n    if (\n      currentBufferedCount < this._maxBufferSize &&\n      !this._readyQueue.length &&\n      !this._flushState.isFlushing\n    ) {\n      return Promise.resolve();\n    }\n\n    const { promise: readyPromise, reject, resolve } = getPromiseParts<void>();\n    this._readyQueue.push({ resolve, reject });\n\n    return readyPromise;\n  }\n\n  /**\n   * Starts the loop that creates batches and sends them to the Event Hub.\n   *\n   * The loop will run until the `_loopAbortSignal` is aborted.\n   */\n  private async _startPublishLoop() {\n    let batch: EventDataBatch | undefined;\n    // `eventToAddToBatch` is used to keep track of an event that has been removed\n    // from the queue, but has not yet been added to a batch.\n    // This prevents losing an event if a `sendBatch` or `createBatch` call fails\n    // before the event is added to a batch.\n    let eventToAddToBatch: EventData | AmqpAnnotatedMessage | undefined;\n    while (!this._loopAbortSignal.aborted) {\n      try {\n        if (!isDefined(batch)) {\n          batch = await this._createBatch();\n        }\n        const timeSinceLastBatchCreation = Date.now() - this._lastBatchCreationTime;\n        const maximumTimeToWaitForEvent = batch.count\n          ? Math.max(this._maxWaitTimeInMs - timeSinceLastBatchCreation, 0)\n          : this._maxWaitTimeInMs;\n\n        const event =\n          eventToAddToBatch ??\n          (await cancelablePromiseRace<[EventData | AmqpAnnotatedMessage, void]>(\n            [\n              (abortOptions: AbortOptions) => this._eventQueue.shift(abortOptions),\n              (abortOptions: AbortOptions) =>\n                delay<void>(\n                  maximumTimeToWaitForEvent,\n                  abortOptions.abortSignal,\n                  abortOptions.abortErrorMsg,\n                ),\n            ],\n            { abortSignal: this._loopAbortSignal },\n          ));\n\n        if (!event) {\n          // We didn't receive an event within the allotted time.\n          // Send the existing batch if it has events in it.\n          if (batch.count) {\n            await this._producer.sendBatch(batch);\n            this._reportSuccess();\n            batch = await this._createBatch();\n          }\n          continue;\n        } else if (!eventToAddToBatch) {\n          eventToAddToBatch = event;\n        }\n\n        const didAdd = batch.tryAdd(event);\n        if (didAdd) {\n          // This event will definitely make it to one of the user-provided handlers\n          // since it was added to a batch.\n          // Store it so we can return it in a handler later.\n          this._batchedEvents.push(event);\n          // Clear reference to existing event since it has been added to the batch.\n          eventToAddToBatch = undefined;\n        }\n\n        if (didAdd && batch.count >= this._maxBufferSize) {\n          // Whenever batch.count exceeds the max count of buffered events, send the batch.\n          await this._producer.sendBatch(batch);\n          this._reportSuccess();\n          batch = await this._createBatch();\n        } else if (!didAdd && batch.count) {\n          // If the event wasn't able to be added and the current batch isn't empty,\n          // attempt to send the current batch and add the event to a new batch.\n          await this._producer.sendBatch(batch);\n          this._reportSuccess();\n          batch = await this._createBatch();\n        }\n\n        if (!didAdd && !batch.tryAdd(event)) {\n          // TODO: Report MaxMesageSizeExceeded error. Mimic service's error.\n          this._reportFailure(new Error(\"Placeholder for max message size exceeded\"), event);\n        } else if (!didAdd) {\n          // Handles the case where the event _was_ successfull added to the new batch.\n          this._batchedEvents.push(event);\n        }\n        // Clear reference to existing event since it has been added to the batch.\n        eventToAddToBatch = undefined;\n      } catch (err: any) {\n        if (!isObjectWithProperties(err, [\"name\"]) || err.name !== \"AbortError\") {\n          this._reportFailure(err);\n          batch = undefined;\n          this._batchedEvents = [];\n        }\n      }\n    }\n  }\n\n  /**\n   * Helper method that returns an `EventDataBatch`.\n   * This also has the side effects of\n   *  - keeping track of batch creation time: needed for maxWaitTime calculations.\n   *  - clearing reference to batched events.\n   *  - incrementing the readiness: creating a new batch indicates the buffer\n   *    should have room, so we can resolve some pending `ready()` calls.\n   */\n  private async _createBatch(): Promise<EventDataBatch> {\n    this._lastBatchCreationTime = Date.now();\n    this._batchedEvents = [];\n    const batch = await this._producer.createBatch({\n      partitionId: this._partitionId,\n    });\n    this._incrementReadiness();\n    return batch;\n  }\n\n  /**\n   * This method will resolve as many pending `ready()` calls as it can\n   * based on how much space remains in the buffer.\n   *\n   * If the channel is currently flushing, this is a no-op. This prevents\n   * `enqueueEvent` calls from adding the event to the buffer until flushing\n   * completes.\n   */\n  private _incrementReadiness() {\n    if (this._flushState.isFlushing) {\n      return;\n    }\n    const currentBufferedCount = this.getCurrentBufferedCount();\n    const num = Math.min(this._maxBufferSize - currentBufferedCount, this._readyQueue.length);\n    for (let i = 0; i < num; i++) {\n      this._readyQueue.shift()?.resolve();\n    }\n  }\n\n  /**\n   * Calls the user-provided `onSendEventsSuccessHandler` with the events\n   * that were successfully sent.\n   */\n  private _reportSuccess() {\n    this._bufferCount = this._bufferCount - this._batchedEvents.length;\n    this._updateFlushState();\n    try {\n      this._onSendEventsSuccessHandler?.({\n        events: this._batchedEvents,\n        partitionId: this._partitionId,\n      });\n    } catch (e: unknown) {\n      logger.error(\n        `The following error occurred in the onSendEventsSuccessHandler: ${JSON.stringify(\n          e,\n          undefined,\n          \"  \",\n        )}`,\n      );\n    }\n  }\n\n  /**\n   * Calls the user-provided `onSendEventsErrorHandler` with an error and the events\n   * that were not successfully sent.\n   */\n  private _reportFailure(err: any, event?: EventData | AmqpAnnotatedMessage) {\n    this._bufferCount = this._bufferCount - (event ? 1 : this._batchedEvents.length);\n    this._updateFlushState();\n    try {\n      this._onSendEventsErrorHandler({\n        error: err,\n        events: event ? [event] : this._batchedEvents,\n        partitionId: this._partitionId,\n      });\n    } catch (e: unknown) {\n      logger.error(\n        `The following error occurred in the onSendEventsErrorHandler: ${JSON.stringify(\n          e,\n          undefined,\n          \"  \",\n        )}`,\n      );\n    }\n  }\n\n  /**\n   * Updates the channel's flush state once the size of the\n   * event buffer has decreased to 0.\n   */\n  private _updateFlushState() {\n    const state = this._flushState;\n    if (!state.isFlushing || this.getCurrentBufferedCount() !== 0) {\n      return;\n    }\n\n    state.resolve();\n\n    this._flushState = { isFlushing: false };\n    this._incrementReadiness();\n  }\n}\n"]}