move memory logging to util

Author: molly-moen

org-code-javabuilder/lib/src/main/java/org/code/javabuilder/CodeExecutionManager.java

@@ -6,9 +6,6 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.PrintStream;
-import java.lang.management.ClassLoadingMXBean;
-import java.lang.management.ManagementFactory;
-import java.lang.management.MemoryPoolMXBean;
 import java.nio.file.Files;
 import java.util.List;
 import org.code.javabuilder.util.LambdaUtils;
@@ -37,8 +34,6 @@ public class CodeExecutionManager {
   private PrintStream systemOutputStream;
   private boolean isInitialized;
 
-  private static final boolean LOG_MEMORY_USAGE = false;
-
   static class CodeBuilderRunnableFactory {
     public CodeBuilderRunnable createCodeBuilderRunnable(
         ProjectFileLoader fileLoader,
@@ -156,7 +151,6 @@ private void onPostExecute() {
     LambdaUtils.safelySendMessage(
         this.outputAdapter, new StatusMessage(StatusMessageKey.EXITED), false);
     JavabuilderContext.getInstance().onExecutionEnded();
-    this.logMemoryUsage();
     try {
       // Close custom input/output streams
       this.overrideInputStream.close();
@@ -176,25 +170,4 @@ private void onPostExecute() {
       this.isInitialized = false;
     }
   }
-
-  private void logMemoryUsage() {
-    if (LOG_MEMORY_USAGE) {
-      List<MemoryPoolMXBean> memoryPoolMXBeans = ManagementFactory.getMemoryPoolMXBeans();
-      for (MemoryPoolMXBean memoryPoolMXBean : memoryPoolMXBeans) {
-        String usageMessage =
-            String.format(
-                "Memory pool %s\n\t Usage: %d\n\tPeak Usage %d",
-                memoryPoolMXBean.getName(),
-                memoryPoolMXBean.getUsage().getUsed(),
-                memoryPoolMXBean.getPeakUsage().getUsed());
-        LoggerUtils.logInfo(usageMessage);
-      }
-      ClassLoadingMXBean classLoadingMXBean = ManagementFactory.getClassLoadingMXBean();
-      String classLoadingMessage =
-          String.format(
-              "Loaded classes: %d, Unloaded classes: %d",
-              classLoadingMXBean.getLoadedClassCount(), classLoadingMXBean.getUnloadedClassCount());
-      LoggerUtils.logInfo(classLoadingMessage);
-    }
-  }
 }

org-code-javabuilder/lib/src/main/java/org/code/javabuilder/util/ProfilingUtils.java

@@ -0,0 +1,35 @@
+package org.code.javabuilder.util;
+
+import java.lang.management.ClassLoadingMXBean;
+import java.lang.management.ManagementFactory;
+import java.lang.management.MemoryPoolMXBean;
+import java.util.List;
+import org.code.protocol.LoggerUtils;
+
+public class ProfilingUtils {
+  private ProfilingUtils() {
+    throw new UnsupportedOperationException("Instantiation of utility class is not allowed.");
+  }
+
+  /*
+  Log current and peak memory usage across all memory pools. Also log currently loaded and unload class counts.
+   */
+  public static void logMemoryUsage() {
+    List<MemoryPoolMXBean> memoryPoolMXBeans = ManagementFactory.getMemoryPoolMXBeans();
+    for (MemoryPoolMXBean memoryPoolMXBean : memoryPoolMXBeans) {
+      String usageMessage =
+          String.format(
+              "Memory pool %s\n\t Usage: %d\n\tPeak Usage %d",
+              memoryPoolMXBean.getName(),
+              memoryPoolMXBean.getUsage().getUsed(),
+              memoryPoolMXBean.getPeakUsage().getUsed());
+      LoggerUtils.logInfo(usageMessage);
+    }
+    ClassLoadingMXBean classLoadingMXBean = ManagementFactory.getClassLoadingMXBean();
+    String classLoadingMessage =
+        String.format(
+            "Loaded classes: %d, Unloaded classes: %d",
+            classLoadingMXBean.getLoadedClassCount(), classLoadingMXBean.getUnloadedClassCount());
+    LoggerUtils.logInfo(classLoadingMessage);
+  }
+}