From a2af3f8f7e3693f3e445d2998938bf448d47c35f Mon Sep 17 00:00:00 2001
From: MD <1917406+md678685@users.noreply.github.com>
Date: Tue, 23 Jun 2020 12:02:43 +0100
Subject: Include platform info in sampler and heap summary data (#58)

---
 .../main/java/me/lucko/spark/common/heapdump/HeapDumpSummary.java | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'spark-common/src/main/java/me/lucko/spark/common/heapdump')

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 56958d1..f7a562c 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
@@ -21,6 +21,7 @@
 package me.lucko.spark.common.heapdump;
 
 import me.lucko.spark.common.command.sender.CommandSender;
+import me.lucko.spark.common.PlatformInfo;
 import me.lucko.spark.proto.SparkProtos;
 import me.lucko.spark.proto.SparkProtos.HeapData;
 import me.lucko.spark.proto.SparkProtos.HeapEntry;
@@ -127,9 +128,10 @@ public final class HeapDumpSummary {
         this.entries = entries;
     }
 
-    private HeapData toProto(CommandSender creator) {
+    private HeapData toProto(PlatformInfo platformInfo, CommandSender creator) {
         HeapData.Builder proto = HeapData.newBuilder();
         proto.setMetadata(SparkProtos.HeapMetadata.newBuilder()
+                .setPlatform(platformInfo.toData().toProto())
                 .setUser(creator.toData().toProto())
                 .build()
         );
@@ -141,10 +143,10 @@ public final class HeapDumpSummary {
         return proto.build();
     }
 
-    public byte[] formCompressedDataPayload(CommandSender creator) {
+    public byte[] formCompressedDataPayload(PlatformInfo platformInfo, CommandSender creator) {
         ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
         try (OutputStream out = new GZIPOutputStream(byteOut)) {
-            toProto(creator).writeTo(out);
+            toProto(platformInfo, creator).writeTo(out);
         } catch (IOException e) {
             throw new RuntimeException(e);
         }
-- 
cgit