Updates to print thread dump only if appConfig enabled

vega113 · Nov 18, 2023 · 2351296 · 2351296
1 parent 793fcd5
commit 2351296
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/app/ShutdownHook.scala b/app/ShutdownHook.scala
@@ -1,15 +1,15 @@
 import org.slf4j.Logger
+import play.api.Configuration
 
 import scala.concurrent.Future
 import play.api.inject.ApplicationLifecycle
 
 import javax.inject._
 
-class ShutdownHook @Inject()(lifecycle: ApplicationLifecycle) {
+class ShutdownHook @Inject()(lifecycle: ApplicationLifecycle, config: Configuration) {
   val logger: Logger = org.slf4j.LoggerFactory.getLogger("ShutdownHook")
 
-  lifecycle.addStopHook { () =>
-    logger.info("ShutdownHook is running")
+  private def threadDump(): Unit = {
     val threadMXBean = java.lang.management.ManagementFactory.getThreadMXBean
     val threadInfos = threadMXBean.getThreadInfo(threadMXBean.getAllThreadIds, 100)
     val stackTrace = threadInfos.map(threadInfo => {
@@ -19,6 +19,13 @@ class ShutdownHook @Inject()(lifecycle: ApplicationLifecycle) {
       """
     }).mkString("\n")
     logger.info(s"Thread dump:\n$stackTrace")
+  }
+
+  lifecycle.addStopHook { () =>
+    logger.info("ShutdownHook is running")
+    if (config.get[Boolean]("emo.threadDumpOnShutdown")) {
+      threadDump()
+    }
     Future.successful(())
   }
 }
diff --git a/conf/application.conf b/conf/application.conf
@@ -32,6 +32,8 @@ play {
 emo.config {
     loginTokenExpirationTime = 365 days
     loginTokenExpirationTime = ${?LOGIN_TOKEN_EXPIRATION_TIME}
+    threadDumpOnShutdown = false
+    threadDumpOnShutdown = ${?THREAD_DUMP_ON_SHUTDOWN}
 }
 
 database.dispatcher {