Examples with ZkMarshallingError org.I0Itec.zkclient.exception.ZkMarshallingError used on opensource projects

Example 1 with ZkMarshallingError

use of org.I0Itec.zkclient.exception.ZkMarshallingError in project nifi by apache.

the class KafkaUtils method retrievePartitionCountForTopic.

/**
 * Will retrieve the amount of partitions for a given Kafka topic.
 */
static int retrievePartitionCountForTopic(String zookeeperConnectionString, String topicName) {
    ZkClient zkClient = null;
    try {
        zkClient = new ZkClient(zookeeperConnectionString);
        zkClient.setZkSerializer(new ZkSerializer() {

            @Override
            public byte[] serialize(Object o) throws ZkMarshallingError {
                return ZKStringSerializer.serialize(o);
            }

            @Override
            public Object deserialize(byte[] bytes) throws ZkMarshallingError {
                return ZKStringSerializer.deserialize(bytes);
            }
        });
        scala.collection.Set<TopicMetadata> topicMetadatas = AdminUtils.fetchTopicMetadataFromZk(JavaConversions.asScalaSet(Collections.singleton(topicName)), zkClient);
        if (topicMetadatas != null && topicMetadatas.size() > 0) {
            return JavaConversions.asJavaSet(topicMetadatas).iterator().next().partitionsMetadata().size();
        } else {
            throw new IllegalStateException("Failed to get metadata for topic " + topicName);
        }
    } catch (Exception e) {
        throw new IllegalStateException("Failed to retrieve partitions for topic " + topicName, e);
    } finally {
        try {
            zkClient.close();
        } catch (Exception e2) {
        // ignore
        }
    }
}