logos-messaging/logos-messaging-nim
1
0
Fork 0
mirror of https://github.com/logos-messaging/logos-messaging-nim.git synced 2026-05-11 21:09:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

only allow furter limits for store live streams

Browse Source
This commit is contained in:
Ivan FB 2026-05-04 13:11:15 +02:00
parent f1dcfefcac
commit fb5c5d3824
No known key found for this signature in database
GPG Key ID: DF0C67A04C543270
1 changed files with 5 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
waku/node/peer_manager/peer_manager.nim
View File

@ -404,25 +404,13 @@ proc disconnectNode*(pm: PeerManager, peer: RemotePeerInfo) {.async.} =
let peerId = peer.peerId
await pm.disconnectNode(peerId)
proc activeServiceStreams(pm: PeerManager, peerId: PeerId): seq[string] =
## Returns the protocol strings of any open service streams for peerId.
## Relay streams are excluded: they are exactly what eviction manages.
for connPeerId, muxers in pm.switch.connManager.getConnections():
if connPeerId != peerId:
continue
for m in muxers:
for stream in m.getStreams():
if stream.protocol.len > 0 and stream.protocol != WakuRelayCodec:
result.add(stream.protocol)
proc evictPeer(pm: PeerManager, peerId: PeerId) {.async.} =
## Policy-based eviction (relay-peer limit, IP colocation, pruning).
## Skips the disconnect when the peer has an in-flight stream for any
## service protocol to avoid aborting active requests.
let serviceStreams = pm.activeServiceStreams(peerId)
if serviceStreams.len > 0:
trace "skipping peer eviction: active service streams",
peerId = peerId, protocols = serviceStreams
## Skips the disconnect when the peer has an in-flight store stream to
## avoid aborting active store requests.
let (storeInPeers, storeOutPeers) = pm.connectedPeers(WakuStoreCodec)
if storeInPeers.contains(peerId) or storeOutPeers.contains(peerId):
trace "skipping peer eviction: active store stream", peerId = peerId
return
await pm.switch.disconnect(peerId)