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

Example 1 with TxnAction

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

the class TransactionMetadataStoreService method endTransaction.

public CompletableFuture<Void> endTransaction(TxnID txnID, int txnAction, boolean isTimeout, CompletableFuture<Void> completableFuture) {
    TxnStatus newStatus;
    switch(txnAction) {
        case TxnAction.COMMIT_VALUE:
            newStatus = COMMITTING;
            break;
        case TxnAction.ABORT_VALUE:
            newStatus = ABORTING;
            break;
        default:
            TransactionCoordinatorException.UnsupportedTxnActionException exception = new TransactionCoordinatorException.UnsupportedTxnActionException(txnID, txnAction);
            LOG.error(exception.getMessage());
            completableFuture.completeExceptionally(exception);
            return completableFuture;
    }
    getTxnMeta(txnID).thenAccept(txnMeta -> {
        TxnStatus txnStatus = txnMeta.status();
        if (txnStatus == TxnStatus.OPEN) {
            updateTxnStatus(txnID, newStatus, TxnStatus.OPEN, isTimeout).thenAccept(v -> endTxnInTransactionBuffer(txnID, txnAction).thenAccept(a -> completableFuture.complete(null)).exceptionally(e -> {
                if (!isRetryableException(e.getCause())) {
                    LOG.error("EndTxnInTransactionBuffer fail! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                } else {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("EndTxnInTransactionBuffer retry! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                    }
                    transactionOpRetryTimer.newTimeout(timeout -> endTransaction(txnID, txnAction, isTimeout, completableFuture), endTransactionRetryIntervalTime, TimeUnit.MILLISECONDS);
                    return null;
                }
                completableFuture.completeExceptionally(e.getCause());
                return null;
            })).exceptionally(e -> {
                if (!isRetryableException(e.getCause())) {
                    LOG.error("EndTransaction UpdateTxnStatus fail! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                } else {
                    if (LOG.isDebugEnabled()) {
                        LOG.debug("EndTransaction UpdateTxnStatus op retry! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                    }
                    transactionOpRetryTimer.newTimeout(timeout -> endTransaction(txnID, txnAction, isTimeout, completableFuture), endTransactionRetryIntervalTime, TimeUnit.MILLISECONDS);
                    return null;
                }
                completableFuture.completeExceptionally(e.getCause());
                return null;
            });
        } else {
            if ((txnStatus == COMMITTING && txnAction == TxnAction.COMMIT.getValue()) || (txnStatus == ABORTING && txnAction == TxnAction.ABORT.getValue())) {
                endTxnInTransactionBuffer(txnID, txnAction).thenAccept(k -> completableFuture.complete(null)).exceptionally(e -> {
                    if (isRetryableException(e.getCause())) {
                        if (LOG.isDebugEnabled()) {
                            LOG.debug("EndTxnInTransactionBuffer retry! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                        }
                        transactionOpRetryTimer.newTimeout(timeout -> endTransaction(txnID, txnAction, isTimeout, completableFuture), endTransactionRetryIntervalTime, TimeUnit.MILLISECONDS);
                        return null;
                    } else {
                        LOG.error("EndTxnInTransactionBuffer fail! TxnId : {}, " + "TxnAction : {}", txnID, txnAction, e);
                    }
                    completableFuture.completeExceptionally(e.getCause());
                    return null;
                });
            } else {
                if (LOG.isDebugEnabled()) {
                    LOG.debug("EndTxnInTransactionBuffer op retry! TxnId : {}, TxnAction : {}", txnID, txnAction);
                }
                completableFuture.completeExceptionally(new InvalidTxnStatusException(txnID, newStatus, txnStatus));
            }
        }
    }).exceptionally(e -> {
        if (isRetryableException(e.getCause())) {
            if (LOG.isDebugEnabled()) {
                LOG.debug("End transaction op retry! TxnId : {}, TxnAction : {}", txnID, txnAction, e);
            }
            transactionOpRetryTimer.newTimeout(timeout -> endTransaction(txnID, txnAction, isTimeout, completableFuture), endTransactionRetryIntervalTime, TimeUnit.MILLISECONDS);
            return null;
        }
        completableFuture.completeExceptionally(e.getCause());
        return null;
    });
    return completableFuture;
}

Example 2 with TxnAction

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

the class ServerCnx method handleEndTxnOnPartition.

@Override
protected void handleEndTxnOnPartition(CommandEndTxnOnPartition command) {
    final long requestId = command.getRequestId();
    final String topic = command.getTopic();
    final int txnAction = command.getTxnAction().getValue();
    TxnID txnID = new TxnID(command.getTxnidMostBits(), command.getTxnidLeastBits());
    final long lowWaterMark = command.getTxnidLeastBitsOfLowWatermark();
    if (log.isDebugEnabled()) {
        log.debug("[{}] handleEndTxnOnPartition txnId: [{}], txnAction: [{}]", topic, txnID, txnAction);
    }
    CompletableFuture<Optional<Topic>> topicFuture = service.getTopicIfExists(TopicName.get(topic).toString());
    topicFuture.thenAccept(optionalTopic -> {
        if (optionalTopic.isPresent()) {
            optionalTopic.get().endTxn(txnID, txnAction, lowWaterMark).whenComplete((ignored, throwable) -> {
                if (throwable != null) {
                    log.error("handleEndTxnOnPartition fail!, topic {}, txnId: [{}], " + "txnAction: [{}]", topic, txnID, TxnAction.valueOf(txnAction), throwable);
                    ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, BrokerServiceException.getClientErrorCode(throwable), throwable.getMessage(), txnID.getLeastSigBits(), txnID.getMostSigBits()));
                    return;
                }
                ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, txnID.getLeastSigBits(), txnID.getMostSigBits()));
            });
        } else {
            getBrokerService().getManagedLedgerFactory().asyncExists(TopicName.get(topic).getPersistenceNamingEncoding()).thenAccept((b) -> {
                if (b) {
                    log.error("handleEndTxnOnPartition fail ! The topic {} does not exist in broker, " + "txnId: [{}], txnAction: [{}]", topic, txnID, TxnAction.valueOf(txnAction));
                    ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, ServerError.ServiceNotReady, "The topic " + topic + " does not exist in broker.", txnID.getMostSigBits(), txnID.getLeastSigBits()));
                } else {
                    log.warn("handleEndTxnOnPartition fail ! The topic {} has not been created, " + "txnId: [{}], txnAction: [{}]", topic, txnID, TxnAction.valueOf(txnAction));
                    ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, txnID.getLeastSigBits(), txnID.getMostSigBits()));
                }
            }).exceptionally(e -> {
                log.error("handleEndTxnOnPartition fail ! topic {}, " + "txnId: [{}], txnAction: [{}]", topic, txnID, TxnAction.valueOf(txnAction), e.getCause());
                ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, ServerError.ServiceNotReady, e.getMessage(), txnID.getLeastSigBits(), txnID.getMostSigBits()));
                return null;
            });
        }
    }).exceptionally(e -> {
        log.error("handleEndTxnOnPartition fail ! topic {}, " + "txnId: [{}], txnAction: [{}]", topic, txnID, TxnAction.valueOf(txnAction), e.getCause());
        ctx.writeAndFlush(Commands.newEndTxnOnPartitionResponse(requestId, ServerError.ServiceNotReady, e.getMessage(), txnID.getLeastSigBits(), txnID.getMostSigBits()));
        return null;
    });
}

Example 3 with TxnAction

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

the class ServerCnx method handleEndTxnOnSubscription.

@Override
protected void handleEndTxnOnSubscription(CommandEndTxnOnSubscription command) {
    final long requestId = command.getRequestId();
    final long txnidMostBits = command.getTxnidMostBits();
    final long txnidLeastBits = command.getTxnidLeastBits();
    final String topic = command.getSubscription().getTopic();
    final String subName = command.getSubscription().getSubscription();
    final int txnAction = command.getTxnAction().getValue();
    final TxnID txnID = new TxnID(txnidMostBits, txnidLeastBits);
    final long lowWaterMark = command.getTxnidLeastBitsOfLowWatermark();
    if (log.isDebugEnabled()) {
        log.debug("[{}] [{}] handleEndTxnOnSubscription txnId: [{}], txnAction: [{}]", topic, subName, new TxnID(txnidMostBits, txnidLeastBits), txnAction);
    }
    CompletableFuture<Optional<Topic>> topicFuture = service.getTopicIfExists(TopicName.get(topic).toString());
    topicFuture.thenAccept(optionalTopic -> {
        if (optionalTopic.isPresent()) {
            Subscription subscription = optionalTopic.get().getSubscription(subName);
            if (subscription == null) {
                log.warn("handleEndTxnOnSubscription fail! " + "topic {} subscription {} does not exist. txnId: [{}], txnAction: [{}]", optionalTopic.get().getName(), subName, txnID, TxnAction.valueOf(txnAction));
                ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnidLeastBits, txnidMostBits));
                return;
            }
            CompletableFuture<Void> completableFuture = subscription.endTxn(txnidMostBits, txnidLeastBits, txnAction, lowWaterMark);
            completableFuture.whenComplete((ignored, e) -> {
                if (e != null) {
                    log.error("handleEndTxnOnSubscription fail ! topic: {}, subscription: {}" + "txnId: [{}], txnAction: [{}]", topic, subName, txnID, TxnAction.valueOf(txnAction), e.getCause());
                    ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnidLeastBits, txnidMostBits, BrokerServiceException.getClientErrorCode(e), "Handle end txn on subscription failed."));
                    return;
                }
                ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnidLeastBits, txnidMostBits));
            });
        } else {
            getBrokerService().getManagedLedgerFactory().asyncExists(TopicName.get(topic).getPersistenceNamingEncoding()).thenAccept((b) -> {
                if (b) {
                    log.error("handleEndTxnOnSubscription fail! The topic {} does not exist in broker, " + "subscription: {}, txnId: [{}], txnAction: [{}]", topic, subName, new TxnID(txnidMostBits, txnidLeastBits), TxnAction.valueOf(txnAction));
                    ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnID.getLeastSigBits(), txnID.getMostSigBits(), ServerError.ServiceNotReady, "The topic " + topic + " does not exist in broker."));
                } else {
                    log.warn("handleEndTxnOnSubscription fail ! The topic {} has not been created, " + "subscription: {} txnId: [{}], txnAction: [{}]", topic, subName, txnID, TxnAction.valueOf(txnAction));
                    ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnID.getLeastSigBits(), txnID.getMostSigBits()));
                }
            }).exceptionally(e -> {
                log.error("handleEndTxnOnSubscription fail ! topic {}, subscription: {}" + "txnId: [{}], txnAction: [{}]", topic, subName, txnID, TxnAction.valueOf(txnAction), e.getCause());
                ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnID.getLeastSigBits(), txnID.getMostSigBits(), ServerError.ServiceNotReady, e.getMessage()));
                return null;
            });
        }
    }).exceptionally(e -> {
        log.error("handleEndTxnOnSubscription fail ! topic: {}, subscription: {}" + "txnId: [{}], txnAction: [{}]", topic, subName, txnID, TxnAction.valueOf(txnAction), e.getCause());
        ctx.writeAndFlush(Commands.newEndTxnOnSubscriptionResponse(requestId, txnidLeastBits, txnidMostBits, ServerError.ServiceNotReady, "Handle end txn on subscription failed."));
        return null;
    });
}