diff options
Diffstat (limited to 'weed/mq/topic/local_manager.go')
| -rw-r--r-- | weed/mq/topic/local_manager.go | 29 |
1 files changed, 25 insertions, 4 deletions
diff --git a/weed/mq/topic/local_manager.go b/weed/mq/topic/local_manager.go index 0c54f2bb1..173df090d 100644 --- a/weed/mq/topic/local_manager.go +++ b/weed/mq/topic/local_manager.go @@ -23,10 +23,7 @@ func NewLocalTopicManager() *LocalTopicManager { func (manager *LocalTopicManager) AddTopicPartition(topic Topic, localPartition *LocalPartition) { localTopic, ok := manager.topics.Get(topic.String()) if !ok { - localTopic = &LocalTopic{ - Topic: topic, - Partitions: make([]*LocalPartition, 0), - } + localTopic = NewLocalTopic(topic) } if !manager.topics.SetIfAbsent(topic.String(), localTopic) { localTopic, _ = manager.topics.Get(topic.String()) @@ -59,6 +56,22 @@ func (manager *LocalTopicManager) RemoveTopicPartition(topic Topic, partition Pa return localTopic.removePartition(partition) } +func (manager *LocalTopicManager) ClosePublishers(topic Topic, unixTsNs int64) (removed bool) { + localTopic, ok := manager.topics.Get(topic.String()) + if !ok { + return false + } + return localTopic.closePartitionPublishers(unixTsNs) +} + +func (manager *LocalTopicManager) CloseSubscribers(topic Topic, unixTsNs int64) (removed bool) { + localTopic, ok := manager.topics.Get(topic.String()) + if !ok { + return false + } + return localTopic.closePartitionSubscribers(unixTsNs) +} + func (manager *LocalTopicManager) CollectStats(duration time.Duration) *mq_pb.BrokerStats { stats := &mq_pb.BrokerStats{ Stats: make(map[string]*mq_pb.TopicPartitionStats), @@ -101,3 +114,11 @@ func (manager *LocalTopicManager) CollectStats(duration time.Duration) *mq_pb.Br return stats } + +func (manager *LocalTopicManager) WaitUntilNoPublishers(topic Topic) { + localTopic, ok := manager.topics.Get(topic.String()) + if !ok { + return + } + localTopic.WaitUntilNoPublishers() +} |