Set activity flag in WS (#721)

vacp2p · May 24, 2022 · 1696d0c · 1696d0c
1 parent d4ff1c8
commit 1696d0c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libp2p/transports/wstransport.nim b/libp2p/transports/wstransport.nim
@@ -71,12 +71,14 @@ method readOnce*(
 
   if res == 0 and s.session.readyState == ReadyState.Closed:
     raise newLPStreamEOFError()
+  s.activity = true # reset activity flag
   return res
 
 method write*(
   s: WsStream,
   msg: seq[byte]): Future[void] {.async.} =
   mapExceptions(await s.session.send(msg, Opcode.Binary))
+  s.activity = true # reset activity flag
 
 method closeImpl*(s: WsStream): Future[void] {.async.} =
   await s.session.close()