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

Example 1 with CommandSeek

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

the class ServerCnx method handleSeek.

@Override
protected void handleSeek(CommandSeek seek) {
    checkArgument(state == State.Connected);
    final long requestId = seek.getRequestId();
    CompletableFuture<Consumer> consumerFuture = consumers.get(seek.getConsumerId());
    if (!seek.hasMessageId() && !seek.hasMessagePublishTime()) {
        commandSender.sendErrorResponse(requestId, ServerError.MetadataError, "Message id and message publish time were not present");
        return;
    }
    boolean consumerCreated = consumerFuture != null && consumerFuture.isDone() && !consumerFuture.isCompletedExceptionally();
    if (consumerCreated && seek.hasMessageId()) {
        Consumer consumer = consumerFuture.getNow(null);
        Subscription subscription = consumer.getSubscription();
        MessageIdData msgIdData = seek.getMessageId();
        long[] ackSet = null;
        if (msgIdData.getAckSetsCount() > 0) {
            ackSet = new long[msgIdData.getAckSetsCount()];
            for (int i = 0; i < ackSet.length; i++) {
                ackSet[i] = msgIdData.getAckSetAt(i);
            }
        }
        Position position = new PositionImpl(msgIdData.getLedgerId(), msgIdData.getEntryId(), ackSet);
        subscription.resetCursor(position).thenRun(() -> {
            log.info("[{}] [{}][{}] Reset subscription to message id {}", remoteAddress, subscription.getTopic().getName(), subscription.getName(), position);
            commandSender.sendSuccessResponse(requestId);
        }).exceptionally(ex -> {
            log.warn("[{}][{}] Failed to reset subscription: {}", remoteAddress, subscription, ex.getMessage(), ex);
            commandSender.sendErrorResponse(requestId, ServerError.UnknownError, "Error when resetting subscription: " + ex.getCause().getMessage());
            return null;
        });
    } else if (consumerCreated && seek.hasMessagePublishTime()) {
        Consumer consumer = consumerFuture.getNow(null);
        Subscription subscription = consumer.getSubscription();
        long timestamp = seek.getMessagePublishTime();
        subscription.resetCursor(timestamp).thenRun(() -> {
            log.info("[{}] [{}][{}] Reset subscription to publish time {}", remoteAddress, subscription.getTopic().getName(), subscription.getName(), timestamp);
            commandSender.sendSuccessResponse(requestId);
        }).exceptionally(ex -> {
            log.warn("[{}][{}] Failed to reset subscription: {}", remoteAddress, subscription, ex.getMessage(), ex);
            commandSender.sendErrorResponse(requestId, ServerError.UnknownError, "Reset subscription to publish time error: " + ex.getCause().getMessage());
            return null;
        });
    } else {
        commandSender.sendErrorResponse(requestId, ServerError.MetadataError, "Consumer not found");
    }
}

Example 2 with CommandSeek

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

the class Commands method newSeek.

public static ByteBuf newSeek(long consumerId, long requestId, long ledgerId, long entryId, long[] ackSet) {
    BaseCommand cmd = localCmd(Type.SEEK);
    CommandSeek seek = cmd.setSeek().setConsumerId(consumerId).setRequestId(requestId);
    MessageIdData messageId = seek.setMessageId().setLedgerId(ledgerId).setEntryId(entryId);
    for (int i = 0; i < ackSet.length; i++) {
        messageId.addAckSet(ackSet[i]);
    }
    return serializeWithSize(cmd);
}