Merge pull request #365 from ThemiraChathumina/fix-lag

Fix lagging issues in the version 0.96.0

Author: NipunaRanasinghe

src/main/java/org/wso2/lsp4intellij/client/DefaultLanguageClient.java

@@ -44,6 +44,7 @@
 import org.eclipse.lsp4j.WorkDoneProgressCreateParams;
 import org.eclipse.lsp4j.WorkDoneProgressEnd;
 import org.eclipse.lsp4j.WorkDoneProgressNotification;
+import org.eclipse.lsp4j.WorkDoneProgressReport;
 import org.eclipse.lsp4j.WorkspaceFolder;
 import org.eclipse.lsp4j.services.LanguageClient;
 import org.jetbrains.annotations.NotNull;
@@ -302,7 +303,7 @@ public CompletableFuture<Void> createProgress(WorkDoneProgressCreateParams param
         }
         Tuple2<String, String> progressNotificationItem = new Tuple2<>("LSP Progress Notification", "");
         progressNotificationItems.put(token, progressNotificationItem);
-        return null;
+        return CompletableFuture.completedFuture(null);
     }
 
     @Override
@@ -330,6 +331,11 @@ public void notifyProgress(ProgressParams params) {
             } else {
                 progressNotificationItems.put(token, progressNotificationItem);
             }
+        } else if (progressNotification instanceof WorkDoneProgressReport) {
+            message = ((WorkDoneProgressReport) progressNotification).getMessage();
+            if (progressNotificationItems.containsKey(token)) {
+                title = progressNotificationItems.get(token).getFirst();
+            }
         } else if (progressNotification instanceof WorkDoneProgressEnd) {
             message = ((WorkDoneProgressEnd) progressNotification).getMessage();
             if (progressNotificationItems.containsKey(token)) {

src/main/java/org/wso2/lsp4intellij/client/languageserver/wrapper/MessageHandler.java

@@ -43,11 +43,7 @@ public MessageConsumer apply(MessageConsumer messageConsumer) {
         return message -> {
             if(isRunning.getAsBoolean()) {
                 handleMessage(message);
-                try {
-                    messageConsumer.consume(message);
-                } catch (Exception e) {
-                    Log.warn("Error while consuming message", e);
-                }
+                messageConsumer.consume(message);
             }
         };
     }