Update BehaviourContextWithFSM.kt

This commit is contained in:
InsanusMokrassar 2022-12-29 08:12:02 +06:00
parent dbe2607994
commit 90ad34f114
1 changed files with 4 additions and 2 deletions

View File

@ -173,8 +173,10 @@ class DefaultBehaviourContextWithFSM<T : State>(
fun Job.enableRemoveOnCompletion(state: T) {
invokeOnCompletion {
launchSafelyWithoutExceptions {
if (this@enableRemoveOnCompletion === statesJobs[state]) {
statesJobs.remove(state)
statesJobsMutex.withLock {
if (this@enableRemoveOnCompletion === statesJobs[state]) {
statesJobs.remove(state)
}
}
}
}