Examples with CommandPartitionedTopicMetadata org.apache.pulsar.common.api.proto.CommandPartitionedTopicMetadata used on opensource projects

Example 1 with CommandPartitionedTopicMetadata

use of org.apache.pulsar.common.api.proto.CommandPartitionedTopicMetadata in project pulsar by apache.

the class LookupProxyHandler method handlePartitionMetadataResponse.

/**
 *   Always get partition metadata from broker service.
 */
private void handlePartitionMetadataResponse(CommandPartitionedTopicMetadata partitionMetadata, long clientRequestId) {
    TopicName topicName = TopicName.get(partitionMetadata.getTopic());
    URI brokerURI;
    try {
        String availableBrokerServiceURL = getBrokerServiceUrl(clientRequestId);
        if (availableBrokerServiceURL == null) {
            log.warn("No available broker for {} to lookup partition metadata", topicName);
            return;
        }
        brokerURI = new URI(availableBrokerServiceURL);
    } catch (URISyntaxException e) {
        proxyConnection.ctx().writeAndFlush(Commands.newPartitionMetadataResponse(ServerError.MetadataError, e.getMessage(), clientRequestId));
        return;
    }
    InetSocketAddress addr = new InetSocketAddress(brokerURI.getHost(), brokerURI.getPort());
    if (log.isDebugEnabled()) {
        log.debug("Getting connections to '{}' for Looking up topic '{}' with clientReq Id '{}'", addr, topicName.getPartitionedTopicName(), clientRequestId);
    }
    proxyConnection.getConnectionPool().getConnection(addr).thenAccept(clientCnx -> {
        // Connected to backend broker
        long requestId = proxyConnection.newRequestId();
        ByteBuf command;
        command = Commands.newPartitionMetadataRequest(topicName.toString(), requestId);
        clientCnx.newLookup(command, requestId).whenComplete((r, t) -> {
            if (t != null) {
                log.warn("[{}] failed to get Partitioned metadata : {}", topicName.toString(), t.getMessage(), t);
                proxyConnection.ctx().writeAndFlush(Commands.newLookupErrorResponse(getServerError(t), t.getMessage(), clientRequestId));
            } else {
                proxyConnection.ctx().writeAndFlush(Commands.newPartitionMetadataResponse(r.partitions, clientRequestId));
            }
            proxyConnection.getConnectionPool().releaseConnection(clientCnx);
        });
    }).exceptionally(ex -> {
        // Failed to connect to backend broker
        proxyConnection.ctx().writeAndFlush(Commands.newPartitionMetadataResponse(ServerError.ServiceNotReady, ex.getMessage(), clientRequestId));
        return null;
    });
}

Example 2 with CommandPartitionedTopicMetadata

use of org.apache.pulsar.common.api.proto.CommandPartitionedTopicMetadata in project pulsar by apache.

the class ServerCnx method handlePartitionMetadataRequest.

@Override
protected void handlePartitionMetadataRequest(CommandPartitionedTopicMetadata partitionMetadata) {
    final long requestId = partitionMetadata.getRequestId();
    if (log.isDebugEnabled()) {
        log.debug("[{}] Received PartitionMetadataLookup from {} for {}", partitionMetadata.getTopic(), remoteAddress, requestId);
    }
    TopicName topicName = validateTopicName(partitionMetadata.getTopic(), requestId, partitionMetadata);
    if (topicName == null) {
        return;
    }
    final Semaphore lookupSemaphore = service.getLookupRequestSemaphore();
    if (lookupSemaphore.tryAcquire()) {
        if (invalidOriginalPrincipal(originalPrincipal)) {
            final String msg = "Valid Proxy Client role should be provided for getPartitionMetadataRequest ";
            log.warn("[{}] {} with role {} and proxyClientAuthRole {} on topic {}", remoteAddress, msg, authRole, originalPrincipal, topicName);
            commandSender.sendPartitionMetadataResponse(ServerError.AuthorizationError, msg, requestId);
            lookupSemaphore.release();
            return;
        }
        isTopicOperationAllowed(topicName, TopicOperation.LOOKUP).thenApply(isAuthorized -> {
            if (isAuthorized) {
                unsafeGetPartitionedTopicMetadataAsync(getBrokerService().pulsar(), topicName).handle((metadata, ex) -> {
                    if (ex == null) {
                        int partitions = metadata.partitions;
                        commandSender.sendPartitionMetadataResponse(partitions, requestId);
                    } else {
                        if (ex instanceof PulsarClientException) {
                            log.warn("Failed to authorize {} at [{}] on topic {} : {}", getRole(), remoteAddress, topicName, ex.getMessage());
                            commandSender.sendPartitionMetadataResponse(ServerError.AuthorizationError, ex.getMessage(), requestId);
                        } else {
                            log.warn("Failed to get Partitioned Metadata [{}] {}: {}", remoteAddress, topicName, ex.getMessage(), ex);
                            ServerError error = (ex instanceof RestException) && ((RestException) ex).getResponse().getStatus() < 500 ? ServerError.MetadataError : ServerError.ServiceNotReady;
                            commandSender.sendPartitionMetadataResponse(error, ex.getMessage(), requestId);
                        }
                    }
                    lookupSemaphore.release();
                    return null;
                });
            } else {
                final String msg = "Proxy Client is not authorized to Get Partition Metadata";
                log.warn("[{}] {} with role {} on topic {}", remoteAddress, msg, getPrincipal(), topicName);
                ctx.writeAndFlush(Commands.newPartitionMetadataResponse(ServerError.AuthorizationError, msg, requestId));
                lookupSemaphore.release();
            }
            return null;
        }).exceptionally(ex -> {
            logAuthException(remoteAddress, "partition-metadata", getPrincipal(), Optional.of(topicName), ex);
            final String msg = "Exception occurred while trying to authorize get Partition Metadata";
            ctx.writeAndFlush(Commands.newPartitionMetadataResponse(ServerError.AuthorizationError, msg, requestId));
            lookupSemaphore.release();
            return null;
        });
    } else {
        if (log.isDebugEnabled()) {
            log.debug("[{}] Failed Partition-Metadata lookup due to too many lookup-requests {}", remoteAddress, topicName);
        }
        commandSender.sendPartitionMetadataResponse(ServerError.TooManyRequests, "Failed due to too many pending lookup requests", requestId);
    }
}