Examples with Commands org.apache.pulsar.common.protocol.Commands used on opensource projects

Example 1 with Commands

use of org.apache.pulsar.common.protocol.Commands in project pulsar by apache.

the class PersistentTopicsBase method internalTerminatePartitionedTopic.

protected void internalTerminatePartitionedTopic(AsyncResponse asyncResponse, boolean authoritative) {
    CompletableFuture<Void> future;
    if (topicName.isGlobal()) {
        future = validateGlobalNamespaceOwnershipAsync(namespaceName);
    } else {
        future = CompletableFuture.completedFuture(null);
    }
    future.thenCompose(__ -> validateTopicOperationAsync(topicName, TopicOperation.TERMINATE).thenCompose(unused -> getPartitionedTopicMetadataAsync(topicName, authoritative, false)).thenAccept(partitionMetadata -> {
        if (partitionMetadata.partitions == 0) {
            String msg = "Termination of a non-partitioned topic is not allowed using partitioned-terminate" + ", please use terminate commands";
            log.error("[{}] [{}] {}", clientAppId(), topicName, msg);
            asyncResponse.resume(new RestException(Status.METHOD_NOT_ALLOWED, msg));
            return;
        }
        if (partitionMetadata.partitions > 0) {
            Map<Integer, MessageId> messageIds = new ConcurrentHashMap<>(partitionMetadata.partitions);
            final List<CompletableFuture<MessageId>> futures = Lists.newArrayListWithCapacity(partitionMetadata.partitions);
            for (int i = 0; i < partitionMetadata.partitions; i++) {
                TopicName topicNamePartition = topicName.getPartition(i);
                try {
                    int finalI = i;
                    futures.add(pulsar().getAdminClient().topics().terminateTopicAsync(topicNamePartition.toString()).whenComplete((messageId, throwable) -> {
                        if (throwable != null) {
                            log.error("[{}] Failed to terminate topic {}", clientAppId(), topicNamePartition, throwable);
                            asyncResponse.resume(new RestException(throwable));
                        }
                        messageIds.put(finalI, messageId);
                    }));
                } catch (Exception e) {
                    log.error("[{}] Failed to terminate topic {}", clientAppId(), topicNamePartition, e);
                    throw new RestException(e);
                }
            }
            FutureUtil.waitForAll(futures).handle((result, exception) -> {
                if (exception != null) {
                    Throwable t = exception.getCause();
                    if (t instanceof NotFoundException) {
                        asyncResponse.resume(new RestException(Status.NOT_FOUND, "Topic not found"));
                    } else {
                        log.error("[{}] Failed to terminate topic {}", clientAppId(), topicName, t);
                        asyncResponse.resume(new RestException(t));
                    }
                }
                asyncResponse.resume(messageIds);
                return null;
            });
        }
    }).exceptionally(ex -> {
        // If the exception is not redirect exception we need to log it.
        if (!isRedirectException(ex)) {
            log.error("[{}] Failed to terminate topic {}", clientAppId(), topicName, ex);
        }
        resumeAsyncResponseExceptionally(asyncResponse, ex);
        return null;
    })).exceptionally(ex -> {
        // If the exception is not redirect exception we need to log it.
        if (!isRedirectException(ex)) {
            log.error("[{}] Failed to terminate topic {}", clientAppId(), topicName, ex);
        }
        resumeAsyncResponseExceptionally(asyncResponse, ex);
        return null;
    });
}