migrate from preloading activity to startup activity.

Author: wangmingliang-ms

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/java/com/microsoft/azure/toolkit/intellij/cosmos/dbtools/DbToolsWorkaround.java

@@ -10,12 +10,16 @@
 import com.intellij.database.dataSource.DatabaseDriverManagerImpl;
 import com.intellij.database.dataSource.url.ui.UrlPropertiesPanel;
 import com.intellij.openapi.application.ApplicationManager;
-import com.intellij.openapi.application.PreloadingActivity;
+import com.intellij.openapi.project.DumbAware;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.startup.ProjectActivity;
 import com.intellij.openapi.util.JDOMUtil;
 import com.intellij.openapi.util.registry.Registry;
 import com.microsoft.azure.toolkit.intellij.common.IntelliJAzureIcons;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
+import kotlin.Unit;
+import kotlin.coroutines.Continuation;
 import lombok.SneakyThrows;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.commons.lang3.reflect.FieldUtils;
@@ -29,7 +33,7 @@
 import java.util.Objects;
 import java.util.Optional;
 
-public class DbToolsWorkaround extends PreloadingActivity {
+public class DbToolsWorkaround implements ProjectActivity, DumbAware {
     private static final String PARAM_NAME = "account";
     public static final String COSMOS_MONGO_ICON = "icons/Microsoft.DocumentDB/databaseAccounts/mongo.svg";
     public static final String COSMOS_MONGO_DRIVER_ID = "az_cosmos_mongo";
@@ -39,7 +43,7 @@ public class DbToolsWorkaround extends PreloadingActivity {
     public static final String COSMOS_CASSANDRA_DRIVER_CONFIG = "databaseDrivers/azure-cosmos-cassandra-drivers.xml";
 
     @Override
-    public void preload() {
+    public Object execute(@Nonnull Project project, @Nonnull Continuation<? super Unit> continuation) {
         ApplicationManager.getApplication().executeOnPooledThread(() -> {
             try {
                 DbToolsWorkaround.makeAccountShowAtTop();
@@ -50,6 +54,7 @@ public void preload() {
                 AzureTelemeter.log(AzureTelemetry.Type.ERROR, new HashMap<>(), t);
             }
         });
+        return null;
     }
 
     private static void loadMongoDriver() {

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-cosmos/src/main/resources/META-INF/azure-intellij-plugin-cosmos-dbtools.xml

@@ -15,7 +15,7 @@
         <registryKey key="azure.toolkit.cosmos_cassandra.dbtools.enabled"
                      description="Enables DB tools related features of Azure cosmos DB API for cassandra"
                      restartRequired="true" defaultValue="false"/>
-        <preloadingActivity implementation="com.microsoft.azure.toolkit.intellij.cosmos.dbtools.DbToolsWorkaround"/>
+        <postStartupActivity implementation="com.microsoft.azure.toolkit.intellij.cosmos.dbtools.DbToolsWorkaround"/>
     </extensions>
     <extensions defaultExtensionNs="com.microsoft.tooling.msservices.intellij.azure">
         <actions implementation="com.microsoft.azure.toolkit.intellij.cosmos.IntelliJCosmosActionsContributorForUltimate"/>

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/java/com/microsoft/azure/toolkit/intellij/database/dbtools/DatabaseDbToolsWorkaround.java

@@ -9,18 +9,26 @@
 import com.intellij.database.dataSource.DatabaseDriverManager;
 import com.intellij.database.dataSource.url.template.UrlTemplate;
 import com.intellij.openapi.application.ApplicationManager;
-import com.intellij.openapi.application.PreloadingActivity;
+import com.intellij.openapi.project.DumbAware;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.startup.ProjectActivity;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemeter;
 import com.microsoft.azure.toolkit.lib.common.telemetry.AzureTelemetry;
+import kotlin.Unit;
+import kotlin.coroutines.Continuation;
 
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
 import java.util.HashMap;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Objects;
 
-public class DatabaseDbToolsWorkaround extends PreloadingActivity {
+public class DatabaseDbToolsWorkaround implements ProjectActivity, DumbAware {
+
+    @Nullable
     @Override
-    public void preload() {
+    public Object execute(@Nonnull Project project, @Nonnull Continuation<? super Unit> continuation) {
         ApplicationManager.getApplication().executeOnPooledThread(() -> {
             try {
                 loadMySqlAzureTemplates();
@@ -32,6 +40,7 @@ public void preload() {
                 AzureTelemeter.log(AzureTelemetry.Type.ERROR, new HashMap<>(), t);
             }
         });
+        return null;
     }
 
     private static void loadMySqlAzureTemplates() {

PluginsAndFeatures/azure-toolkit-for-intellij/azure-intellij-plugin-database/src/main/resources/META-INF/azure-intellij-plugin-database-dbtools.xml

@@ -5,6 +5,6 @@
         <urlParamEditorUiProvider implementation="com.microsoft.azure.toolkit.intellij.database.dbtools.DatabaseServerTypeUIFactory" order="first"/>
     </extensions>
     <extensions defaultExtensionNs="com.intellij">
-        <preloadingActivity implementation="com.microsoft.azure.toolkit.intellij.database.dbtools.DatabaseDbToolsWorkaround"/>
+        <postStartupActivity implementation="com.microsoft.azure.toolkit.intellij.database.dbtools.DatabaseDbToolsWorkaround"/>
     </extensions>
 </idea-plugin>