Reduce waiting time for head restart (#483)

2024-05-21 21:08:23 +07:00
parent 1a825667b6
commit 8f8d89ae74
1 changed files with 1 additions and 1 deletions
--- 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 {