From 73dd214ae66bab483ee8b4f0ed03881466da92e8 Mon Sep 17 00:00:00 2001
From: Luck <git@lucko.me>
Date: Tue, 12 Jul 2022 22:37:59 +0100
Subject: Improve/fix game thread dumper

---
 .../main/java/me/lucko/spark/common/sampler/ThreadDumper.java  | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

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

diff --git a/spark-common/src/main/java/me/lucko/spark/common/sampler/ThreadDumper.java b/spark-common/src/main/java/me/lucko/spark/common/sampler/ThreadDumper.java
index 9d54f50..fe3a6a7 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/sampler/ThreadDumper.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/sampler/ThreadDumper.java
@@ -83,10 +83,8 @@ public interface ThreadDumper {
             return Objects.requireNonNull(this.dumper, "dumper");
         }
 
-        public void ensureSetup() {
-            if (this.dumper == null) {
-                this.dumper = new Specific(new long[]{Thread.currentThread().getId()});
-            }
+        public void setThread(Thread thread) {
+            this.dumper = new Specific(new long[]{thread.getId()});
         }
     }
 
@@ -98,6 +96,10 @@ public interface ThreadDumper {
         private Set<Thread> threads;
         private Set<String> threadNamesLowerCase;
 
+        public Specific(Thread thread) {
+            this.ids = new long[]{thread.getId()};
+        }
+
         public Specific(long[] ids) {
             this.ids = ids;
         }
-- 
cgit