@Override
public CompletableFuture<Void> visitGetKeysInGroupCommand(
final InvocationContext ctx, GetKeysInGroupCommand command) throws Throwable {
final String groupName = command.getGroupName();
if (!command.isGroupOwner()) {
return ctx.continueInvocation();
}
final KeyFilter<Object> keyFilter =
new CompositeKeyFilter<>(
new GroupFilter<>(groupName, groupManager),
new CollectionKeyFilter<>(ctx.getLookedUpEntries().keySet()));
dataContainer.executeTask(
keyFilter,
(o, internalCacheEntry) -> {
synchronized (ctx) {
// the process can be made in multiple threads, so we need to synchronize in the
// context.
entryFactory.wrapExternalEntry(
ctx,
internalCacheEntry.getKey(),
internalCacheEntry,
EntryFactory.Wrap.STORE,
false);
}
});
return ctx.continueInvocation();
}
@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> 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;
});
}
@Override
public CompletableFuture<Void> visitApplyDeltaCommand(
InvocationContext ctx, ApplyDeltaCommand command) throws Throwable {
entryFactory.wrapEntryForDelta(ctx, command.getKey(), command.getDelta());
return ctx.continueInvocation();
}