Reduce waiting time for head restart (#483)

emeraldpay · May 21, 2024 · 8f8d89a · 8f8d89a
1 parent 1a82566
commit 8f8d89a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/kotlin/io/emeraldpay/dshackle/upstream/AbstractHead.kt b/src/main/kotlin/io/emeraldpay/dshackle/upstream/AbstractHead.kt
@@ -87,7 +87,7 @@ abstract class AbstractHead @JvmOverloads constructor(
                 // close internal stream if upstream is finished, otherwise it gets stuck,
                 // but technically it should never happen during normal work, only when the Head
                 // is stopping
-                if (it == SignalType.ON_ERROR && !stopping) {
+                if ((it == SignalType.ON_ERROR || it == SignalType.CANCEL || it == SignalType.ON_COMPLETE) && !stopping) {
                     log.warn("Received signal $upstreamId $it unexpectedly - restart head")
                     lastHeadUpdated = 0L
                 } else {