acquire lock for stop()

amplitude · Dec 12, 2024 · bb784a6 · bb784a6
1 parent 03a824e
commit bb784a6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/kotlin/deployment/DeploymentRunner.kt b/src/main/kotlin/deployment/DeploymentRunner.kt
@@ -97,8 +97,10 @@ internal class DeploymentRunner(
     }
 
     fun stop() {
-        poller.shutdown()
-        flagConfigUpdater.shutdown()
+        lock.once {
+            poller.shutdown()
+            flagConfigUpdater.shutdown()
+        }
     }
 
     private fun getCohortPollingInterval(): Long {