diff --git a/sdk/eventhub/event-hubs/src/eventHubSender.ts b/sdk/eventhub/event-hubs/src/eventHubSender.ts index e5477d731b73..9f10ee229559 100644 --- a/sdk/eventhub/event-hubs/src/eventHubSender.ts +++ b/sdk/eventhub/event-hubs/src/eventHubSender.ts @@ -454,6 +454,13 @@ export class EventHubSender extends LinkEntity { options = {}; } + const operationTimeoutInMs = + options.retryOptions == undefined || + options.retryOptions.operationTimeoutInMs == undefined || + options.retryOptions.operationTimeoutInMs < Constants.defaultOperationTimeoutInSeconds * 1000 + ? Constants.defaultOperationTimeoutInSeconds * 1000 + : options.retryOptions.operationTimeoutInMs; + const abortSignal: AbortSignalLike | undefined = options.abortSignal; const sendEventPromise = () => new Promise((resolve, reject) => { @@ -577,15 +584,7 @@ export class EventHubSender extends LinkEntity { this._sender!.on(SenderEvents.rejected, onRejected); this._sender!.on(SenderEvents.modified, onModified); this._sender!.on(SenderEvents.released, onReleased); - waitTimer = setTimeout( - actionAfterTimeout, - options == undefined || - options.retryOptions == undefined || - options.retryOptions.operationTimeoutInMs == undefined || - options.retryOptions.operationTimeoutInMs < 0 - ? Constants.defaultOperationTimeoutInSeconds * 1000 - : options.retryOptions.operationTimeoutInMs - ); + waitTimer = setTimeout(actionAfterTimeout, operationTimeoutInMs); const delivery = this._sender!.send(message, tag, 0x80013700); log.sender( "[%s] Sender '%s', sent message with delivery id: %d and tag: %s", diff --git a/sdk/eventhub/event-hubs/src/managementClient.ts b/sdk/eventhub/event-hubs/src/managementClient.ts index 6b7d2be3a210..51c74512786f 100644 --- a/sdk/eventhub/event-hubs/src/managementClient.ts +++ b/sdk/eventhub/event-hubs/src/managementClient.ts @@ -292,16 +292,18 @@ export class ManagementClient extends LinkEntity { options = {}; } + const operationTimeoutInMs = + options.retryOptions == undefined || + options.retryOptions.operationTimeoutInMs == undefined || + options.retryOptions.operationTimeoutInMs < Constants.defaultOperationTimeoutInSeconds * 1000 + ? Constants.defaultOperationTimeoutInSeconds * 1000 + : options.retryOptions.operationTimeoutInMs; + const sendRequestOptions: SendRequestOptions = { maxRetries: options.retryOptions && options.retryOptions.maxRetries, abortSignal: options.abortSignal, requestName: options.requestName, - timeoutInSeconds: - options.retryOptions == undefined || - options.retryOptions.operationTimeoutInMs == undefined || - options.retryOptions.operationTimeoutInMs < 0 - ? Constants.defaultOperationTimeoutInSeconds - : options.retryOptions.operationTimeoutInMs / 1000, + timeoutInSeconds: operationTimeoutInMs / 1000, delayInSeconds: options.retryOptions && options.retryOptions.retryInterval && options.retryOptions.retryInterval >= 0 ? options.retryOptions.retryInterval / 1000