How to use

AdminZkClient

in

kafka.zk

/**
 * Override to set up your specific external resource.
 *
 * @throws Throwable if setup fails (which will disable {@code after}
 */
@Override protected void before() throws Throwable {
 // Start the ZK and the Broker
 LOG.info("init embedded Zookeeper");
 zkServer = new EmbeddedZookeeper();
 tmpLogDir = Files.createTempDirectory("kafka-log-dir-").toAbsolutePath();
 String zkConnect = "127.0.0.1:" + zkServer.port();
 LOG.info("init kafka broker");
 Properties brokerProps = new Properties();
 brokerProps.setProperty("zookeeper.connect", zkConnect);
 brokerProps.setProperty("broker.id", "0");
 brokerProps.setProperty("log.dir", tmpLogDir.toString());
 brokerProps.setProperty("listeners", "PLAINTEXT://" + BROKER_IP_PORT);
 brokerProps.setProperty("offsets.topic.replication.factor", "1");
 brokerProps.setProperty("transaction.state.log.replication.factor", "1");
 brokerProps.setProperty("transaction.state.log.min.isr", "1");
 KafkaConfig config = new KafkaConfig(brokerProps);
 kafkaServer = TestUtils.createServer(config, Time.SYSTEM);
 kafkaServer.startup();
 kafkaServer.zkClient();
 adminZkClient = new AdminZkClient(kafkaServer.zkClient());
 LOG.info("Creating kafka TOPIC [{}]", TOPIC);
 adminZkClient.createTopic(TOPIC, 1, 1, new Properties(), RackAwareMode.Disabled$.MODULE$);
}

 void deleteTopic(@SuppressWarnings("SameParameterValue") String topic) {
  adminZkClient.deleteTopic(topic);
 }
}

void maybeAddPartitions(int minPartitionNum) {
 KafkaZkClient zkClient = KafkaZkClient.apply(_zkConnect, JaasUtils.isZkSecurityEnabled(), ZK_SESSION_TIMEOUT_MS,
   ZK_CONNECTION_TIMEOUT_MS, Integer.MAX_VALUE, Time.SYSTEM, METRIC_GROUP_NAME, "SessionExpireListener");
 AdminZkClient adminZkClient = new AdminZkClient(zkClient);
 try {
  scala.collection.Map<Object, scala.collection.Seq<Object>> existingAssignment = getPartitionAssignment(zkClient, _topic);
  int partitionNum = existingAssignment.size();
  if (partitionNum < minPartitionNum) {
   LOG.info("MultiClusterTopicManagementService will increase partition of the topic {} "
     + "in cluster {} from {} to {}.", _topic, _zkConnect, partitionNum, minPartitionNum);
   scala.Option<scala.collection.Map<java.lang.Object, scala.collection.Seq<java.lang.Object>>> replicaAssignment = scala.Option.apply(null);
   scala.Option<Seq<Object>> brokerList = scala.Option.apply(null);
   adminZkClient.addPartitions(_topic, existingAssignment, adminZkClient.getBrokerMetadatas(RackAwareMode.Disabled$.MODULE$, brokerList), minPartitionNum, replicaAssignment, false);
  }
 } finally {
  zkClient.close();
 }
}

  server.startup();
  LOGGER.info("Started Kafka server {} at {} with storage in {}", brokerId, getConnection(), logsDir.getAbsolutePath());
  adminZkClient = new AdminZkClient(server.zkClient());
  return this;
} catch (RuntimeException e) {

/**
 * Create the specified topic.
 * 
 * @param topic the name of the topic to create
 * @param numPartitions the number of partitions for the topic
 * @param replicationFactor the replication factor for the topic
 */
public void createTopic( String topic, int numPartitions, int replicationFactor ) {
  RackAwareMode rackAwareMode = null;
  getAdminZkClient().createTopic(topic, numPartitions, replicationFactor, new Properties(), rackAwareMode);
}

@Override
public void updateTopic(Topic topic) {
  Properties config = createTopicConfig(topic.getRetentionTime().getDuration(), topicProperties, topic);
  KafkaTopics kafkaTopics = kafkaNamesMapper.toKafkaTopics(topic);
  if (isMigrationToNewKafkaTopic(kafkaTopics)) {
    adminZkClient.createTopic(
        kafkaTopics.getPrimary().name().asString(),
        topicProperties.getPartitions(),
        topicProperties.getReplicationFactor(),
        config,
        kafka.admin.RackAwareMode.Enforced$.MODULE$
    );
  } else {
    adminZkClient.changeTopicConfig(kafkaTopics.getPrimary().name().asString(), config);
  }
  kafkaTopics.getSecondary().ifPresent(secondary ->
      adminZkClient.changeTopicConfig(secondary.name().asString(), config)
  );
}

private AdminZkClient adminZkClient(KafkaZkClient kafkaZkClient) {
  return new AdminZkClient(kafkaZkClient);
}

@Override
public void createTopic(Topic topic) {
  Properties config = createTopicConfig(topic.getRetentionTime().getDuration(), topicProperties, topic);
  kafkaNamesMapper.toKafkaTopics(topic).forEach(k ->
      adminZkClient.createTopic(
          k.name().asString(),
          topicProperties.getPartitions(),
          topicProperties.getReplicationFactor(),
          config,
          kafka.admin.RackAwareMode.Enforced$.MODULE$
      )
  );
}

private void createTopic(String topicName, KafkaZkClient kafkaZkClient) {
  Topic topic = topic(topicName).build();
  kafkaNamesMapper.toKafkaTopics(topic).forEach(kafkaTopic -> {
    AdminZkClient adminZkClient = new AdminZkClient(kafkaZkClient);
    adminZkClient.createTopic(kafkaTopic.name().asString(), DEFAULT_PARTITIONS, DEFAULT_REPLICATION_FACTOR, new Properties(), RackAwareMode.Enforced$.MODULE$);
    waitAtMost(adjust(Duration.ONE_MINUTE)).until(() -> {
          kafkaZkClient.topicExists(kafkaTopic.name().asString());
        }
    );
  });
}

@Override
public void removeTopic(Topic topic) {
  kafkaNamesMapper.toKafkaTopics(topic).forEach(k -> adminZkClient.deleteTopic(k.name().asString()));
}

private void createTopic(String topicName, KafkaZkClient kafkaZkClient) {
  Topic topic = topic(topicName).build();
  kafkaNamesMapper.toKafkaTopics(topic).forEach(kafkaTopic -> {
    AdminZkClient adminZkClient = new AdminZkClient(kafkaZkClient);
    adminZkClient.createTopic(kafkaTopic.name().asString(), DEFAULT_PARTITIONS, DEFAULT_REPLICATION_FACTOR, new Properties(), RackAwareMode.Enforced$.MODULE$);
    waitAtMost(adjust(Duration.ONE_MINUTE)).until(() -> {
          kafkaZkClient.topicExists(kafkaTopic.name().asString());
        }
    );
  });
}