From 316464fc71e75c146f09f55e8c739c8a0fc02f51 Mon Sep 17 00:00:00 2001 From: Csaba Kiraly Date: Wed, 17 May 2023 08:08:12 +0200 Subject: [PATCH] dht: waitMessage: expose timeout as parameter, keeping default defults to ResponseTimeout as before Signed-off-by: Csaba Kiraly --- codexdht/private/eth/p2p/discoveryv5/protocol.nim | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codexdht/private/eth/p2p/discoveryv5/protocol.nim b/codexdht/private/eth/p2p/discoveryv5/protocol.nim index ab6b4b6..36e8de4 100644 --- a/codexdht/private/eth/p2p/discoveryv5/protocol.nim +++ b/codexdht/private/eth/p2p/discoveryv5/protocol.nim @@ -480,12 +480,12 @@ proc waitResponse*[T: SomeMessage](d: Protocol, node: Node, msg: T): result = d.waitMessage(node, reqId) sendRequest(d, node, msg, reqId) -proc waitMessage(d: Protocol, fromNode: Node, reqId: RequestId): +proc waitMessage(d: Protocol, fromNode: Node, reqId: RequestId, timeout = ResponseTimeout): Future[Option[Message]] = result = newFuture[Option[Message]]("waitMessage") let res = result let key = (fromNode.id, reqId) - sleepAsync(ResponseTimeout).addCallback() do(data: pointer): + sleepAsync(timeout).addCallback() do(data: pointer): d.awaitedMessages.del(key) if not res.finished: res.complete(none(Message))