|
Steps To Reproduce: |
public class PartitionFinder {
public static final String TM_TRANSPORT_PLAN_INTERNAL = "tm.transportPlan.internal";
public static final String KEY = "\"1989e567-b6d1-fc12-910c-f027bc56d7f0\"";
private static Logger logger = LoggerFactory.getLogger(PartitionFinder.class);
public static void main(String[] args) throws Exception {
String bootstrapServers = "widmpp01.lkw-walter.com:9092";
Properties config = new Properties();
config.put(AdminClientConfig.BOOTSTRAP_SERVERS_CONFIG, bootstrapServers);
AdminClient adminClient = AdminClient.create(config);
DescribeClusterResult describeClusterResult = adminClient.describeCluster();
System.out.println(describeClusterResult.clusterId().get());
System.out.println(describeClusterResult.nodes().get());
List<TopicPartitionInfo> topicPartitionInfo =
adminClient.describeTopics(List.of(TM_TRANSPORT_PLAN_INTERNAL)).values().get(TM_TRANSPORT_PLAN_INTERNAL).get().partitions();
List<PartitionInfo> partitionInfos = topicPartitionInfo.stream().map(new Function<TopicPartitionInfo, PartitionInfo>() {
@Override
public PartitionInfo apply(TopicPartitionInfo topicPartitionInfo) {
PartitionInfo partitionInfo = new PartitionInfo(
TM_TRANSPORT_PLAN_INTERNAL,topicPartitionInfo.partition(),topicPartitionInfo.leader(),topicPartitionInfo.replicas().toArray(new Node[]{}),topicPartitionInfo.isr().toArray(new Node[]{})
);
return partitionInfo;
}
@Override
public <V> Function<V, PartitionInfo> compose(Function<? super V, ? extends TopicPartitionInfo> before) {
return null;
}
@Override
public <V> Function<TopicPartitionInfo, V> andThen(Function<? super PartitionInfo, ? extends V> after) {
return null;
}
}).collect(Collectors.toList());
System.out.println(partitionInfos.toString());
Cluster cluster = new Cluster(describeClusterResult.clusterId().get(),describeClusterResult.nodes().get(),partitionInfos,new HashSet<>(),new HashSet<>(),null);
Partitioner partitioner = new DefaultPartitioner();
System.out.println(cluster.toString() + " : " + cluster.partitionCountForTopic(TM_TRANSPORT_PLAN_INTERNAL));
int partition = partitioner.partition(TM_TRANSPORT_PLAN_INTERNAL, KEY, KEY.getBytes(),null,null,cluster);
System.out.println(partition);
System.exit(0);
}
} |