StakeSquid/dshackle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Control max heap dumps via env var (#529)

Browse Source
This commit is contained in:
KirillPamPam
2024-07-19 18:07:44 +04:00
committed by GitHub
parent 4911c0c333
commit 366dbd3213
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/kotlin/io/emeraldpay/dshackle/HeapDumpCreator.kt
View File

@@ -16,8 +16,6 @@ import java.util.concurrent.TimeUnit
class HeapDumpCreator {
companion object {
private const val MAX_DUMPS = 3
private val log = LoggerFactory.getLogger(HeapDumpCreator::class.java)
private val executorService = Executors.newSingleThreadScheduledExecutor()
@@ -25,6 +23,7 @@ class HeapDumpCreator {
val envVars = System.getenv()
val enableCreateDumps = envVars["HEAP_DUMP_ENABLE"]?.toBoolean() ?: true
val dumpsPath = envVars["HEAP_DUMP_PATH"] ?: "/etc/dshackle/dumps"
val maxDumps = envVars["MAX_HEAP_DUMPS"]?.toInt() ?: 3
val dumpsPathExists = heapDumpsPathExists(dumpsPath)
.also {
if (!it) {
@@ -33,7 +32,7 @@ class HeapDumpCreator {
}
if (enableCreateDumps && dumpsPathExists) {
executorService.scheduleAtFixedRate({ dumpCheckTask(dumpsPath) }, 0, 30, TimeUnit.MINUTES)
executorService.scheduleAtFixedRate({ dumpCheckTask(dumpsPath, maxDumps) }, 0, 30, TimeUnit.MINUTES)
val appName = envVars["DSHACKLE_APP_NAME"] ?: "dshackle"
@@ -64,7 +63,7 @@ class HeapDumpCreator {
}
}
private fun dumpCheckTask(dumpsPath: String) {
private fun dumpCheckTask(dumpsPath: String, maxDumps: Int) {
val files = File(dumpsPath).listFiles()
?.filter { it.name.endsWith("hprof") }
?: emptyList()
@@ -72,7 +71,7 @@ class HeapDumpCreator {
if (files.isNotEmpty()) {
log.warn("There are some heap dumps, please send them to dshackle developers")
}
if (files.size >= MAX_DUMPS) {
if (files.size >= maxDumps) {
files.minBy {
Files.readAttributes(Paths.get(it.absolutePath), BasicFileAttributes::class.java).creationTime()
}.run {