Examples with DEFAULT_OPERATION_TIMEOUT_SEC org.apache.pulsar.broker.resources.PulsarResources.DEFAULT_OPERATION_TIMEOUT_SEC used on opensource projects

Example 1 with DEFAULT_OPERATION_TIMEOUT_SEC

use of org.apache.pulsar.broker.resources.PulsarResources.DEFAULT_OPERATION_TIMEOUT_SEC in project pulsar by apache.

the class PersistentTopicsBase method internalUpdatePartitionedTopic.

/**
 * It updates number of partitions of an existing partitioned topic. It requires partitioned-topic to
 * already exist and number of new partitions must be greater than existing number of partitions. Decrementing
 * number of partitions requires deletion of topic which is not supported.
 *
 * Already created partitioned producers and consumers can't see newly created partitions and it requires to
 * recreate them at application so, newly created producers and consumers can connect to newly added partitions as
 * well. Therefore, it can violate partition ordering at producers until all producers are restarted at application.
 *
 * @param numPartitions
 */
protected void internalUpdatePartitionedTopic(int numPartitions, boolean updateLocalTopicOnly, boolean authoritative, boolean force) {
    if (numPartitions <= 0) {
        throw new RestException(Status.NOT_ACCEPTABLE, "Number of partitions should be more than 0");
    }
    validateTopicOwnership(topicName, authoritative);
    validateTopicPolicyOperation(topicName, PolicyName.PARTITION, PolicyOperation.WRITE);
    // Only do the validation if it's the first hop.
    if (!updateLocalTopicOnly && !force) {
        validatePartitionTopicUpdate(topicName.getLocalName(), numPartitions);
    }
    final int maxPartitions = pulsar().getConfig().getMaxNumPartitionsPerPartitionedTopic();
    if (maxPartitions > 0 && numPartitions > maxPartitions) {
        throw new RestException(Status.NOT_ACCEPTABLE, "Number of partitions should be less than or equal to " + maxPartitions);
    }
    if (topicName.isGlobal() && isNamespaceReplicated(topicName.getNamespaceObject())) {
        Set<String> clusters = getNamespaceReplicatedClusters(topicName.getNamespaceObject());
        if (!clusters.contains(pulsar().getConfig().getClusterName())) {
            log.error("[{}] local cluster is not part of replicated cluster for namespace {}", clientAppId(), topicName);
            throw new RestException(Status.FORBIDDEN, "Local cluster is not part of replicate cluster list");
        }
        try {
            tryCreatePartitionsAsync(numPartitions).get(DEFAULT_OPERATION_TIMEOUT_SEC, TimeUnit.SECONDS);
            createSubscriptions(topicName, numPartitions).get(DEFAULT_OPERATION_TIMEOUT_SEC, TimeUnit.SECONDS);
        } catch (Exception e) {
            if (e.getCause() instanceof RestException) {
                throw (RestException) e.getCause();
            }
            log.error("[{}] Failed to update partitioned topic {}", clientAppId(), topicName, e);
            throw new RestException(e);
        }
        // other clusters and then update number of partitions.
        if (!updateLocalTopicOnly) {
            CompletableFuture<Void> updatePartition = new CompletableFuture<>();
            updatePartitionInOtherCluster(numPartitions, clusters).thenRun(() -> {
                try {
                    namespaceResources().getPartitionedTopicResources().updatePartitionedTopicAsync(topicName, p -> new PartitionedTopicMetadata(numPartitions)).thenAccept(r -> updatePartition.complete(null)).exceptionally(ex -> {
                        updatePartition.completeExceptionally(ex.getCause());
                        return null;
                    });
                } catch (Exception e) {
                    updatePartition.completeExceptionally(e);
                }
            }).exceptionally(ex -> {
                updatePartition.completeExceptionally(ex);
                return null;
            });
            try {
                updatePartition.get(DEFAULT_OPERATION_TIMEOUT_SEC, TimeUnit.SECONDS);
            } catch (Exception e) {
                log.error("{} Failed to update number of partitions in zk for topic {} and partitions {}", clientAppId(), topicName, numPartitions, e);
                if (e.getCause() instanceof RestException) {
                    throw (RestException) e.getCause();
                }
                throw new RestException(e);
            }
        }
        return;
    }
    try {
        tryCreatePartitionsAsync(numPartitions).get(DEFAULT_OPERATION_TIMEOUT_SEC, TimeUnit.SECONDS);
        updatePartitionedTopic(topicName, numPartitions, force).get(DEFAULT_OPERATION_TIMEOUT_SEC, TimeUnit.SECONDS);
    } catch (Exception e) {
        if (e.getCause() instanceof RestException) {
            throw (RestException) e.getCause();
        }
        log.error("[{}] Failed to update partitioned topic {}", clientAppId(), topicName, e);
        throw new RestException(e);
    }
}