private CompletableFuture<Void> visitSecondPhaseCommand(
TxInvocationContext ctx,
TransactionBoundaryCommand command,
boolean commit,
ExtendedStatistic duration,
ExtendedStatistic counter)
throws Throwable {
GlobalTransaction globalTransaction = command.getGlobalTransaction();
if (trace) {
log.tracef(
"Visit 2nd phase command %s. Is it local? %s. Transaction is %s",
command, ctx.isOriginLocal(), globalTransaction.globalId());
}
long start = timeService.time();
return ctx.onReturn(
(rCtx, rCommand, rv, throwable) -> {
if (throwable != null) {
throw throwable;
}
long end = timeService.time();
updateTime(duration, counter, start, end, globalTransaction, rCtx.isOriginLocal());
cacheStatisticManager.setTransactionOutcome(
commit, globalTransaction, rCtx.isOriginLocal());
cacheStatisticManager.terminateTransaction(globalTransaction, true, true);
return null;
});
}
@Override
public CompletableFuture<Void> visitGetAllCommand(InvocationContext ctx, GetAllCommand command)
throws Throwable {
if (trace) {
log.tracef(
"Visit Get All Command %s. Is it in transaction scope? %s. Is it local? %s",
command, ctx.isInTxScope(), ctx.isOriginLocal());
}
if (!ctx.isInTxScope()) {
return ctx.continueInvocation();
}
long start = timeService.time();
return ctx.onReturn(
(rCtx, rCommand, rv, throwable) -> {
if (throwable != null) {
throw throwable;
}
long end = timeService.time();
initStatsIfNecessary(rCtx);
int numRemote = 0;
Collection<?> keys = ((GetAllCommand) rCommand).getKeys();
for (Object key : keys) {
if (isRemote(key)) numRemote++;
}
// TODO: tbh this seems like it doesn't work properly for statistics as each
// one will have the duration of all the time for all gets... Maybe do an average
// instead ? Either way this isn't very indicative
if (numRemote > 0) {
cacheStatisticManager.add(
NUM_REMOTE_GET, numRemote, getGlobalTransaction(rCtx), rCtx.isOriginLocal());
cacheStatisticManager.add(
REMOTE_GET_EXECUTION,
timeService.timeDuration(start, end, NANOSECONDS),
getGlobalTransaction(rCtx),
rCtx.isOriginLocal());
}
cacheStatisticManager.add(
ALL_GET_EXECUTION,
timeService.timeDuration(start, end, NANOSECONDS),
getGlobalTransaction(rCtx),
rCtx.isOriginLocal());
cacheStatisticManager.add(
NUM_GET, keys.size(), getGlobalTransaction(rCtx), rCtx.isOriginLocal());
return null;
});
}
@Override
public CompletableFuture<Void> visitPrepareCommand(
TxInvocationContext ctx, PrepareCommand command) throws Throwable {
GlobalTransaction globalTransaction = command.getGlobalTransaction();
if (trace) {
log.tracef(
"Visit Prepare command %s. Is it local?. Transaction is %s",
command, ctx.isOriginLocal(), globalTransaction.globalId());
}
initStatsIfNecessary(ctx);
cacheStatisticManager.onPrepareCommand(globalTransaction, ctx.isOriginLocal());
if (command.hasModifications()) {
cacheStatisticManager.markAsWriteTransaction(globalTransaction, ctx.isOriginLocal());
}
long start = timeService.time();
return ctx.onReturn(
(rCtx, rCommand, rv, throwable) -> {
if (throwable != null) {
processWriteException(rCtx, globalTransaction, throwable);
} else {
long end = timeService.time();
updateTime(
PREPARE_EXECUTION_TIME,
NUM_PREPARE_COMMAND,
start,
end,
globalTransaction,
rCtx.isOriginLocal());
}
if (((PrepareCommand) rCommand).isOnePhaseCommit()) {
boolean local = rCtx.isOriginLocal();
boolean success = throwable == null;
cacheStatisticManager.setTransactionOutcome(
success, globalTransaction, rCtx.isOriginLocal());
cacheStatisticManager.terminateTransaction(globalTransaction, local, !local);
}
return null;
});
}
@Override
public CompletableFuture<Void> visitGetKeyValueCommand(
InvocationContext ctx, GetKeyValueCommand command) throws Throwable {
if (trace) {
log.tracef(
"Visit Get Key Value command %s. Is it in transaction scope? %s. Is it local? %s",
command, ctx.isInTxScope(), ctx.isOriginLocal());
}
if (!ctx.isInTxScope()) {
return ctx.continueInvocation();
}
long start = timeService.time();
return ctx.onReturn(
(rCtx, rCommand, rv, throwable) -> {
if (throwable != null) {
throw throwable;
}
long end = timeService.time();
initStatsIfNecessary(rCtx);
Object key = ((GetKeyValueCommand) rCommand).getKey();
if (isRemote(key)) {
cacheStatisticManager.increment(
NUM_REMOTE_GET, getGlobalTransaction(rCtx), rCtx.isOriginLocal());
cacheStatisticManager.add(
REMOTE_GET_EXECUTION,
timeService.timeDuration(start, end, NANOSECONDS),
getGlobalTransaction(rCtx),
rCtx.isOriginLocal());
}
cacheStatisticManager.add(
ALL_GET_EXECUTION,
timeService.timeDuration(start, end, NANOSECONDS),
getGlobalTransaction(rCtx),
rCtx.isOriginLocal());
cacheStatisticManager.increment(
NUM_GET, getGlobalTransaction(rCtx), rCtx.isOriginLocal());
return null;
});
}
private CompletableFuture<Void> visitWriteCommand(
InvocationContext ctx, WriteCommand command, Object key) throws Throwable {
if (trace) {
log.tracef(
"Visit write command %s. Is it in transaction scope? %s. Is it local? %s",
command, ctx.isInTxScope(), ctx.isOriginLocal());
}
if (!ctx.isInTxScope()) {
return ctx.continueInvocation();
}
long start = timeService.time();
return ctx.onReturn(
(rCtx, rCommand, rv, throwable) -> {
long end = timeService.time();
initStatsIfNecessary(rCtx);
if (throwable != null) {
processWriteException(rCtx, getGlobalTransaction(rCtx), throwable);
} else {
if (isRemote(key)) {
cacheStatisticManager.add(
REMOTE_PUT_EXECUTION,
timeService.timeDuration(start, end, NANOSECONDS),
getGlobalTransaction(rCtx),
rCtx.isOriginLocal());
cacheStatisticManager.increment(
NUM_REMOTE_PUT, getGlobalTransaction(rCtx), rCtx.isOriginLocal());
}
}
cacheStatisticManager.increment(
NUM_PUT, getGlobalTransaction(rCtx), rCtx.isOriginLocal());
cacheStatisticManager.markAsWriteTransaction(
getGlobalTransaction(rCtx), rCtx.isOriginLocal());
return null;
});
}