From d5cb2150839a5c59a186df2446f8a396e594d3e8 Mon Sep 17 00:00:00 2001
From: Levente Pap <levente.pap@iota.org>
Date: Sun, 12 Jul 2020 22:23:50 +0200
Subject: [PATCH] Get message requester queue size
---
.../messagelayer/messagerequester/messagerequester.go | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/packages/binary/messagelayer/messagerequester/messagerequester.go b/packages/binary/messagelayer/messagerequester/messagerequester.go
index 06845164..caad920c 100644
--- a/packages/binary/messagelayer/messagerequester/messagerequester.go
+++ b/packages/binary/messagelayer/messagerequester/messagerequester.go
@@ -69,3 +69,10 @@ func (requester *MessageRequester) reRequest(id message.Id) {
requester.scheduledRequests[id] = time.AfterFunc(requester.options.retryInterval, func() { requester.reRequest(id) })
}
}
+
+// RequestQueueSize returns the number of scheduled message requests.
+func (requester *MessageRequester) RequestQueueSize() int {
+ requester.scheduledRequestsMutex.RLock()
+ defer requester.scheduledRequestsMutex.RUnlock()
+ return len(requester.scheduledRequests)
+}
--
GitLab