MINOR: Prevent java.lang.UnsupportedOperationException in MockAdminClient (#14955)

Reviewers: Divij Vaidya <diviv@amazon.com>, Sagar Rao <sagarmeansocean@gmail.com>
This commit is contained in:
James Faulkner 2024-01-04 21:15:51 +00:00 committed by GitHub
parent c703ce2563
commit 29108b6701
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions

View File

@ -1017,7 +1017,13 @@ public class MockAdminClient extends AdminClient {
for (Node node : nodes) {
Map<String, LogDirDescription> logDirDescriptionMap = unwrappedResults.get(node.id());
LogDirDescription logDirDescription = logDirDescriptionMap.getOrDefault(partitionLogDirs.get(0), new LogDirDescription(null, new HashMap<>()));
logDirDescription.replicaInfos().put(new TopicPartition(topicName, topicPartitionInfo.partition()), new ReplicaInfo(0, 0, false));
Map<TopicPartition, ReplicaInfo> topicPartitionReplicaInfoMap = new HashMap<>(logDirDescription.replicaInfos());
topicPartitionReplicaInfoMap.put(new TopicPartition(topicName, topicPartitionInfo.partition()), new ReplicaInfo(0, 0, false));
logDirDescriptionMap.put(partitionLogDirs.get(0), new LogDirDescription(
logDirDescription.error(),
topicPartitionReplicaInfoMap,
logDirDescription.totalBytes().orElse(DescribeLogDirsResponse.UNKNOWN_VOLUME_BYTES),
logDirDescription.usableBytes().orElse(DescribeLogDirsResponse.UNKNOWN_VOLUME_BYTES)));
}
}
}