Merge pull request #5279 from microsoft/fix-preload

[#0] Preloader cause failure in AzureService loading(SPI)

Author: Flanker32

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-lib/src/main/java/com/microsoft/azure/toolkit/intellij/common/preload/AzurePreloadingActivity.java

@@ -8,16 +8,30 @@
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.application.PreloadingActivity;
 import com.intellij.openapi.progress.ProgressIndicator;
+import com.microsoft.azure.toolkit.lib.AzureService;
 import com.microsoft.azure.toolkit.lib.common.cache.Preloader;
 import lombok.extern.java.Log;
 
+import java.util.logging.Level;
+
+import static com.microsoft.azure.toolkit.lib.Azure.az;
+
 @Log
 public class AzurePreloadingActivity extends PreloadingActivity {
 
     @Override
     public void preload(@org.jetbrains.annotations.NotNull final ProgressIndicator indicator) {
-        // Using progress manager as azure task manager is not initialized
-        // TODO: remove this comment
-        // ApplicationManager.getApplication().executeOnPooledThread(Preloader::load);
+        final ClassLoader current = Thread.currentThread().getContextClassLoader();
+        try {
+            final ClassLoader classLoader = AzurePreloadingActivity.class.getClassLoader();
+            Thread.currentThread().setContextClassLoader(classLoader);
+            az(AzureService.class);
+        } catch (final Exception e) {
+            log.log(Level.WARNING, "failed to load AzureServices.", e);
+        } finally {
+            Thread.currentThread().setContextClassLoader(current);
+        }
+        // Using application manager as azure task manager is not initialized
+        ApplicationManager.getApplication().executeOnPooledThread(Preloader::load);
     }
 }