Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) "Neo4j"
* Neo4j Sweden AB [https://neo4j.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*/
package org.neo4j.fabric.transaction.parent;
import static org.neo4j.kernel.api.exceptions.Status.Transaction.TransactionCommitFailed;
import static org.neo4j.kernel.api.exceptions.Status.Transaction.TransactionRollbackFailed;
import static org.neo4j.kernel.api.exceptions.Status.Transaction.TransactionTerminationFailed;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.function.Function;
import java.util.function.Supplier;
import org.neo4j.fabric.executor.Exceptions;
import org.neo4j.fabric.executor.FabricException;
import org.neo4j.fabric.executor.Location;
import org.neo4j.fabric.transaction.ErrorReporter;
import org.neo4j.fabric.transaction.TransactionMode;
import org.neo4j.graphdb.TransactionTerminatedException;
import org.neo4j.kernel.api.TerminationMark;
import org.neo4j.kernel.api.exceptions.Status;
import org.neo4j.time.SystemNanoClock;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
/**
* Implements transaction actions for transactions that consist of child transactions
*/
public abstract class AbstractCompoundTransaction
implements CompoundTransaction {
private final ErrorReporter errorReporter;
private final SystemNanoClock clock;
private final ReadWriteLock transactionLock = new ReentrantReadWriteLock();
protected final Lock nonExclusiveLock = transactionLock.readLock();
protected final Lock exclusiveLock = transactionLock.writeLock();
protected State state = State.OPEN;
protected TerminationMark terminationMark;
private final Set autocommitQueries = ConcurrentHashMap.newKeySet();
protected final Set> readingTransactions = ConcurrentHashMap.newKeySet();
protected Child writingTransaction;
private record ReadingChildTransaction(Tx inner, boolean readingOnly) {}
protected enum State {
OPEN,
CLOSED,
TERMINATED
}
protected record ErrorRecord(String message, Throwable error) {}
protected AbstractCompoundTransaction(ErrorReporter errorReporter, SystemNanoClock clock) {
this.errorReporter = errorReporter;
this.clock = clock;
}
@Override
public Tx registerNewChildTransaction(
Location location, TransactionMode mode, Supplier transactionSupplier) throws FabricException {
return switch (mode) {
case DEFINITELY_WRITE -> startWritingTransaction(location, transactionSupplier);
case MAYBE_WRITE -> startReadingTransaction(false, transactionSupplier);
case DEFINITELY_READ -> startReadingTransaction(true, transactionSupplier);
};
}
private Tx startWritingTransaction(Location location, Supplier writeTransactionSupplier)
throws FabricException {
exclusiveLock.lock();
try {
checkTransactionOpenForStatementExecution();
if (writingTransaction != null) {
throw multipleWriteError(location, writingTransaction.location());
}
var tx = writeTransactionSupplier.get();
writingTransaction = tx;
return tx;
} finally {
exclusiveLock.unlock();
}
}
private TX startReadingTransaction(boolean readOnly, Supplier readingTransactionSupplier)
throws FabricException {
nonExclusiveLock.lock();
try {
checkTransactionOpenForStatementExecution();
var tx = readingTransactionSupplier.get();
readingTransactions.add(new ReadingChildTransaction<>(tx, readOnly));
return tx;
} finally {
nonExclusiveLock.unlock();
}
}
@Override
public void upgradeToWritingTransaction(Tx childTransaction) throws FabricException {
if (this.writingTransaction == childTransaction) {
return;
}
exclusiveLock.lock();
try {
if (this.writingTransaction == childTransaction) {
return;
}
if (this.writingTransaction != null) {
throw multipleWriteError(childTransaction.location(), this.writingTransaction.location());
}
var readingTransaction = readingTransactions.stream()
.filter(readingTx -> readingTx.inner == childTransaction)
.findAny()
.orElseThrow(
() -> new IllegalArgumentException("The supplied transaction has not been registered"));
if (readingTransaction.readingOnly) {
throw new IllegalStateException("Upgrading reading-only transaction to a writing one is not allowed");
}
readingTransactions.remove(readingTransaction);
this.writingTransaction = readingTransaction.inner;
} finally {
exclusiveLock.unlock();
}
}
@Override
public void commit() {
exclusiveLock.lock();
try {
if (state == State.TERMINATED) {
// Wait for all children to be rolled back. Ignore errors
doRollbackAndIgnoreErrors(this::childTransactionRollback);
throw new TransactionTerminatedException(terminationMark.getReason());
}
if (state == State.CLOSED) {
throw new FabricException(TransactionCommitFailed, "Trying to commit closed transaction");
}
state = State.CLOSED;
var allFailures = new ArrayList();
try {
doOnChildren(readingTransactions, null, this::childTransactionCommit)
.forEach(error ->
allFailures.add(new ErrorRecord("Failed to commit a child read transaction", error)));
if (!allFailures.isEmpty()) {
doOnChildren(List.of(), writingTransaction, this::childTransactionRollback)
.forEach(error -> allFailures.add(
new ErrorRecord("Failed to rollback a child write transaction", error)));
} else {
doOnChildren(List.of(), writingTransaction, this::childTransactionCommit)
.forEach(error -> allFailures.add(
new ErrorRecord("Failed to commit a child write transaction", error)));
}
} catch (Exception e) {
allFailures.add(new ErrorRecord("Failed to commit composite transaction", commitFailedError()));
} finally {
closeContextsAndRemoveTransaction();
}
throwIfNonEmpty(allFailures, TransactionCommitFailed);
} finally {
exclusiveLock.unlock();
}
}
@Override
public void rollback() {
exclusiveLock.lock();
try {
// guard against someone calling rollback after 'begin' failure
if (isUninitialized()) {
return;
}
if (state == State.TERMINATED) {
// Wait for all children to be rolled back. Ignore errors
doRollbackAndIgnoreErrors(this::childTransactionRollback);
return;
}
if (state == State.CLOSED) {
return;
}
state = State.CLOSED;
doRollback(this::childTransactionRollback);
} finally {
exclusiveLock.unlock();
}
}
private void doRollback(Function> operation) {
var allFailures = new ArrayList();
try {
doOnChildren(readingTransactions, writingTransaction, operation)
.forEach(
error -> allFailures.add(new ErrorRecord("Failed to rollback a child transaction", error)));
} catch (Exception e) {
allFailures.add(new ErrorRecord("Failed to rollback composite transaction", rollbackFailedError()));
} finally {
closeContextsAndRemoveTransaction();
}
throwIfNonEmpty(allFailures, TransactionRollbackFailed);
}
private void doRollbackAndIgnoreErrors(Function> operation) {
try {
doOnChildren(readingTransactions, writingTransaction, operation);
} finally {
closeContextsAndRemoveTransaction();
}
}
@Override
public boolean markForTermination(Status reason) {
// While state is open, take the lock by polling.
// We do this to re-check state, which could be set by another thread committing or rolling back.
while (true) {
try {
if (state != State.OPEN) {
return false;
} else {
if (exclusiveLock.tryLock(100, java.util.concurrent.TimeUnit.MILLISECONDS)) {
break;
}
}
} catch (InterruptedException e) {
throw terminationFailedError();
}
}
try {
if (state != State.OPEN) {
return false;
}
terminationMark = new TerminationMark(reason, clock.nanos());
state = State.TERMINATED;
terminateChildren(reason);
autocommitQueries.forEach(q -> q.terminate(reason));
} finally {
exclusiveLock.unlock();
}
return true;
}
@Override
public void childTransactionTerminated(Status reason) {
if (!isOpen()) {
return;
}
markForTermination(reason);
}
@Override
public void registerAutocommitQuery(AutocommitQuery autocommitQuery) {
autocommitQueries.add(autocommitQuery);
// Handle a case when we are registering to an already terminated transaction
if (state == State.TERMINATED) {
autocommitQuery.terminate(terminationMark.getReason());
}
}
@Override
public void unRegisterAutocommitQuery(AutocommitQuery autocommitQuery) {
autocommitQueries.remove(autocommitQuery);
}
private void terminateChildren(Status reason) {
var allFailures = new ArrayList();
try {
doOnChildren(
readingTransactions,
writingTransaction,
singleDbTransaction -> childTransactionTerminate(singleDbTransaction, reason))
.forEach(error ->
allFailures.add(new ErrorRecord("Failed to terminate a child transaction", error)));
} catch (Exception e) {
allFailures.add(new ErrorRecord("Failed to terminate composite transaction", terminationFailedError()));
}
throwIfNonEmpty(allFailures, TransactionTerminationFailed);
}
public boolean isOpen() {
return state == State.OPEN;
}
public Optional getTerminationMark() {
return Optional.ofNullable(terminationMark);
}
protected void checkTransactionOpenForStatementExecution() {
if (state == State.TERMINATED) {
throw new TransactionTerminatedException(terminationMark.getReason());
}
if (state == State.CLOSED) {
throw new FabricException(
Status.Statement.ExecutionFailed, "Trying to execute query in a closed transaction");
}
}
private List doOnChildren(
Iterable> readingTransactions,
Child writingTransaction,
Function> operation) {
var failures = Flux.fromIterable(readingTransactions)
.map(txWrapper -> txWrapper.inner)
.concatWith(Mono.justOrEmpty(writingTransaction))
.flatMap(tx -> catchErrors(operation.apply(tx)))
.collectList()
.block();
return failures == null ? List.of() : failures;
}
private Mono catchErrors(Mono action) {
return action.flatMap(v -> Mono.empty()).onErrorResume(Mono::just);
}
private void throwIfNonEmpty(List failures, Status defaultStatusCode) {
if (!failures.isEmpty()) {
// The main exception is not logged, because it will be logged by Bolt
// and the log would contain two lines reporting the same thing without any additional info.
var mainException = Exceptions.transform(defaultStatusCode, failures.get(0).error);
for (int i = 1; i < failures.size(); i++) {
var errorRecord = failures.get(i);
mainException.addSuppressed(errorRecord.error);
errorReporter.report(errorRecord.message, errorRecord.error, defaultStatusCode);
}
throw mainException;
}
}
private FabricException multipleWriteError(Location attempt, Location current) {
// There are two situations and the error should reflect them in order not to confuse the users:
// 1. This is actually the same database, but the location has changed, because of leader switch in the cluster.
if (current.getUuid().equals(attempt.getUuid())) {
return new FabricException(
Status.Transaction.LeaderSwitch,
"Could not write to a database due to a cluster leader switch that occurred during the transaction. "
+ "Previous leader: %s, Current leader: %s.",
current,
attempt);
}
// 2. The user is really trying to write to two different databases.
return new FabricException(
Status.Statement.AccessMode,
"Writing to more than one database per transaction is not allowed. Attempted write to %s, currently writing to %s",
attempt.databaseReference().toPrettyString(),
current.databaseReference().toPrettyString());
}
private FabricException commitFailedError() {
return new FabricException(TransactionCommitFailed, "Failed to commit composite transaction");
}
private FabricException rollbackFailedError() {
return new FabricException(TransactionRollbackFailed, "Failed to rollback composite transaction");
}
private FabricException terminationFailedError() {
return new FabricException(TransactionTerminationFailed, "Failed to terminate composite transaction");
}
protected abstract boolean isUninitialized();
protected abstract void closeContextsAndRemoveTransaction();
protected abstract Mono childTransactionCommit(Child child);
protected abstract Mono childTransactionRollback(Child child);
protected abstract Mono childTransactionTerminate(Child child, Status reason);
}
