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

Example 1 with CommandLookupTopic

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

the class Commands method newLookup.

public static ByteBuf newLookup(String topic, String listenerName, boolean authoritative, long requestId) {
    BaseCommand cmd = localCmd(Type.LOOKUP);
    CommandLookupTopic lookup = cmd.setLookupTopic().setTopic(topic).setRequestId(requestId).setAuthoritative(authoritative);
    if (StringUtils.isNotBlank(listenerName)) {
        lookup.setAdvertisedListenerName(listenerName);
    }
    return serializeWithSize(cmd);
}

Example 2 with CommandLookupTopic

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

the class ServerCnx method handleLookup.

@Override
protected void handleLookup(CommandLookupTopic lookup) {
    final long requestId = lookup.getRequestId();
    final boolean authoritative = lookup.isAuthoritative();
    // use the connection-specific listener name by default.
    final String advertisedListenerName = lookup.hasAdvertisedListenerName() && StringUtils.isNotBlank(lookup.getAdvertisedListenerName()) ? lookup.getAdvertisedListenerName() : this.listenerName;
    if (log.isDebugEnabled()) {
        log.debug("[{}] Received Lookup from {} for {} requesting listener {}", lookup.getTopic(), remoteAddress, requestId, StringUtils.isNotBlank(advertisedListenerName) ? advertisedListenerName : "(none)");
    }
    TopicName topicName = validateTopicName(lookup.getTopic(), requestId, lookup);
    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 lookup ";
            log.warn("[{}] {} with role {} and proxyClientAuthRole {} on topic {}", remoteAddress, msg, authRole, originalPrincipal, topicName);
            ctx.writeAndFlush(newLookupErrorResponse(ServerError.AuthorizationError, msg, requestId));
            lookupSemaphore.release();
            return;
        }
        isTopicOperationAllowed(topicName, TopicOperation.LOOKUP).thenApply(isAuthorized -> {
            if (isAuthorized) {
                lookupTopicAsync(getBrokerService().pulsar(), topicName, authoritative, getPrincipal(), getAuthenticationData(), requestId, advertisedListenerName).handle((lookupResponse, ex) -> {
                    if (ex == null) {
                        ctx.writeAndFlush(lookupResponse);
                    } else {
                        // it should never happen
                        log.warn("[{}] lookup failed with error {}, {}", remoteAddress, topicName, ex.getMessage(), ex);
                        ctx.writeAndFlush(newLookupErrorResponse(ServerError.ServiceNotReady, ex.getMessage(), requestId));
                    }
                    lookupSemaphore.release();
                    return null;
                });
            } else {
                final String msg = "Proxy Client is not authorized to Lookup";
                log.warn("[{}] {} with role {} on topic {}", remoteAddress, msg, getPrincipal(), topicName);
                ctx.writeAndFlush(newLookupErrorResponse(ServerError.AuthorizationError, msg, requestId));
                lookupSemaphore.release();
            }
            return null;
        }).exceptionally(ex -> {
            logAuthException(remoteAddress, "lookup", getPrincipal(), Optional.of(topicName), ex);
            final String msg = "Exception occurred while trying to authorize lookup";
            ctx.writeAndFlush(newLookupErrorResponse(ServerError.AuthorizationError, msg, requestId));
            lookupSemaphore.release();
            return null;
        });
    } else {
        if (log.isDebugEnabled()) {
            log.debug("[{}] Failed lookup due to too many lookup-requests {}", remoteAddress, topicName);
        }
        ctx.writeAndFlush(newLookupErrorResponse(ServerError.TooManyRequests, "Failed due to too many pending lookup requests", requestId));
    }
}