Merge pull request #365 from code-dot-org/molly-add-reflection-access

Add reflection access for validation

Author: molly-moen

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

@@ -138,5 +138,6 @@ private boolean isInAllowedPackage(String[] allowedPackageList, String name) {
       };
 
   // Allowed packages for code with elevated permissions, such as validation code.
-  private static final String[] validatorAllowedPackages = new String[] {"org.code.validation"};
+  private static final String[] validatorAllowedPackages =
+      new String[] {"org.code.validation", "java.lang.reflect"};
 }

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);
+  }
+}