From badc14d8b4018fdae91a6af5a7181406eb9c6f20 Mon Sep 17 00:00:00 2001
From: Tolya Korniltsev <korniltsev.anatoly@gmail.com>
Date: Tue, 10 May 2022 20:13:44 +0300
Subject: [PATCH] Add 'contextId' parsing for async-profiler

Add support for parsing new fields from https://github.com/jvm-profiling-tools/async-profiler/pull/576/files
---
 parser/event_types.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/parser/event_types.go b/parser/event_types.go
index 0f70d34..d71f0b3 100644
--- a/parser/event_types.go
+++ b/parser/event_types.go
@@ -622,6 +622,7 @@ type ExecutionSample struct {
 	SampledThread *Thread
 	StackTrace    *StackTrace
 	State         *ThreadState
+	ContextId     int64
 }
 
 func (es *ExecutionSample) parseField(name string, p ParseResolvable) (err error) {
@@ -634,6 +635,8 @@ func (es *ExecutionSample) parseField(name string, p ParseResolvable) (err error
 		es.StackTrace, err = toStackTrace(p)
 	case "state":
 		es.State, err = toThreadState(p)
+	case "contextId":
+		es.ContextId, err = toLong(p)
 	}
 	return err
 }
@@ -846,6 +849,7 @@ type JavaMonitorEnter struct {
 	MonitorClass  *Class
 	PreviousOwner *Thread
 	Address       int64
+	ContextId     int64
 }
 
 func (jme *JavaMonitorEnter) parseField(name string, p ParseResolvable) (err error) {
@@ -864,6 +868,8 @@ func (jme *JavaMonitorEnter) parseField(name string, p ParseResolvable) (err err
 		jme.PreviousOwner, err = toThread(p)
 	case "address":
 		jme.Address, err = toLong(p)
+	case "contextId":
+		jme.ContextId, err = toLong(p)
 	}
 	return err
 }
@@ -1146,6 +1152,7 @@ type ObjectAllocationInNewTLAB struct {
 	ObjectClass    *Class
 	AllocationSize int64
 	TLABSize       int64
+	ContextId      int64
 }
 
 func (oa *ObjectAllocationInNewTLAB) parseField(name string, p ParseResolvable) (err error) {
@@ -1162,7 +1169,10 @@ func (oa *ObjectAllocationInNewTLAB) parseField(name string, p ParseResolvable)
 		oa.AllocationSize, err = toLong(p)
 	case "tlabSize":
 		oa.TLABSize, err = toLong(p)
+	case "contextId":
+		oa.ContextId, err = toLong(p)
 	}
+
 	return err
 }
 
@@ -1176,6 +1186,7 @@ type ObjectAllocationOutsideTLAB struct {
 	StackTrace     *StackTrace
 	ObjectClass    *Class
 	AllocationSize int64
+	ContextId      int64
 }
 
 func (oa *ObjectAllocationOutsideTLAB) parseField(name string, p ParseResolvable) (err error) {
@@ -1190,6 +1201,8 @@ func (oa *ObjectAllocationOutsideTLAB) parseField(name string, p ParseResolvable
 		oa.ObjectClass, err = toClass(p)
 	case "allocationSize":
 		oa.AllocationSize, err = toLong(p)
+	case "contextId":
+		oa.ContextId, err = toLong(p)
 	}
 	return err
 }
@@ -1552,6 +1565,7 @@ type ThreadPark struct {
 	Timeout     int64
 	Until       int64
 	Address     int64
+	ContextId   int64
 }
 
 func (tp *ThreadPark) parseField(name string, p ParseResolvable) (err error) {
@@ -1572,6 +1586,8 @@ func (tp *ThreadPark) parseField(name string, p ParseResolvable) (err error) {
 		tp.Until, err = toLong(p)
 	case "address":
 		tp.Address, err = toLong(p)
+	case "contextId": // todo this one seems to be unimplemented in the profiler yet
+		tp.ContextId, err = toLong(p)
 	}
 	return err
 }