public ControlledShutdownResponse(short errorCode, Set<TopicPartition> partitionsRemaining) {
super(new Struct(CURRENT_SCHEMA));
struct.set(ERROR_CODE_KEY_NAME, errorCode);
List<Struct> partitionsRemainingList = new ArrayList<>(partitionsRemaining.size());
for (TopicPartition topicPartition : partitionsRemaining) {
Struct topicPartitionStruct = struct.instance(PARTITIONS_REMAINING_KEY_NAME);
topicPartitionStruct.set(TOPIC_KEY_NAME, topicPartition.topic());
topicPartitionStruct.set(PARTITION_KEY_NAME, topicPartition.partition());
partitionsRemainingList.add(topicPartitionStruct);
}
struct.set(PARTITIONS_REMAINING_KEY_NAME, partitionsRemainingList.toArray());
this.errorCode = errorCode;
this.partitionsRemaining = partitionsRemaining;
}
/**
* Constructor for MetadataResponse where there are errors for some of the topics, error data take
* precedence over cluster information for particular topic
*/
public MetadataResponse(Cluster cluster, Map<String, Errors> errors) {
super(new Struct(CURRENT_SCHEMA));
List<Struct> brokerArray = new ArrayList<Struct>();
for (Node node : cluster.nodes()) {
Struct broker = struct.instance(BROKERS_KEY_NAME);
broker.set(NODE_ID_KEY_NAME, node.id());
broker.set(HOST_KEY_NAME, node.host());
broker.set(PORT_KEY_NAME, node.port());
brokerArray.add(broker);
}
struct.set(BROKERS_KEY_NAME, brokerArray.toArray());
List<Struct> topicArray = new ArrayList<Struct>();
for (Map.Entry<String, Errors> errorEntry : errors.entrySet()) {
Struct topicData = struct.instance(TOPIC_METATDATA_KEY_NAME);
topicData.set(TOPIC_KEY_NAME, errorEntry.getKey());
topicData.set(TOPIC_ERROR_CODE_KEY_NAME, errorEntry.getValue().code());
topicData.set(PARTITION_METADATA_KEY_NAME, new Struct[0]);
topicArray.add(topicData);
}
for (String topic : cluster.topics()) {
if (!errors.containsKey(topic)) {
Struct topicData = struct.instance(TOPIC_METATDATA_KEY_NAME);
topicData.set(TOPIC_KEY_NAME, topic);
topicData.set(TOPIC_ERROR_CODE_KEY_NAME, Errors.NONE.code());
List<Struct> partitionArray = new ArrayList<Struct>();
for (PartitionInfo fetchPartitionData : cluster.partitionsForTopic(topic)) {
Struct partitionData = topicData.instance(PARTITION_METADATA_KEY_NAME);
partitionData.set(PARTITION_ERROR_CODE_KEY_NAME, Errors.NONE.code());
partitionData.set(PARTITION_KEY_NAME, fetchPartitionData.partition());
partitionData.set(LEADER_KEY_NAME, fetchPartitionData.leader().id());
ArrayList<Integer> replicas = new ArrayList<Integer>();
for (Node node : fetchPartitionData.replicas()) replicas.add(node.id());
partitionData.set(REPLICAS_KEY_NAME, replicas.toArray());
ArrayList<Integer> isr = new ArrayList<Integer>();
for (Node node : fetchPartitionData.inSyncReplicas()) isr.add(node.id());
partitionData.set(ISR_KEY_NAME, isr.toArray());
partitionArray.add(partitionData);
}
topicData.set(PARTITION_METADATA_KEY_NAME, partitionArray.toArray());
topicArray.add(topicData);
}
}
struct.set(TOPIC_METATDATA_KEY_NAME, topicArray.toArray());
this.cluster = cluster;
this.errors = errors;
}