Examples with PartitionedManagedLedgerInfo org.apache.pulsar.common.naming.PartitionedManagedLedgerInfo used on opensource projects

Example 1 with PartitionedManagedLedgerInfo

use of org.apache.pulsar.common.naming.PartitionedManagedLedgerInfo in project pulsar by apache.

the class AdminApiTest method testGetPartitionedInternalInfo.

@Test
public void testGetPartitionedInternalInfo() throws Exception {
    String partitionedTopic = "my-topic" + UUID.randomUUID().toString();
    assertEquals(admin.topics().getPartitionedTopicList("prop-xyz/ns1"), Lists.newArrayList());
    final String partitionedTopicName = "persistent://prop-xyz/ns1/" + partitionedTopic;
    admin.topics().createPartitionedTopic(partitionedTopicName, 2);
    assertEquals(admin.topics().getPartitionedTopicList("prop-xyz/ns1"), Lists.newArrayList(partitionedTopicName));
    assertEquals(admin.topics().getPartitionedTopicMetadata(partitionedTopicName).partitions, 2);
    String partitionTopic0 = partitionedTopicName + "-partition-0";
    String partitionTopic1 = partitionedTopicName + "-partition-1";
    String partitionTopic0InfoResponse = admin.topics().getInternalInfo(partitionTopic0);
    String partitionTopic1InfoResponse = admin.topics().getInternalInfo(partitionTopic1);
    // expected managed info
    PartitionedManagedLedgerInfo partitionedManagedLedgerInfo = new PartitionedManagedLedgerInfo();
    partitionedManagedLedgerInfo.version = 0L;
    partitionedManagedLedgerInfo.partitions.put(partitionTopic0, ObjectMapperFactory.getThreadLocal().readValue(partitionTopic0InfoResponse, ManagedLedgerInfo.class));
    partitionedManagedLedgerInfo.partitions.put(partitionTopic1, ObjectMapperFactory.getThreadLocal().readValue(partitionTopic1InfoResponse, ManagedLedgerInfo.class));
    String expectedResult = ObjectMapperFactory.getThreadLocal().writeValueAsString(partitionedManagedLedgerInfo);
    String partitionTopicInfoResponse = admin.topics().getInternalInfo(partitionedTopicName);
    assertEquals(partitionTopicInfoResponse, expectedResult);
}

Example 2 with PartitionedManagedLedgerInfo

use of org.apache.pulsar.common.naming.PartitionedManagedLedgerInfo in project pulsar by apache.

the class PersistentTopicsBase method internalGetManagedLedgerInfo.

protected void internalGetManagedLedgerInfo(AsyncResponse asyncResponse, boolean authoritative) {
    CompletableFuture<Void> future;
    if (topicName.isGlobal()) {
        future = validateGlobalNamespaceOwnershipAsync(namespaceName);
    } else {
        future = CompletableFuture.completedFuture(null);
    }
    future.thenAccept(__ -> {
        // If the topic name is a partition name, no need to get partition topic metadata again
        if (topicName.isPartitioned()) {
            internalGetManagedLedgerInfoForNonPartitionedTopic(asyncResponse);
        } else {
            getPartitionedTopicMetadataAsync(topicName, authoritative, false).thenAccept(partitionMetadata -> {
                if (partitionMetadata.partitions > 0) {
                    final List<CompletableFuture<String>> futures = Lists.newArrayListWithCapacity(partitionMetadata.partitions);
                    PartitionedManagedLedgerInfo partitionedManagedLedgerInfo = new PartitionedManagedLedgerInfo();
                    for (int i = 0; i < partitionMetadata.partitions; i++) {
                        TopicName topicNamePartition = topicName.getPartition(i);
                        try {
                            futures.add(pulsar().getAdminClient().topics().getInternalInfoAsync(topicNamePartition.toString()).whenComplete((response, throwable) -> {
                                if (throwable != null) {
                                    log.error("[{}] Failed to get managed info for {}", clientAppId(), topicNamePartition, throwable);
                                    asyncResponse.resume(new RestException(throwable));
                                }
                                try {
                                    partitionedManagedLedgerInfo.partitions.put(topicNamePartition.toString(), jsonMapper().readValue(response, ManagedLedgerInfo.class));
                                } catch (JsonProcessingException ex) {
                                    log.error("[{}] Failed to parse ManagedLedgerInfo for {} from [{}]", clientAppId(), topicNamePartition, response, ex);
                                }
                            }));
                        } catch (PulsarServerException e) {
                            log.error("[{}] Failed to get admin client while get managed info for {}", 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 get managed info for {}", clientAppId(), topicName, t);
                                asyncResponse.resume(new RestException(t));
                            }
                        }
                        asyncResponse.resume((StreamingOutput) output -> {
                            jsonMapper().writer().writeValue(output, partitionedManagedLedgerInfo);
                        });
                        return null;
                    });
                } else {
                    internalGetManagedLedgerInfoForNonPartitionedTopic(asyncResponse);
                }
            }).exceptionally(ex -> {
                // If the exception is not redirect exception we need to log it.
                if (!isRedirectException(ex)) {
                    log.error("[{}] Failed to get partitioned metadata while get managed info for {}", 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 validate the global namespace ownership while get managed info for {}", clientAppId(), topicName, ex);
        }
        resumeAsyncResponseExceptionally(asyncResponse, ex);
        return null;
    });
}