Redefine HCR event. (#133)

yaohaizh · web-flow · commit 36604caeaff2 · 2017-12-18T21:33:53.000-08:00
diff --git a/com.microsoft.java.debug.plugin/src/main/java/com/microsoft/java/debug/plugin/internal/JavaHotCodeReplaceProvider.java b/com.microsoft.java.debug.plugin/src/main/java/com/microsoft/java/debug/plugin/internal/JavaHotCodeReplaceProvider.java
@@ -248,15 +248,41 @@ private IResource getSourceFile(IJavaProject project, String qualifiedName, Stri
         }
     }
 
+    /**
+     * Hot code replace event type.
+     */
+    enum EventType {
+        ERROR(-1),
+
+        WARNING(-2),
+
+        STARTING(1),
+
+        END(2);
+
+        private int value;
+
+        private EventType(int value) {
+            this.value = value;
+        }
+
+        public int getValue() {
+            return this.value;
+        }
+    }
+
     class HotCodeReplaceEvent extends DebugEvent {
 
+        public EventType eventType;
+
         public String message;
 
         /**
          * Constructor.
          */
-        public HotCodeReplaceEvent(String message) {
+        public HotCodeReplaceEvent(EventType eventType, String message) {
             super("hotCodeReplace");
+            this.eventType = eventType;
             this.message = message;
         }
     }
@@ -311,6 +337,9 @@ private void doHotCodeReplace(List<IResource> resourcesToReplace, List<String> q
             return;
         }
 
+        context.getProtocolServer()
+                .sendEvent(new HotCodeReplaceEvent(EventType.STARTING, "Start hot code replacement procedure..."));
+
         try {
             List<ThreadReference> poppedThreads = new ArrayList<>();
             boolean framesPopped = false;
@@ -329,7 +358,8 @@ private void doHotCodeReplace(List<IResource> resourcesToReplace, List<String> q
             }
 
             if (containsObsoleteMethods()) {
-                context.getProtocolServer().sendEvent(new HotCodeReplaceEvent("JVM contains obsolete methods"));
+                context.getProtocolServer()
+                        .sendEvent(new HotCodeReplaceEvent(EventType.ERROR, "JVM contains obsolete methods"));
             }
 
             if (currentDebugSession.getVM().canPopFrames() && framesPopped) {
@@ -339,6 +369,8 @@ private void doHotCodeReplace(List<IResource> resourcesToReplace, List<String> q
             }
         } catch (DebugException e) {
             logger.log(Level.SEVERE, "Failed to complete hot code replace: " + e.getMessage(), e);
+        } finally {
+            context.getProtocolServer().sendEvent(new HotCodeReplaceEvent(EventType.END, "Completed hot code replace"));
         }
 
         threadFrameMap.clear();
@@ -592,7 +624,7 @@ private void redefineTypesJDK(List<IResource> resources, List<String> qualifiedN
             currentDebugSession.getVM().redefineClasses(typesToBytes);
         } catch (UnsupportedOperationException | NoClassDefFoundError | VerifyError | ClassFormatError
                 | ClassCircularityError e) {
-            context.getProtocolServer().sendEvent(new HotCodeReplaceEvent(e.getMessage()));
+            context.getProtocolServer().sendEvent(new HotCodeReplaceEvent(EventType.ERROR, e.getMessage()));
             throw new DebugException("Failed to redefine classes: " + e.getMessage());
         }
     }
