Atomically set subscription

drpcorg · Sep 25, 2024 · 1c34e7c · 1c34e7c
1 parent 8b5e3bd
commit 1c34e7c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/main/kotlin/io/emeraldpay/dshackle/upstream/ethereum/GenericWsHead.kt b/src/main/kotlin/io/emeraldpay/dshackle/upstream/ethereum/GenericWsHead.kt
@@ -91,9 +91,7 @@ class GenericWsHead(
         )
         this.subscription.set(super.follow(heads))
 
-        if (headResubSubscription.get() == null) {
-            headResubSubscription.set(registerHeadResubscribeFlux())
-        }
+        headResubSubscription.compareAndSet(null, registerHeadResubscribeFlux())
     }
 
     override fun onNoHeadUpdates() {
@@ -182,6 +180,7 @@ class GenericWsHead(
     }
 
     private fun registerHeadResubscribeFlux(): Disposable {
+        println("here")
         val connectionStates = wsSubscriptions.connectionInfoFlux()
             .map {
                 if (it.connectionId == connectionId.get() && it.connectionState == WsConnection.ConnectionState.DISCONNECTED) {