EventHubs: Adding retry for create-batch api (Azure#32751)

Author: anuchandy

sdk/eventhubs/azure-messaging-eventhubs/CHANGELOG.md

@@ -10,6 +10,8 @@
 
 ### Bugs Fixed
 
+- Added retry for createBatch API as this API also makes network calls similar to its companion send API.
+
 ### Other Changes
 
 ## 5.15.0 (2022-11-16)

sdk/eventhubs/azure-messaging-eventhubs/src/main/java/com/azure/messaging/eventhubs/EventHubProducerAsyncClient.java

@@ -306,7 +306,7 @@ public Mono<EventDataBatch> createBatch(CreateBatchOptions options) {
                 MAX_PARTITION_KEY_LENGTH)));
         }
 
-        return getSendLink(partitionId)
+        final Mono<EventDataBatch> createBatch = getSendLink(partitionId)
             .flatMap(link -> link.getLinkSize()
                 .flatMap(size -> {
                     final int maximumLinkSize = size > 0
@@ -326,6 +326,12 @@ public Mono<EventDataBatch> createBatch(CreateBatchOptions options) {
 
                     return Mono.just(new EventDataBatch(batchSize, partitionId, partitionKey, link::getErrorContext, instrumentation));
                 }));
+
+        // Similar to the companion API 'send', the 'create-batch' can also make network calls, so retry in case
+        // of transient errors.
+        return withRetry(createBatch, retryOptions,
+            String.format("partitionId[%s]: Creating batch timed out.", partitionId))
+            .publishOn(scheduler);
     }
 
     /**