refactor

weed/mq/topic/local_manager.go

@@ -28,10 +28,7 @@ func (manager *LocalTopicManager) AddLocalPartition(topic Topic, localPartition
 	if !manager.topics.SetIfAbsent(topic.String(), localTopic) {
 		localTopic, _ = manager.topics.Get(topic.String())
 	}
-	if localTopic.findPartition(localPartition.Partition) != nil {
-		return
-	}
-	localTopic.Partitions = append(localTopic.Partitions, localPartition)
+	localTopic.AddPartition(localPartition)
 }
 
 // GetLocalPartition gets a topic from the local topic manager

weed/mq/topic/local_topic.go

@@ -37,6 +37,12 @@ func (localTopic *LocalTopic) removePartition(partition Partition) bool {
 	localTopic.Partitions = append(localTopic.Partitions[:foundPartitionIndex], localTopic.Partitions[foundPartitionIndex+1:]...)
 	return true
 }
+func (localTopic *LocalTopic) AddPartition(localPartition *LocalPartition) {
+	if localTopic.findPartition(localPartition.Partition) != nil {
+		return
+	}
+	localTopic.Partitions = append(localTopic.Partitions, localPartition)
+}
 
 func (localTopic *LocalTopic) closePartitionPublishers(unixTsNs int64) bool {
 	var wg sync.WaitGroup