Refactor simplipy partitionSubset logic

Signed-off-by: Su Ko <[email protected]>

Author: bandalgomsu

spring-kafka/src/main/java/org/springframework/kafka/listener/ConcurrentMessageListenerContainer.java

@@ -316,30 +316,28 @@ private KafkaMessageListenerContainer<K, V> constructContainer(ContainerProperti
 	}
 
 	private @Nullable TopicPartitionOffset @Nullable [] partitionSubset(ContainerProperties containerProperties, int index) {
-		@Nullable TopicPartitionOffset @Nullable [] topicPartitions = containerProperties.getTopicPartitions();
+		TopicPartitionOffset[] topicPartitions = containerProperties.getTopicPartitions();
 		if (topicPartitions == null) {
 			return null;
 		}
+
 		if (this.concurrency == 1) {
 			return topicPartitions;
 		}
-		else {
-			int numPartitions = topicPartitions.length;
-			if (numPartitions == this.concurrency) {
-				return new TopicPartitionOffset[] { topicPartitions[index] };
-			}
-			else {
-				int perContainer = numPartitions / this.concurrency;
-				TopicPartitionOffset[] subset;
-				if (index == this.concurrency - 1) {
-					subset = Arrays.copyOfRange(topicPartitions, index * perContainer, topicPartitions.length);
-				}
-				else {
-					subset = Arrays.copyOfRange(topicPartitions, index * perContainer, (index + 1) * perContainer);
-				}
-				return subset;
-			}
+
+		int numPartitions = topicPartitions.length;
+
+		if (numPartitions == this.concurrency) {
+			return new TopicPartitionOffset[] { topicPartitions[index] };
 		}
+
+		int perContainer = numPartitions / this.concurrency;
+		int start = index * perContainer;
+		int end = (index == this.concurrency - 1)
+				? numPartitions
+				: start + perContainer;
+
+		return Arrays.copyOfRange(topicPartitions, start, end);
 	}
 
 	/*