瀏覽代碼

Change type of unvisitedPartitions from []string to bool to economize

Scott Kidder 6 年之前
父節點
當前提交
b29086bdaa
共有 1 個文件被更改,包括 2 次插入2 次删除
  1. 2 2
      balance_strategy.go

+ 2 - 2
balance_strategy.go

@@ -215,9 +215,9 @@ func (s *stickyBalanceStrategy) Plan(members map[string]ConsumerGroupMemberMetad
 
 
 	// create a mapping of each partition to its current consumer, where possible
 	// create a mapping of each partition to its current consumer, where possible
 	currentPartitionConsumers := make(map[topicPartitionAssignment]string, len(currentAssignment))
 	currentPartitionConsumers := make(map[topicPartitionAssignment]string, len(currentAssignment))
-	unvisitedPartitions := make(map[topicPartitionAssignment][]string, len(partition2AllPotentialConsumers))
+	unvisitedPartitions := make(map[topicPartitionAssignment]bool, len(partition2AllPotentialConsumers))
 	for partition := range partition2AllPotentialConsumers {
 	for partition := range partition2AllPotentialConsumers {
-		unvisitedPartitions[partition] = []string{}
+		unvisitedPartitions[partition] = true
 	}
 	}
 	var unassignedPartitions []topicPartitionAssignment
 	var unassignedPartitions []topicPartitionAssignment
 	for memberID, partitions := range currentAssignment {
 	for memberID, partitions := range currentAssignment {