MINOR Capture heap dump after OOM on CI

.github/actions/run-gradle/action.yml

@@ -86,7 +86,10 @@ runs:
       run: |
         set +e
         ./.github/scripts/thread-dump.sh &
-        timeout ${TIMEOUT_MINUTES}m ./gradlew --build-cache --continue --no-scan \
+        mkdir -p heap-dumps
+        HEAP_DUMP_DIR=$(readlink -f heap-dumps)
+        timeout ${TIMEOUT_MINUTES}m ./gradlew --continue --no-scan \
+        -Dorg.gradle.jvmargs="-Xmx4g -Xss4m -XX:+UseParallelGC -XX:+ExitOnOutOfMemoryError -XX:+HeapDumpOnOutOfMemoryError -XX:HeapDumpPath=$HEAP_DUMP_DIR" \
         -PtestLoggingEvents=started,passed,skipped,failed \
         -PmaxParallelForks=4 \
         -PmaxTestRetries=$TEST_RETRIES -PmaxTestRetryFailures=10 \
@@ -96,9 +99,11 @@ runs:
         -Pkafka.test.xml.output.dir=$TEST_XML_OUTPUT_DIR \
         -Pkafka.cluster.test.repeat=$TEST_REPEAT \
         -Pkafka.test.verbose=$TEST_VERBOSE \
+        -PheapDumpPath=$HEAP_DUMP_DIR \
         -PcommitId=xxxxxxxxxxxxxxxx \
         $TEST_TASK
         exitcode="$?"
+        find heap-dumps
         echo "exitcode=$exitcode" >> $GITHUB_OUTPUT
     - name: Archive build scan (${{ inputs.test-task }})
       if: always()

.github/workflows/build.yml

@@ -260,6 +260,17 @@ jobs:
           compression-level: 9
           if-no-files-found: ignore
 
+      - name: Archive Heap Dumps
+        id: archive-heap-dump
+        uses: actions/upload-artifact@v4
+        with:
+          name: junit-heap-dumps-${{ env.job-variation }}
+          path: |
+            heap-dumps/*
+          compression-level: 9
+          retention-days: 5
+          if-no-files-found: ignore
+
       - name: Parse JUnit tests
         env:
           GITHUB_WORKSPACE: ${{ github.workspace }}

build.gradle

@@ -53,8 +53,10 @@ ext {
 
   buildVersionFileName = "kafka-version.properties"
 
+  heapDumpPath = project.hasProperty('heapDumpPath') ? heapDumpPath : "/tmp"
+  heapDumpPathArg = "-XX:HeapDumpPath=$heapDumpPath".toString()
   defaultMaxHeapSize = "2g"
-  defaultJvmArgs = ["-Xss4m", "-XX:+UseParallelGC"]
+  defaultJvmArgs = ["-Xss4m", "-XX:+UseParallelGC", "-XX:+ExitOnOutOfMemoryError", "-XX:+HeapDumpOnOutOfMemoryError", heapDumpPathArg]
 
   // "JEP 403: Strongly Encapsulate JDK Internals" causes some tests to fail when they try
   // to access internals (often via mocking libraries). We use `--add-opens` as a workaround