tursom/TursomServer
0
0
Fork 0
mirror of https://github.com/tursom/TursomServer.git synced 2025-02-08 18:50:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

update ShutdownHook

Browse Source
This commit is contained in:
tursom 2021-07-13 10:38:27 +08:00
parent 84029c8613
commit bbed157b4a
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ts-core/src/main/kotlin/cn/tursom/core/ShutdownHook.kt
View File

@ -16,6 +16,19 @@ object ShutdownHook {
fun get(): (() -> Unit)?
}
internal class SoftHookReference(
hook: () -> Unit,
) : HookReference {
private val ref = SoftReference(hook)
override fun get(): (() -> Unit)? = ref.get()
}
internal class HardHookReference(
private val hook: () -> Unit,
) : HookReference {
override fun get(): () -> Unit = hook
}
class Hook internal constructor(
private val hook: () -> Unit,
private val reference: HookReference,
@ -44,12 +57,7 @@ object ShutdownHook {
addWorkThread()
}
val reference = if (softReference) object : HookReference {
private val ref = SoftReference(hook)
override fun get(): (() -> Unit)? = ref.get()
} else object : HookReference {
override fun get(): () -> Unit = hook
}
val reference = if (softReference) SoftHookReference(hook) else HardHookReference(hook)
shutdownHooks.add(reference)
return Hook(hook, reference)