From 1efbba29f3764fc759fdac4a77b65be460205ea4 Mon Sep 17 00:00:00 2001
From: a10zn8 <a1.cyb3r@gmail.com>
Date: Tue, 17 Jan 2023 20:13:25 +0400
Subject: [PATCH] set lag = 0 in case single upstream in multisream

---
 .../kotlin/io/emeraldpay/dshackle/upstream/Multistream.kt    | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/kotlin/io/emeraldpay/dshackle/upstream/Multistream.kt b/src/main/kotlin/io/emeraldpay/dshackle/upstream/Multistream.kt
index 8ad24d15..0233583b 100644
--- a/src/main/kotlin/io/emeraldpay/dshackle/upstream/Multistream.kt
+++ b/src/main/kotlin/io/emeraldpay/dshackle/upstream/Multistream.kt
@@ -198,8 +198,9 @@ abstract class Multistream(
             }
             lagObserver?.stop()
             lagObserver = null
-            if (upstreams.size > 1) {
-                lagObserver = makeLagObserver()
+            when {
+                upstreams.size == 1 -> upstreams[0].setLag(0)
+                upstreams.size > 1 -> lagObserver = makeLagObserver()
             }
         }
     }