Include platform and system statistics in profiler viewer payload

1 files changed, 9 insertions, 7 deletions

diff --git a/spark-common/src/main/java/me/lucko/spark/common/heapdump/HeapDumpSummary.java b/spark-common/src/main/java/me/lucko/spark/common/heapdump/HeapDumpSummary.java
index 34fd6c4..7bb411d 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/heapdump/HeapDumpSummary.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/heapdump/HeapDumpSummary.java
@@ -20,11 +20,11 @@
 
 package me.lucko.spark.common.heapdump;
 
+import me.lucko.spark.common.SparkPlatform;
 import me.lucko.spark.common.command.sender.CommandSender;
-import me.lucko.spark.common.platform.PlatformInfo;
-import me.lucko.spark.proto.SparkProtos;
-import me.lucko.spark.proto.SparkProtos.HeapData;
-import me.lucko.spark.proto.SparkProtos.HeapEntry;
+import me.lucko.spark.proto.SparkHeapProtos.HeapData;
+import me.lucko.spark.proto.SparkHeapProtos.HeapEntry;
+import me.lucko.spark.proto.SparkHeapProtos.HeapMetadata;
 
 import org.objectweb.asm.Type;
 
@@ -125,10 +125,12 @@ public final class HeapDumpSummary {
         this.entries = entries;
     }
 
-    public HeapData toProto(PlatformInfo platformInfo, CommandSender creator) {
+    public HeapData toProto(SparkPlatform platform, CommandSender creator) {
         HeapData.Builder proto = HeapData.newBuilder();
-        proto.setMetadata(SparkProtos.HeapMetadata.newBuilder()
-                .setPlatformMetadata(platformInfo.toData().toProto())
+        proto.setMetadata(HeapMetadata.newBuilder()
+                .setPlatformMetadata(platform.getPlugin().getPlatformInfo().toData().toProto())
+                .setPlatformStatistics(platform.getStatisticsProvider().getPlatformStatistics(null))
+                .setSystemStatistics(platform.getStatisticsProvider().getSystemStatistics())
                 .setCreator(creator.toData().toProto())
                 .build()
         );