block() function is not allowed to run in nio thread.

use subscribe() to get string in non-blocking way.
microsoft · Sep 16, 2022 · bb74188 · bb74188
1 parent af8a98c
commit bb74188
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/...ommon-lib/src/main/java/com/microsoft/azure/toolkit/lib/common/messager/AzureMessage.java b/...ommon-lib/src/main/java/com/microsoft/azure/toolkit/lib/common/messager/AzureMessage.java
@@ -39,6 +39,7 @@
 import java.util.Optional;
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Supplier;
 import java.util.stream.Collectors;
 
@@ -105,19 +106,20 @@ protected String getCause(@Nonnull Throwable throwable) {
         if (Objects.isNull(root)) {
             return ExceptionUtils.getRootCause(throwable).toString();
         }
-        String cause = null;
+        AtomicReference<String> cause = new AtomicReference<>(null);
         if (root instanceof ManagementException) {
-            cause = Optional.of((ManagementException) root)
-                .map(ManagementException::getValue)
-                .map(ManagementError::getMessage)
-                .orElse("Unknown cause");
+            cause.set(Optional.of((ManagementException) root)
+                    .map(ManagementException::getValue)
+                    .map(ManagementError::getMessage)
+                    .orElse("Unknown cause"));
         } else if (root instanceof HttpResponseException) {
-            cause = Optional.of((HttpResponseException) root)
+            Optional.of((HttpResponseException) root)
                 .map(HttpResponseException::getResponse)
                 .map(HttpResponse::getBodyAsString)
-                .map(Mono::block).orElse("Unknown cause");
+                .orElse(Mono.just("Unknown cause"))
+                .subscribe(cause::set);
         }
-        final String causeMsg = StringUtils.firstNonBlank(cause, root.getMessage());
+        final String causeMsg = StringUtils.firstNonBlank(cause.get(), root.getMessage());
         return Optional.ofNullable(causeMsg)
             .filter(StringUtils::isNotBlank)
             .map(StringUtils::uncapitalize)