package org.infinispan.interceptors.locking;
import java.util.Collection;
import java.util.function.Predicate;
import org.infinispan.InvalidCacheUsageException;
import org.infinispan.commands.DataCommand;
import org.infinispan.commands.FlagAffectedCommand;
import org.infinispan.commands.read.GetAllCommand;
import org.infinispan.commands.write.DataWriteCommand;
import org.infinispan.context.InvocationContext;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;
/**
* Locking interceptor to be used for non-transactional caches.
*
* @author Mircea Markus
*/
public class NonTransactionalLockingInterceptor extends AbstractLockingInterceptor {
private static final Log log = LogFactory.getLog(NonTransactionalLockingInterceptor.class);
private final Predicate<Object> shouldLockKey = this::shouldLockKey;
@Override
protected Log getLog() {
return log;
}
@Override
protected final Object visitDataReadCommand(InvocationContext ctx, DataCommand command) throws Throwable {
assertNonTransactional(ctx);
return invokeNext(ctx, command);
}
@Override
protected Object visitDataWriteCommand(InvocationContext ctx, DataWriteCommand command) throws Throwable {
assertNonTransactional(ctx);
return visitNonTxDataWriteCommand(ctx, command);
}
@Override
public Object visitGetAllCommand(InvocationContext ctx, GetAllCommand command) throws Throwable {
assertNonTransactional(ctx);
return invokeNext(ctx, command);
}
@Override
protected <K> Object handleWriteManyCommand(InvocationContext ctx, FlagAffectedCommand command, Collection<K> keys, boolean forwarded) throws Throwable {
assertNonTransactional(ctx);
if (forwarded || hasSkipLocking(command)) {
return invokeNext(ctx, command);
}
try {
lockAllAndRecord(ctx, keys.stream().filter(shouldLockKey), getLockTimeoutMillis(command));
} catch (Throwable t) {
lockManager.unlockAll(ctx);
throw t;
}
return invokeNextAndFinally(ctx, command, unlockAllReturnHandler);
}
private void assertNonTransactional(InvocationContext ctx) {
//this only happens if the cache is used in a transaction's scope
if (ctx.isInTxScope()) {
throw new InvalidCacheUsageException(
"This is a non-transactional cache and cannot be accessed with a transactional InvocationContext.");
}
}
}