Adding try/catch when unable to reschedule scheduler task.

Azure · Jun 22, 2019 · ef030ee · ef030ee
1 parent e100b23
commit ef030ee
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.../src/main/java/com/azure/messaging/eventhubs/implementation/ActiveClientTokenManager.java b/.../src/main/java/com/azure/messaging/eventhubs/implementation/ActiveClientTokenManager.java
@@ -83,7 +83,12 @@ Mono<Void> authorize() {
                     lastRefreshInterval.set(refreshIntervalMS);
 
                     // This converts it to milliseconds
-                    this.timer.schedule(new RefreshAuthorizationToken(), refreshIntervalMS);
+                    try {
+                        this.timer.schedule(new RefreshAuthorizationToken(), refreshIntervalMS);
+                    } catch (IllegalStateException e) {
+                        logger.asWarning().log("Unable to reschedule timer task.", e);
+                        hasScheduled.set(false);
+                    }
                 }
 
                 return expiresOn;
@@ -123,8 +128,11 @@ public void run() {
                     logger.asInfo().log("Success. Rescheduling refresh authorization task.");
                     sink.next(AmqpResponseCode.ACCEPTED);
 
-                    if (hasScheduled.getAndSet(true)) {
+                    try {
                         timer.schedule(new RefreshAuthorizationToken(), lastRefreshInterval.get());
+                    } catch (IllegalStateException e) {
+                        logger.asWarning().log("Unable to reschedule timer task.", e);
+                        hasScheduled.set(false);
                     }
                 });
         }