Cluster cluster2 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic2, 0), t2p0)); Cluster cluster3 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic2, 2), t0p2)); Cluster cluster4 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic0, 0), t0p0DifferentOrder)); Cluster cluster5 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic0, 0), t0p0DifferentAssignment)); Cluster cluster6 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic0, 0), t0p0DifferentLeader)); Cluster cluster7 = cluster1.withPartitions(Collections.singletonMap(new TopicPartition(topic0, 0), t0p0DifferentIsr));
public void addTopic(String topic, int partitions) { Map<TopicPartition, PartitionInfo> partitionInfo = new HashMap<>(); List<Node> nodes = cluster.nodes(); for (int i = 0; i < partitions; i++) { Node node = nodes.get(i % nodes.size()); Node[] replicas = new Node[]{node}; TopicPartition topicPartition = new TopicPartition(topic, i); partitionInfo.put(topicPartition, new PartitionInfo(topic, i, node, replicas, replicas)); logs.put(topicPartition, new ArrayList<>()); uncommittedMessages.put(topicPartition, new ArrayList<>()); } cluster = cluster.withPartitions(partitionInfo); }
final Cluster fullMetadata = metadata.withPartitions(allRepartitionTopicPartitions); taskManager.setClusterMetadata(fullMetadata);
taskManager.setClusterMetadata(Cluster.empty().withPartitions(topicToPartitionInfo)); taskManager.setPartitionsByHostState(partitionsByHost); taskManager.setAssignmentMetadata(activeTasks, info.standbyTasks());