add size()

2 files changed, 14 insertions, 0 deletions

diff --git a/weed/mq/topic/local_partition_publishers.go b/weed/mq/topic/local_partition_publishers.go
index 367ccce5f..c12f66336 100644
--- a/weed/mq/topic/local_partition_publishers.go
+++ b/weed/mq/topic/local_partition_publishers.go
@@ -50,3 +50,10 @@ func (p *LocalPartitionPublishers) IsEmpty() bool {
 
 	return len(p.publishers) == 0
 }
+
+func (p *LocalPartitionPublishers) Size() int {
+	p.publishersLock.RLock()
+	defer p.publishersLock.RUnlock()
+
+	return len(p.publishers)
+}
diff --git a/weed/mq/topic/local_partition_subscribers.go b/weed/mq/topic/local_partition_subscribers.go
index caadff278..d3b989d72 100644
--- a/weed/mq/topic/local_partition_subscribers.go
+++ b/weed/mq/topic/local_partition_subscribers.go
@@ -54,3 +54,10 @@ func (p *LocalPartitionSubscribers) IsEmpty() bool {
 
 	return len(p.Subscribers) == 0
 }
+
+func (p *LocalPartitionSubscribers) Size() int {
+	p.SubscribersLock.RLock()
+	defer p.SubscribersLock.RUnlock()
+
+	return len(p.Subscribers)
+}