/**
* CatchThrowableProxy is a wrapper around interface that does not allow any exception to be
* thrown when invoking methods on that interface. All exceptions are logged but not propagated to
* the caller.
*/
static class CatchThrowableProxy implements java.lang.reflect.InvocationHandler {
private static final Log log = LogFactory.getLog(CatchThrowableProxy.class);
private Object obj;
public static Object newInstance(Object obj) {
return java.lang.reflect.Proxy.newProxyInstance(
obj.getClass().getClassLoader(),
obj.getClass().getInterfaces(),
new CatchThrowableProxy(obj));
}
private CatchThrowableProxy(Object obj) {
this.obj = obj;
}
public Object invoke(Object proxy, Method m, Object[] args) throws Throwable {
Object result = null;
try {
result = m.invoke(obj, args);
} catch (Throwable t) {
log.ignoringException(m.getName(), t.getMessage(), t.getCause());
} finally {
}
return result;
}
}
public static class EntryCreatedInterceptor extends BaseCustomInterceptor {
Log log = LogFactory.getLog(EntryCreatedInterceptor.class);
final CountDownLatch latch;
volatile boolean assertKeySet;
private EntryCreatedInterceptor(CountDownLatch latch) {
this.latch = latch;
}
@Override
public Object visitPutKeyValueCommand(InvocationContext ctx, PutKeyValueCommand command)
throws Throwable {
// First execute the operation itself
Object ret = super.visitPutKeyValueCommand(ctx, command);
assertKeySet = (cache.keySet().size() == 1);
// After entry has been committed to the container
log.info("Cache entry created, now check in different thread");
latch.countDown();
// Force a bit of delay in the listener
TestingUtil.sleepThread(3000);
return ret;
}
}
/**
* // TODO: Document this
*
* @author Pedro Ruivo
* @since 4.0
*/
public class TotalOrderVersionedCommitContextEntries extends NonVersionedCommitContextEntries {
private final Log log = LogFactory.getLog(TotalOrderVersionedCommitContextEntries.class);
private VersionGenerator versionGenerator;
@Inject
public final void injectVersionGenerator(VersionGenerator versionGenerator) {
this.versionGenerator = versionGenerator;
}
@Override
protected Log getLog() {
return log;
}
@Override
protected void commitContextEntry(
CacheEntry entry, InvocationContext ctx, boolean skipOwnershipCheck) {
if (ctx.isInTxScope()) {
ClusteredRepeatableReadEntry clusterMvccEntry = (ClusteredRepeatableReadEntry) entry;
EntryVersion existingVersion = clusterMvccEntry.getVersion();
EntryVersion newVersion;
if (existingVersion == null) {
newVersion = versionGenerator.generateNew();
} else {
newVersion = versionGenerator.increment((IncrementableEntryVersion) existingVersion);
}
commitEntry(entry, newVersion, skipOwnershipCheck);
} else {
// This could be a state transfer call!
commitEntry(entry, entry.getVersion(), skipOwnershipCheck);
}
}
}
private static class HangingCacheManager extends AbstractDelegatingEmbeddedCacheManager {
static Log log = LogFactory.getLog(HangingCacheManager.class);
final CountDownLatch latch;
public HangingCacheManager(EmbeddedCacheManager delegate, CountDownLatch latch) {
super(delegate);
this.latch = latch;
}
@Override
public <K, V> Cache<K, V> getCache(String cacheName) {
log.info("Retrieve cache from hanging cache manager");
// TODO: Hacky but it's the easiest thing to do - consider ByteMan
// ByteMan apparently supports testng since 1.5.1 but no clear
// example out there, with more time it should be considered.
String threadName = Thread.currentThread().getName();
if (threadName.startsWith("HotRod")) {
log.info("Thread is a HotRod server worker thread, so force wait");
try {
// Wait a max of 3 minutes, otherwise socket timeout's not working
latch.await(180, TimeUnit.SECONDS);
log.info("Wait finished, return the cache");
return super.getCache();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
throw new CacheException(e);
}
}
return super.getCache(cacheName);
}
}
public static class CustomPojo implements Serializable {
static final Log log = LogFactory.getLog(CustomPojo.class);
private String name;
public CustomPojo(String name) {
this.name = name;
}
@Override
public boolean equals(Object obj) {
if (obj == null) {
log.debug("null -> false");
return false;
}
log.debug(obj.getClass());
if (getClass() != obj.getClass()) {
log.debug("class not same -> false");
return false;
}
final CustomPojo other = (CustomPojo) obj;
return this.name.equals(other.name);
}
@Override
public int hashCode() {
return name.hashCode();
}
}
/** * @author [email protected] * @since 4.0 */ public class EvictCommand extends RemoveCommand implements LocalCommand { private static final Log log = LogFactory.getLog(EvictCommand.class); public EvictCommand( Object key, CacheNotifier notifier, Set<Flag> flags, CommandInvocationId commandInvocationId) { super(key, null, notifier, flags, null, commandInvocationId); } @Override public Object acceptVisitor(InvocationContext ctx, Visitor visitor) throws Throwable { return visitor.visitEvictCommand(ctx, this); } @Override public Object perform(InvocationContext ctx) throws Throwable { if (key == null) { throw new NullPointerException("Key is null!!"); } super.perform(ctx); return null; } @Override public void notify( InvocationContext ctx, Object value, Metadata previousMetadata, boolean isPre) { // Eviction has no notion of pre/post event since 4.2.0.ALPHA4. // EvictionManagerImpl.onEntryEviction() triggers both pre and post events // with non-null values, so we should do the same here as an ugly workaround. if (!isPre) { if (log.isTraceEnabled()) log.tracef("Notify eviction listeners for key=%", key); notifier.notifyCacheEntryEvicted(key, value, ctx, this); } } @Override public byte getCommandId() { return -1; // these are not meant for replication! } @Override public String toString() { return new StringBuilder() .append("EvictCommand{key=") .append(key) .append(", value=") .append(value) .append(", flags=") .append(flags) .append("}") .toString(); } }
/**
* {@link CacheResult} interceptor implementation. This interceptor uses the following algorithm
* describes in JSR-107.
*
* <p>When a method annotated with {@link CacheResult} is invoked the following must occur.
*
* <ol>
* <li>Generate a key based on InvocationContext using the specified {@linkplain
* CacheKeyGenerator}.
* <li>Use this key to look up the entry in the cache.
* <li>If an entry is found return it as the result and do not call the annotated method.
* <li>If no entry is found invoke the method.
* <li>Use the result to populate the cache with this key/result pair.
* </ol>
*
* There is a skipGet attribute which if set to true will cause the method body to always be invoked
* and the return value put into the cache. The cache is not checked for the key before method body
* invocation, skipping steps 2 and 3 from the list above. This can be used for annotating methods
* that do a cache.put() with no other consequences.
*
* @author Kevin Pollet <*****@*****.**> (C) 2011 SERLI
*/
@Interceptor
public class CacheResultInterceptor implements Serializable {
private static final long serialVersionUID = 5275055951121834315L;
private static final Log log = LogFactory.getLog(CacheResultInterceptor.class, Log.class);
private final CacheResolver cacheResolver;
private final CacheKeyInvocationContextFactory contextFactory;
@Inject
public CacheResultInterceptor(
CacheResolver cacheResolver, CacheKeyInvocationContextFactory contextFactory) {
this.cacheResolver = cacheResolver;
this.contextFactory = contextFactory;
}
@AroundInvoke
public Object cacheResult(InvocationContext invocationContext) throws Exception {
if (log.isTraceEnabled()) {
log.tracef("Interception of method named '%s'", invocationContext.getMethod().getName());
}
final CacheKeyInvocationContext<CacheResult> cacheKeyInvocationContext =
contextFactory.getCacheKeyInvocationContext(invocationContext);
final CacheKeyGenerator cacheKeyGenerator =
cacheKeyInvocationContext
.unwrap(CacheKeyInvocationContextImpl.class)
.getCacheKeyGenerator();
final CacheResult cacheResult = cacheKeyInvocationContext.getCacheAnnotation();
final CacheKey cacheKey = cacheKeyGenerator.generateCacheKey(cacheKeyInvocationContext);
final Cache<CacheKey, Object> cache = cacheResolver.resolveCache(cacheKeyInvocationContext);
Object result = null;
if (!cacheResult.skipGet()) {
result = cache.get(cacheKey);
if (log.isTraceEnabled()) {
log.tracef(
"Entry with value '%s' has been found in cache '%s' with key '%s'",
result, cache.getName(), cacheKey);
}
}
if (result == null) {
result = invocationContext.proceed();
if (result != null) {
cache.put(cacheKey, result);
if (log.isTraceEnabled()) {
log.tracef(
"Value '%s' cached in cache '%s' with key '%s'", result, cache.getName(), cacheKey);
}
}
}
return result;
}
}
/**
* Interceptor that allows for waiting for a command to be invoked, blocking that command and
* subsequently allowing that command to be released.
*
* @author William Burns
* @since 6.0
*/
public class BlockingInterceptor extends CommandInterceptor {
private static final Log log = LogFactory.getLog(BlockingInterceptor.class);
private final CyclicBarrier barrier;
private final Class<? extends VisitableCommand> commandClass;
private final boolean blockAfter;
private final boolean originLocalOnly;
private final AtomicBoolean suspended = new AtomicBoolean();
public BlockingInterceptor(
CyclicBarrier barrier,
Class<? extends VisitableCommand> commandClass,
boolean blockAfter,
boolean originLocalOnly) {
this.barrier = barrier;
this.commandClass = commandClass;
this.blockAfter = blockAfter;
this.originLocalOnly = originLocalOnly;
}
public void suspend(boolean s) {
this.suspended.set(s);
}
private void blockIfNeeded(InvocationContext ctx, VisitableCommand command)
throws BrokenBarrierException, InterruptedException {
if (suspended.get()) {
log.tracef("Suspended, not blocking command %s", command);
return;
}
if (commandClass.equals(command.getClass()) && (!originLocalOnly || ctx.isOriginLocal())) {
log.tracef("Command blocking %s completion of %s", blockAfter ? "after" : "before", command);
// The first arrive and await is to sync with main thread
barrier.await();
// Now we actually block until main thread lets us go
barrier.await();
log.tracef("Command completed blocking completion of %s", command);
} else {
log.trace("Command arrived but already found a blocker");
}
}
@Override
protected Object handleDefault(InvocationContext ctx, VisitableCommand command) throws Throwable {
try {
if (!blockAfter) {
blockIfNeeded(ctx, command);
}
return super.handleDefault(ctx, command);
} finally {
if (blockAfter) {
blockIfNeeded(ctx, command);
}
}
}
}
private static int detectVersion() {
Log log = LogFactory.getLog(LuceneVersionDetector.class, Log.class);
int version = 3;
try {
Class.forName(
"org.apache.lucene.store.IOContext", true, LuceneVersionDetector.class.getClassLoader());
version = 4;
} catch (ClassNotFoundException e) {
}
log.detectedLuceneVersion(version);
return version;
}
/**
* LuceneUserThread: base class to perform activities on the index, as searching, adding to index
* and deleting.
*
* @author Sanne Grinovero
* @since 4.0
*/
public abstract class LuceneUserThread implements Runnable {
private static final Log log = LogFactory.getLog(LuceneUserThread.class);
protected final Directory directory;
protected final SharedState state;
LuceneUserThread(Directory dir, SharedState state) {
this.directory = dir;
this.state = state;
}
@Override
public final void run() {
try {
state.waitForStart();
} catch (InterruptedException e1) {
state.errorManage(e1);
return;
}
try {
beforeLoop();
} catch (IOException e) {
log.error("unexpected error", e);
state.errorManage(e);
}
while (!state.needToQuit()) {
try {
testLoop();
} catch (Exception e) {
log.error("unexpected error", e);
state.errorManage(e);
}
}
try {
cleanup();
} catch (IOException e) {
log.error("unexpected error", e);
state.errorManage(e);
}
}
protected void beforeLoop() throws IOException {
// defaults to no operation
}
protected abstract void testLoop() throws IOException;
protected void cleanup() throws IOException {
// defaults to no operation
}
}
/**
* A per-entry lock container for ReentrantLocks
*
* @author Manik Surtani
* @since 4.0
*/
public class ReentrantPerEntryLockContainer
extends AbstractPerEntryLockContainer<VisibleOwnerRefCountingReentrantLock> {
private static final Log log = LogFactory.getLog(ReentrantPerEntryLockContainer.class);
@Override
protected Log getLog() {
return log;
}
public ReentrantPerEntryLockContainer(int concurrencyLevel, Equivalence<Object> keyEquivalence) {
super(concurrencyLevel, keyEquivalence);
}
@Override
protected VisibleOwnerRefCountingReentrantLock newLock() {
return new VisibleOwnerRefCountingReentrantLock();
}
@Override
public boolean ownsLock(Object key, Object ignored) {
ReentrantLock l = getLockFromMap(key);
return l != null && l.isHeldByCurrentThread();
}
@Override
public boolean isLocked(Object key) {
ReentrantLock l = getLockFromMap(key);
return l != null && l.isLocked();
}
private ReentrantLock getLockFromMap(Object key) {
return locks.get(key);
}
@Override
protected void unlock(VisibleOwnerRefCountingReentrantLock l, Object unused) {
l.unlock();
}
@Override
protected boolean tryLock(
VisibleOwnerRefCountingReentrantLock lock, long timeout, TimeUnit unit, Object unused)
throws InterruptedException {
return lock.tryLock(timeout, unit);
}
@Override
protected void lock(VisibleOwnerRefCountingReentrantLock lock, Object lockOwner) {
lock.lock();
}
}
public final class DistributedTaskLifecycleService {
private static final Log log = LogFactory.getLog(DistributedTaskLifecycleService.class);
private static DistributedTaskLifecycleService service;
private ServiceLoader<DistributedTaskLifecycle> loader;
private DistributedTaskLifecycleService() {
loader = ServiceLoader.load(DistributedTaskLifecycle.class);
}
public static synchronized DistributedTaskLifecycleService getInstance() {
if (service == null) {
service = new DistributedTaskLifecycleService();
}
return service;
}
public <T> void onPreExecute(Callable<T> task) {
try {
Iterator<DistributedTaskLifecycle> i = loader.iterator();
while (i.hasNext()) {
DistributedTaskLifecycle cl = i.next();
cl.onPreExecute(task);
}
} catch (ServiceConfigurationError serviceError) {
log.errorReadingProperties(
new IOException(
"Could not properly load and instantiate DistributedTaskLifecycle service ",
serviceError));
}
}
public <T> void onPostExecute(Callable<T> task) {
try {
Iterator<DistributedTaskLifecycle> i = loader.iterator();
while (i.hasNext()) {
DistributedTaskLifecycle cl = i.next();
cl.onPostExecute(task);
}
} catch (ServiceConfigurationError serviceError) {
log.errorReadingProperties(
new IOException(
"Could not properly load and instantiate DistributedTaskLifecycle service ",
serviceError));
}
}
}
/**
* DistributedLazyIterator.
*
* <p>Lazily iterates on a distributed query
*
* @author Israel Lacerra <*****@*****.**>
* @since 5.1
*/
public class DistributedLazyIterator<E> extends DistributedIterator<E> {
private final UUID queryId;
private final ExecutorService asyncExecutor;
private final ClusteredQueryInvoker invoker;
private static final Log log = LogFactory.getLog(DistributedLazyIterator.class);
public DistributedLazyIterator(
Sort sort,
int fetchSize,
int resultSize,
int maxResults,
int firstResult,
UUID id,
HashMap<UUID, ClusteredTopDocs> topDocsResponses,
ExecutorService asyncExecutor,
AdvancedCache<?, ?> cache) {
super(sort, fetchSize, resultSize, maxResults, firstResult, topDocsResponses, cache);
this.queryId = id;
this.asyncExecutor = asyncExecutor;
this.invoker = new ClusteredQueryInvoker(cache, asyncExecutor);
}
@Override
public void close() {
ClusteredQueryCommand killQuery = ClusteredQueryCommand.destroyLazyQuery(cache, queryId);
ClusteredQueryInvoker invoker = new ClusteredQueryInvoker(cache, asyncExecutor);
try {
invoker.broadcast(killQuery);
} catch (Exception e) {
log.error("Could not close the distributed iterator", e);
}
}
@Override
protected E fetchValue(int scoreIndex, ClusteredTopDocs topDoc) {
Object value = null;
try {
value = invoker.getValue(scoreIndex, topDoc.getNodeAddress(), queryId);
} catch (Exception e) {
log.error("Error while trying to remoting fetch next value: " + e.getMessage());
}
return (E) value;
}
}
public static class PojoValue implements Externalizable {
static AtomicBoolean holdUp = new AtomicBoolean();
Log log = LogFactory.getLog(PojoValue.class);
volatile int value;
public PojoValue() {}
public PojoValue(int value) {
this.value = value;
}
@Override
public void writeExternal(ObjectOutput out) throws IOException {
String threadName = Thread.currentThread().getName();
if (!holdUp.get()) {
log.debug("In streaming...");
holdUp.compareAndSet(false, true);
log.debug("Holding up...");
TestingUtil.sleepThread(1000); // Sleep for 2 seconds to hold up state transfer
}
out.writeInt(value);
}
@Override
public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
value = in.readInt();
}
@Override
public int hashCode() {
return value + 31;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
PojoValue pojo = (PojoValue) o;
if (value != pojo.value) return false;
return true;
}
}
/** * Rpc to obtain all in-doubt prepared transactions stored on remote nodes. A transaction is in * doubt if it is prepared and the node where it started has crashed. * * @author [email protected] * @since 5.0 */ public class GetInDoubtTransactionsCommand extends RecoveryCommand { private static final Log log = LogFactory.getLog(GetInDoubtTransactionsCommand.class); public static final int COMMAND_ID = 21; private GetInDoubtTransactionsCommand() { super(null); // For command id uniqueness test } public GetInDoubtTransactionsCommand(String cacheName) { super(cacheName); } @Override public List<Xid> perform(InvocationContext ctx) throws Throwable { List<Xid> localInDoubtTransactions = recoveryManager.getInDoubtTransactions(); log.tracef("Returning result %s", localInDoubtTransactions); return localInDoubtTransactions; } @Override public byte getCommandId() { return COMMAND_ID; } @Override public Object[] getParameters() { return Util.EMPTY_OBJECT_ARRAY; } @Override public void setParameters(int commandId, Object[] parameters) { if (commandId != COMMAND_ID) throw new IllegalStateException("Expected " + COMMAND_ID + "and received " + commandId); // No parameters } @Override public String toString() { return getClass().getSimpleName() + " { cacheName = " + cacheName + "}"; } }
/** * Rpc to obtain all in-doubt prepared transactions stored on remote nodes. A transaction is in * doubt if it is prepared and the node where it started has crashed. * * @author [email protected] * @since 5.0 */ public class GetInDoubtTransactionsCommand extends RecoveryCommand { private static Log log = LogFactory.getLog(GetInDoubtTransactionsCommand.class); public static final int COMMAND_ID = Ids.GET_IN_DOUBT_TX_COMMAND; public GetInDoubtTransactionsCommand() {} public GetInDoubtTransactionsCommand(String cacheName) { this.cacheName = cacheName; } @Override public List<Xid> perform(InvocationContext ctx) throws Throwable { List<Xid> localInDoubtTransactions = recoveryManager.getLocalInDoubtTransactions(); if (log.isTraceEnabled()) log.trace("Returning result %s", localInDoubtTransactions); return localInDoubtTransactions; } @Override public byte getCommandId() { return COMMAND_ID; } @Override public Object[] getParameters() { return new Object[] {cacheName}; } @Override public void setParameters(int commandId, Object[] parameters) { if (commandId != COMMAND_ID) throw new IllegalStateException("Expected " + COMMAND_ID + "and received " + commandId); cacheName = (String) parameters[0]; } @Override public String toString() { return getClass().getSimpleName() + " { cacheName = " + cacheName + "}"; } }
/**
* This class is an interceptor that will index data only if it has come from a local source.
*
* <p>Currently, this is a property that is determined by setting "infinispan.query.indexLocalOnly"
* as a System property to "true".
*
* @author Navin Surtani
* @since 4.0
*/
public class LocalQueryInterceptor extends QueryInterceptor {
private static final Log log = LogFactory.getLog(LocalQueryInterceptor.class, Log.class);
public LocalQueryInterceptor(SearchFactoryIntegrator searchFactory) {
super(searchFactory);
}
@Override
protected Log getLog() {
return log;
}
@Override
protected boolean shouldModifyIndexes(FlagAffectedCommand command, InvocationContext ctx) {
// will index only local updates that were not flagged with SKIP_INDEXING and are not caused
// internally by state transfer
return ctx.isOriginLocal()
&& !command.hasFlag(Flag.SKIP_INDEXING)
&& !command.hasFlag(Flag.PUT_FOR_STATE_TRANSFER);
}
}
/**
* An extension of {@link ReadCommittedEntry} that provides Repeatable Read semantics
*
* @author Manik Surtani (<a href="mailto:[email protected]">[email protected]</a>)
* @since 4.0
*/
public class RepeatableReadEntry extends ReadCommittedEntry {
private static final Log log = LogFactory.getLog(RepeatableReadEntry.class);
public RepeatableReadEntry(Object key, Object value, long lifespan) {
super(key, value, lifespan);
}
@Override
public void copyForUpdate(DataContainer container, boolean writeSkewCheck) {
if (isChanged()) return; // already copied
// mark entry as changed.
setChanged();
if (writeSkewCheck) {
// check for write skew.
InternalCacheEntry ice = container.get(key);
Object actualValue = ice == null ? null : ice.getValue();
// Note that this identity-check is intentional. We don't *want* to call actualValue.equals()
// since that defeats the purpose.
// the implicit "versioning" we have in R_R creates a new wrapper "value" instance for every
// update.
if (actualValue != null && actualValue != value) {
String errormsg =
new StringBuilder()
.append("Detected write skew on key [")
.append(getKey())
.append("]. Another process has changed the entry since we last read it!")
.toString();
if (log.isWarnEnabled()) log.warn(errormsg + ". Unable to copy entry for update.");
throw new CacheException(errormsg);
}
}
// make a backup copy
oldValue = value;
}
}
/**
* @author Mircea Markus
* @since 5.2
*/
public class BaseBackupInterceptor extends CommandInterceptor {
protected BackupSender backupSender;
protected TransactionTable txTable;
private static final Log log = LogFactory.getLog(BaseBackupInterceptor.class);
@Inject
void init(BackupSender sender, TransactionTable txTable) {
this.backupSender = sender;
this.txTable = txTable;
}
protected boolean isTxFromRemoteSite(GlobalTransaction gtx) {
LocalTransaction remoteTx = txTable.getLocalTransaction(gtx);
return remoteTx != null && remoteTx.isFromRemoteSite();
}
protected boolean shouldInvokeRemoteTxCommand(TxInvocationContext ctx) {
// ISPN-2362: For backups, we should only replicate to the remote site if there are
// modifications to replay.
boolean shouldBackupRemotely =
ctx.isOriginLocal()
&& ctx.hasModifications()
&& !ctx.getCacheTransaction().isFromStateTransfer();
getLog().tracef("Should backup remotely? %s", shouldBackupRemotely);
return shouldBackupRemotely;
}
protected final boolean skipXSiteBackup(FlagAffectedCommand command) {
return command.hasFlag(Flag.SKIP_XSITE_BACKUP);
}
@Override
protected Log getLog() {
return log;
}
}
/**
* Tests the interceptor chain and surrounding logic
*
* @author Manik Surtani
*/
@Test(groups = "functional", testName = "loaders.CacheLoaderFunctionalTest")
public class CacheLoaderFunctionalTest extends AbstractInfinispanTest {
private static final Log log = LogFactory.getLog(CacheLoaderFunctionalTest.class);
Cache cache;
CacheStore store;
TransactionManager tm;
Configuration cfg;
EmbeddedCacheManager cm;
long lifespan = 60000000; // very large lifespan so nothing actually expires
@BeforeTest
public void setUp() {
cfg =
new Configuration()
.fluent()
.loaders()
.addCacheLoader(
new DummyInMemoryCacheStore.Cfg()
.storeName(this.getClass().getName())) // in order to use the same store
.transaction()
.transactionMode(TransactionMode.TRANSACTIONAL)
.build();
cm = TestCacheManagerFactory.createCacheManager(cfg);
cache = cm.getCache();
store = TestingUtil.extractComponent(cache, CacheLoaderManager.class).getCacheStore();
tm = TestingUtil.getTransactionManager(cache);
}
@AfterTest
public void tearDown() {
TestingUtil.killCacheManagers(cm);
cache = null;
cm = null;
cfg = null;
tm = null;
store = null;
}
@AfterMethod
public void afterMethod() throws CacheLoaderException {
if (cache != null) cache.clear();
if (store != null) store.clear();
}
private void assertInCacheAndStore(Object key, Object value) throws CacheLoaderException {
assertInCacheAndStore(key, value, -1);
}
private void assertInCacheAndStore(Object key, Object value, long lifespanMillis)
throws CacheLoaderException {
assertInCacheAndStore(cache, store, key, value, lifespanMillis);
}
private void assertInCacheAndStore(Cache cache, CacheStore store, Object key, Object value)
throws CacheLoaderException {
assertInCacheAndStore(cache, store, key, value, -1);
}
private void assertInCacheAndStore(
Cache cache, CacheStore store, Object key, Object value, long lifespanMillis)
throws CacheLoaderException {
InternalCacheEntry se = cache.getAdvancedCache().getDataContainer().get(key, null);
testStoredEntry(se, value, lifespanMillis, "Cache", key);
se = store.load(key);
testStoredEntry(se, value, lifespanMillis, "Store", key);
}
private void testStoredEntry(
InternalCacheEntry entry,
Object expectedValue,
long expectedLifespan,
String src,
Object key) {
assert entry != null : src + " entry for key " + key + " should NOT be null";
assert entry.getValue().equals(expectedValue)
: src
+ " should contain value "
+ expectedValue
+ " under key "
+ entry.getKey()
+ " but was "
+ entry.getValue()
+ ". Entry is "
+ entry;
assert entry.getLifespan() == expectedLifespan
: src
+ " expected lifespan for key "
+ key
+ " to be "
+ expectedLifespan
+ " but was "
+ entry.getLifespan()
+ ". Entry is "
+ entry;
}
private void assertNotInCacheAndStore(Cache cache, CacheStore store, Object... keys)
throws CacheLoaderException {
for (Object key : keys) {
assert !cache.getAdvancedCache().getDataContainer().containsKey(key, null)
: "Cache should not contain key " + key;
assert !store.containsKey(key) : "Store should not contain key " + key;
}
}
private void assertNotInCacheAndStore(Object... keys) throws CacheLoaderException {
assertNotInCacheAndStore(cache, store, keys);
}
private void assertInStoreNotInCache(Object... keys) throws CacheLoaderException {
assertInStoreNotInCache(cache, store, keys);
}
private void assertInStoreNotInCache(Cache cache, CacheStore store, Object... keys)
throws CacheLoaderException {
for (Object key : keys) {
assert !cache.getAdvancedCache().getDataContainer().containsKey(key, null)
: "Cache should not contain key " + key;
assert store.containsKey(key) : "Store should contain key " + key;
}
}
private void assertInCacheAndNotInStore(Object... keys) throws CacheLoaderException {
assertInCacheAndNotInStore(cache, store, keys);
}
private void assertInCacheAndNotInStore(Cache cache, CacheStore store, Object... keys)
throws CacheLoaderException {
for (Object key : keys) {
assert cache.getAdvancedCache().getDataContainer().containsKey(key, null)
: "Cache should not contain key " + key;
assert !store.containsKey(key) : "Store should contain key " + key;
}
}
public void testStoreAndRetrieve() throws CacheLoaderException {
assertNotInCacheAndStore("k1", "k2", "k3", "k4", "k5", "k6", "k7");
cache.put("k1", "v1");
cache.put("k2", "v2", lifespan, MILLISECONDS);
cache.putAll(Collections.singletonMap("k3", "v3"));
cache.putAll(Collections.singletonMap("k4", "v4"), lifespan, MILLISECONDS);
cache.putIfAbsent("k5", "v5");
cache.putIfAbsent("k6", "v6", lifespan, MILLISECONDS);
cache.putIfAbsent("k5", "v5-SHOULD-NOT-PUT");
cache.putIfAbsent("k6", "v6-SHOULD-NOT-PUT", lifespan, MILLISECONDS);
cache.putForExternalRead("k7", "v7");
cache.putForExternalRead("k7", "v7-SHOULD-NOT-PUT");
for (int i = 1; i < 8; i++) {
// even numbers have lifespans
if (i % 2 == 1) assertInCacheAndStore("k" + i, "v" + i);
else assertInCacheAndStore("k" + i, "v" + i, lifespan);
}
assert !cache.remove("k1", "some rubbish");
for (int i = 1; i < 8; i++) {
// even numbers have lifespans
if (i % 2 == 1) assertInCacheAndStore("k" + i, "v" + i);
else assertInCacheAndStore("k" + i, "v" + i, lifespan);
}
log.info("cache.get(\"k1\") = " + cache.get("k1"));
assert cache.remove("k1", "v1");
log.info("cache.get(\"k1\") = " + cache.get("k1"));
assert cache.remove("k2").equals("v2");
assertNotInCacheAndStore("k1", "k2");
for (int i = 3; i < 8; i++) {
// even numbers have lifespans
if (i % 2 == 1) assertInCacheAndStore("k" + i, "v" + i);
else assertInCacheAndStore("k" + i, "v" + i, lifespan);
}
cache.clear();
assertNotInCacheAndStore("k1", "k2", "k3", "k4", "k5", "k6", "k7");
}
public void testReplaceMethods() throws CacheLoaderException {
assertNotInCacheAndStore("k1", "k2", "k3", "k4");
cache.replace("k1", "v1-SHOULD-NOT-STORE");
assertNoLocks(cache);
cache.replace("k2", "v2-SHOULD-NOT-STORE", lifespan, MILLISECONDS);
assertNoLocks(cache);
assertNotInCacheAndStore("k1", "k2", "k3", "k4");
cache.put("k1", "v1");
assertNoLocks(cache);
cache.put("k2", "v2");
assertNoLocks(cache);
cache.put("k3", "v3");
assertNoLocks(cache);
cache.put("k4", "v4");
assertNoLocks(cache);
for (int i = 1; i < 5; i++) assertInCacheAndStore("k" + i, "v" + i);
cache.replace("k1", "v1-SHOULD-NOT-STORE", "v1-STILL-SHOULD-NOT-STORE");
assertNoLocks(cache);
cache.replace("k2", "v2-SHOULD-NOT-STORE", "v2-STILL-SHOULD-NOT-STORE", lifespan, MILLISECONDS);
assertNoLocks(cache);
for (int i = 1; i < 5; i++) assertInCacheAndStore("k" + i, "v" + i);
cache.replace("k1", "v1-REPLACED");
assertNoLocks(cache);
cache.replace("k2", "v2-REPLACED", lifespan, MILLISECONDS);
assertInCacheAndStore("k2", "v2-REPLACED", lifespan);
assertNoLocks(cache);
cache.replace("k3", "v3", "v3-REPLACED");
assertNoLocks(cache);
cache.replace("k4", "v4", "v4-REPLACED", lifespan, MILLISECONDS);
assertNoLocks(cache);
for (int i = 1; i < 5; i++) {
// even numbers have lifespans
if (i % 2 == 1) assertInCacheAndStore("k" + i, "v" + i + "-REPLACED");
else assertInCacheAndStore("k" + i, "v" + i + "-REPLACED", lifespan);
}
assertNoLocks(cache);
}
public void testLoading() throws CacheLoaderException {
assertNotInCacheAndStore("k1", "k2", "k3", "k4");
for (int i = 1; i < 5; i++) store.store(TestInternalCacheEntryFactory.create("k" + i, "v" + i));
for (int i = 1; i < 5; i++) assert cache.get("k" + i).equals("v" + i);
// make sure we have no stale locks!!
assertNoLocks(cache);
for (int i = 1; i < 5; i++) cache.evict("k" + i);
// make sure we have no stale locks!!
assertNoLocks(cache);
assert cache.putIfAbsent("k1", "v1-SHOULD-NOT-STORE").equals("v1");
assert cache.remove("k2").equals("v2");
assert cache.replace("k3", "v3-REPLACED").equals("v3");
assert cache.replace("k4", "v4", "v4-REPLACED");
// make sure we have no stale locks!!
assertNoLocks(cache);
assert cache.size() == 3
: "Expected the cache to contain 3 elements but contained " + cache.size();
for (int i = 1; i < 5; i++) cache.evict("k" + i);
// make sure we have no stale locks!!
assertNoLocks(cache);
assert cache.isEmpty(); // cache size ops will not trigger a load
cache.clear(); // this should propagate to the loader though
assertNotInCacheAndStore("k1", "k2", "k3", "k4");
// make sure we have no stale locks!!
assertNoLocks(cache);
}
public void testPreloading() throws CacheLoaderException {
Configuration preloadingCfg = cfg.clone();
preloadingCfg.getCacheLoaderManagerConfig().setPreload(true);
((DummyInMemoryCacheStore.Cfg)
preloadingCfg.getCacheLoaderManagerConfig().getFirstCacheLoaderConfig())
.setStoreName("preloadingCache");
cm.defineConfiguration("preloadingCache", preloadingCfg);
Cache preloadingCache = cm.getCache("preloadingCache");
CacheStore preloadingStore =
TestingUtil.extractComponent(preloadingCache, CacheLoaderManager.class).getCacheStore();
assert preloadingCache.getConfiguration().getCacheLoaderManagerConfig().isPreload();
assertNotInCacheAndStore(preloadingCache, preloadingStore, "k1", "k2", "k3", "k4");
preloadingCache.put("k1", "v1");
preloadingCache.put("k2", "v2", lifespan, MILLISECONDS);
preloadingCache.put("k3", "v3");
preloadingCache.put("k4", "v4", lifespan, MILLISECONDS);
for (int i = 1; i < 5; i++) {
if (i % 2 == 1) assertInCacheAndStore(preloadingCache, preloadingStore, "k" + i, "v" + i);
else assertInCacheAndStore(preloadingCache, preloadingStore, "k" + i, "v" + i, lifespan);
}
DataContainer c = preloadingCache.getAdvancedCache().getDataContainer();
assert c.size(null) == 4;
preloadingCache.stop();
assert c.size(null) == 0;
preloadingCache.start();
assert preloadingCache.getConfiguration().getCacheLoaderManagerConfig().isPreload();
c = preloadingCache.getAdvancedCache().getDataContainer();
assert c.size(null) == 4;
for (int i = 1; i < 5; i++) {
if (i % 2 == 1) assertInCacheAndStore(preloadingCache, preloadingStore, "k" + i, "v" + i);
else assertInCacheAndStore(preloadingCache, preloadingStore, "k" + i, "v" + i, lifespan);
}
}
public void testPurgeOnStartup() throws CacheLoaderException {
Configuration purgingCfg = cfg.clone();
CacheStoreConfig firstCacheLoaderConfig =
(CacheStoreConfig) purgingCfg.getCacheLoaderManagerConfig().getFirstCacheLoaderConfig();
firstCacheLoaderConfig.setPurgeOnStartup(true);
((DummyInMemoryCacheStore.Cfg)
purgingCfg.getCacheLoaderManagerConfig().getFirstCacheLoaderConfig())
.setStoreName("purgingCache");
cm.defineConfiguration("purgingCache", purgingCfg);
Cache purgingCache = cm.getCache("purgingCache");
CacheStore purgingStore =
TestingUtil.extractComponent(purgingCache, CacheLoaderManager.class).getCacheStore();
assertNotInCacheAndStore(purgingCache, purgingStore, "k1", "k2", "k3", "k4");
purgingCache.put("k1", "v1");
purgingCache.put("k2", "v2", lifespan, MILLISECONDS);
purgingCache.put("k3", "v3");
purgingCache.put("k4", "v4", lifespan, MILLISECONDS);
for (int i = 1; i < 5; i++) {
if (i % 2 == 1) assertInCacheAndStore(purgingCache, purgingStore, "k" + i, "v" + i);
else assertInCacheAndStore(purgingCache, purgingStore, "k" + i, "v" + i, lifespan);
}
DataContainer c = purgingCache.getAdvancedCache().getDataContainer();
assert c.size(null) == 4;
purgingCache.stop();
assert c.size(null) == 0;
purgingCache.start();
c = purgingCache.getAdvancedCache().getDataContainer();
assert c.size(null) == 0;
assertNotInCacheAndStore(purgingCache, purgingStore, "k1", "k2", "k3", "k4");
}
public void testTransactionalWrites() throws Exception {
assert cache.getStatus() == ComponentStatus.RUNNING;
assertNotInCacheAndStore("k1", "k2");
tm.begin();
cache.put("k1", "v1");
cache.put("k2", "v2", lifespan, MILLISECONDS);
Transaction t = tm.suspend();
assertNotInCacheAndStore("k1", "k2");
tm.resume(t);
tm.commit();
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2", lifespan);
tm.begin();
cache.clear();
t = tm.suspend();
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2", lifespan);
tm.resume(t);
tm.commit();
assertNotInCacheAndStore("k1", "k2");
tm.begin();
cache.put("k1", "v1");
cache.put("k2", "v2", lifespan, MILLISECONDS);
t = tm.suspend();
assertNotInCacheAndStore("k1", "k2");
tm.resume(t);
tm.rollback();
assertNotInCacheAndStore("k1", "k2");
cache.put("k1", "v1");
cache.put("k2", "v2", lifespan, MILLISECONDS);
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2", lifespan);
tm.begin();
cache.clear();
t = tm.suspend();
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2", lifespan);
tm.resume(t);
tm.rollback();
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2", lifespan);
}
public void testTransactionalReplace(Method m) throws Exception {
assert cache.getStatus() == ComponentStatus.RUNNING;
assertNotInCacheAndStore(k(m, 1));
assertNotInCacheAndStore(k(m, 2));
cache.put(k(m, 2), v(m));
tm.begin();
cache.put(k(m, 1), v(m, 1));
cache.replace(k(m, 2), v(m, 1));
Transaction t = tm.suspend();
assertNotInCacheAndStore(k(m, 1));
assertInCacheAndStore(k(m, 2), v(m));
tm.resume(t);
tm.commit();
assertInCacheAndStore(k(m, 1), v(m, 1));
assertInCacheAndStore(k(m, 2), v(m, 1));
}
public void testEvictAndRemove() throws CacheLoaderException {
assertNotInCacheAndStore("k1", "k2");
cache.put("k1", "v1");
cache.put("k2", "v2", lifespan, MILLISECONDS);
cache.evict("k1");
cache.evict("k2");
assert "v1".equals(cache.remove("k1"));
assert "v2".equals(cache.remove("k2"));
}
public void testLoadingToMemory() throws CacheLoaderException {
assertNotInCacheAndStore("k1", "k2");
store.store(TestInternalCacheEntryFactory.create("k1", "v1"));
store.store(TestInternalCacheEntryFactory.create("k2", "v2"));
assertInStoreNotInCache("k1", "k2");
assert "v1".equals(cache.get("k1"));
assert "v2".equals(cache.get("k2"));
assertInCacheAndStore("k1", "v1");
assertInCacheAndStore("k2", "v2");
store.remove("k1");
store.remove("k2");
assertInCacheAndNotInStore("k1", "k2");
assert "v1".equals(cache.get("k1"));
assert "v2".equals(cache.get("k2"));
}
public void testSkipLocking(Method m) {
String name = m.getName();
AdvancedCache advancedCache = cache.getAdvancedCache();
advancedCache.put("k-" + name, "v-" + name);
advancedCache.withFlags(Flag.SKIP_LOCKING).put("k-" + name, "v2-" + name);
}
public void testDuplicatePersistence(Method m) throws Exception {
String key = "k-" + m.getName();
String value = "v-" + m.getName();
cache.put(key, value);
assert value.equals(cache.get(key));
cache.stop();
cache.start();
tm.begin();
cache.containsKey(key); // Necessary call to force locks being acquired in advance
cache.getAdvancedCache().withFlags(Flag.FORCE_WRITE_LOCK).get(key);
cache.put(key, value);
tm.commit();
assert value.equals(cache.get(key));
}
public void testGetCacheLoadersFromConfigAfterStart() {
cache.getConfiguration().getCacheLoaders();
cache.getConfiguration().getCacheLoaders();
}
}
/**
* Parent tests for both transactional and read-only tests are defined in this class.
*
* @author Galder Zamarreño
* @since 4.1
*/
public abstract class AbstractFunctionalTestCase extends SingleNodeTestCase {
static final Log log = LogFactory.getLog(AbstractFunctionalTestCase.class);
@Test
public void testEmptySecondLevelCacheEntry() throws Exception {
sessionFactory().getCache().evictCollectionRegion(Item.class.getName() + ".items");
Statistics stats = sessionFactory().getStatistics();
stats.clear();
SecondLevelCacheStatistics statistics =
stats.getSecondLevelCacheStatistics(Item.class.getName() + ".items");
Map cacheEntries = statistics.getEntries();
assertEquals(0, cacheEntries.size());
}
@Test
public void testInsertDeleteEntity() throws Exception {
final Statistics stats = sessionFactory().getStatistics();
stats.clear();
final Item item = new Item("chris", "Chris's Item");
withTx(
tm,
new Callable<Void>() {
@Override
public Void call() throws Exception {
Session s = openSession();
s.getTransaction().begin();
s.persist(item);
s.getTransaction().commit();
s.close();
return null;
}
});
log.info("Entry persisted, let's load and delete it.");
withTx(
tm,
new Callable<Void>() {
@Override
public Void call() throws Exception {
Session s = openSession();
s.getTransaction().begin();
Item found = (Item) s.load(Item.class, item.getId());
log.info(stats.toString());
assertEquals(item.getDescription(), found.getDescription());
assertEquals(0, stats.getSecondLevelCacheMissCount());
assertEquals(1, stats.getSecondLevelCacheHitCount());
s.delete(found);
s.getTransaction().commit();
s.close();
return null;
}
});
}
@Test
public void testInsertClearCacheDeleteEntity() throws Exception {
final Statistics stats = sessionFactory().getStatistics();
stats.clear();
final Item item = new Item("chris", "Chris's Item");
withTx(
tm,
new Callable<Void>() {
@Override
public Void call() throws Exception {
Session s = openSession();
s.getTransaction().begin();
s.persist(item);
s.getTransaction().commit();
assertEquals(0, stats.getSecondLevelCacheMissCount());
assertEquals(0, stats.getSecondLevelCacheHitCount());
assertEquals(1, stats.getSecondLevelCachePutCount());
s.close();
return null;
}
});
log.info("Entry persisted, let's load and delete it.");
cleanupCache();
withTx(
tm,
new Callable<Void>() {
@Override
public Void call() throws Exception {
Session s = openSession();
s.getTransaction().begin();
Item found = (Item) s.load(Item.class, item.getId());
log.info(stats.toString());
assertEquals(item.getDescription(), found.getDescription());
assertEquals(1, stats.getSecondLevelCacheMissCount());
assertEquals(0, stats.getSecondLevelCacheHitCount());
assertEquals(2, stats.getSecondLevelCachePutCount());
s.delete(found);
s.getTransaction().commit();
s.close();
return null;
}
});
}
}
/**
* RhqPluginDoclet.
*
* @author Galder Zamarreño
* @since 4.0
*/
public class RhqPluginXmlGenerator {
private static final Log log = LogFactory.getLog(RhqPluginXmlGenerator.class);
private static ClassPool classPool;
private static String cp;
public static void main(String[] args) throws Exception {
cp = System.getProperty("java.class.path");
start(null);
}
public static boolean validOptions(String options[][], DocErrorReporter reporter) {
for (String[] option : options) {
if (option[0].equals("-classpath")) cp = option[1];
}
return true;
}
public static boolean start(RootDoc rootDoc) throws Exception {
List<Class<?>> mbeanIspnClasses = getMBeanClasses();
List<Class<?>> globalClasses = new ArrayList<Class<?>>();
List<Class<?>> namedCacheClasses = new ArrayList<Class<?>>();
for (Class<?> clazz : mbeanIspnClasses) {
Scope scope = clazz.getAnnotation(Scope.class);
if (scope != null && scope.value() == Scopes.GLOBAL) {
debug("Add as global class " + clazz);
globalClasses.add(clazz);
} else {
debug("Add as named cache class " + clazz);
namedCacheClasses.add(clazz);
}
}
// Init the Javassist class pool.
classPool = ClassPool.getDefault();
classPool.insertClassPath(new ClassClassPath(RhqPluginXmlGenerator.class));
PluginGen pg = new PluginGen();
Props root = new Props();
root.setPluginName("Infinispan");
root.setPluginDescription("Supports management and monitoring of Infinispan");
root.setName("Infinispan Cache Manager");
root.setPkg("org.infinispan.rhq");
root.setDependsOnJmxPlugin(true);
root.setDiscoveryClass("CacheManagerDiscovery");
root.setComponentClass("CacheManagerComponent");
root.setSingleton(false);
root.setCategory(ResourceCategory.SERVICE);
Set<TypeKey> servers = new HashSet<TypeKey>();
servers.add(new TypeKey("JMX Server", "JMX"));
servers.add(new TypeKey("JBossAS Server", "JBossAS"));
servers.add(new TypeKey("JBossAS Server", "JBossAS5"));
root.setRunsInsides(servers);
populateMetricsAndOperations(globalClasses, root, false);
Props cache = new Props();
cache.setName("Infinispan Cache");
cache.setPkg("org.infinispan.rhq");
cache.setDependsOnJmxPlugin(true);
cache.setDiscoveryClass("CacheDiscovery");
cache.setComponentClass("CacheComponent");
cache.setSingleton(false);
cache.setCategory(ResourceCategory.SERVICE);
populateMetricsAndOperations(namedCacheClasses, cache, true);
root.getChildren().add(cache);
String metaInfDir = "../../../src/main/resources/META-INF";
new File(metaInfDir).mkdirs();
String targetMetaInfDir = "../../../target/classes/META-INF";
new File(targetMetaInfDir).mkdirs();
pg.createFile(root, "descriptor", "rhq-plugin.xml", metaInfDir);
copyFile(
new File(metaInfDir + "/rhq-plugin.xml"), new File(targetMetaInfDir + "/rhq-plugin.xml"));
return true;
}
private static void copyFile(File in, File out) throws IOException {
FileChannel inCh = new FileInputStream(in).getChannel();
FileChannel outCh = new FileOutputStream(out).getChannel();
try {
inCh.transferTo(0, inCh.size(), outCh);
} catch (IOException e) {
throw e;
} finally {
if (inCh != null) inCh.close();
if (outCh != null) outCh.close();
}
}
private static List<Class<?>> getMBeanClasses() throws IOException {
try {
return ClassFinder.withAnnotationDeclared(ClassFinder.infinispanClasses(cp), MBean.class);
} catch (Exception e) {
IOException ioe = new IOException("Unable to get Infinispan classes");
ioe.initCause(e);
throw ioe;
}
}
private static void populateMetricsAndOperations(
List<Class<?>> classes, Props props, boolean withNamePrefix) throws Exception {
props.setHasOperations(true);
props.setHasMetrics(true);
for (Class<?> clazz : classes) {
MBean mbean = clazz.getAnnotation(MBean.class);
String prefix = withNamePrefix ? mbean.objectName() + '.' : "";
CtClass ctClass = classPool.get(clazz.getName());
CtMethod[] ctMethods = ctClass.getMethods();
for (CtMethod ctMethod : ctMethods) {
ManagedAttribute managedAttr =
(ManagedAttribute) ctMethod.getAnnotation(ManagedAttribute.class);
ManagedOperation managedOp =
(ManagedOperation) ctMethod.getAnnotation(ManagedOperation.class);
Metric rhqMetric = (Metric) ctMethod.getAnnotation(Metric.class);
if (rhqMetric != null) {
debug("Metric annotation found " + rhqMetric);
// Property and description resolution are the reason why annotation scanning is done
// here.
// These two fields are calculated from either the method name or the Managed*
// annotations,
// and so, only the infinispan side knows about that.
String property = prefix + getPropertyFromBeanConvention(ctMethod);
if (!rhqMetric.property().isEmpty()) {
property = prefix + rhqMetric.property();
}
MetricProps metric = new MetricProps(property);
String displayName =
withNamePrefix
? "[" + mbean.objectName() + "] " + rhqMetric.displayName()
: rhqMetric.displayName();
metric.setDisplayName(displayName);
metric.setDisplayType(rhqMetric.displayType());
metric.setDataType(rhqMetric.dataType());
metric.setUnits(rhqMetric.units());
if (managedAttr != null) {
debug("Metric has ManagedAttribute annotation " + managedAttr);
metric.setDescription(managedAttr.description());
} else if (managedOp != null) {
debug("Metric has ManagedOperation annotation " + managedOp);
metric.setDescription(managedOp.description());
} else {
log.debug(
"Metric has no managed annotations, so take the description from the display name.");
metric.setDescription(rhqMetric.displayName());
}
props.getMetrics().add(metric);
}
Operation rhqOperation = (Operation) ctMethod.getAnnotation(Operation.class);
if (rhqOperation != null) {
debug("Operation annotation found " + rhqOperation);
String name;
if (!rhqOperation.name().isEmpty()) {
name = prefix + rhqOperation.name();
} else {
name = prefix + ctMethod.getName();
}
OperationProps operation = new OperationProps(name);
String displayName =
withNamePrefix
? "[" + mbean.objectName() + "] " + rhqOperation.displayName()
: rhqOperation.displayName();
operation.setDisplayName(displayName);
if (managedAttr != null) {
debug("Operation has ManagedAttribute annotation " + managedAttr);
operation.setDescription(managedAttr.description());
} else if (managedOp != null) {
debug("Operation has ManagedOperation annotation " + managedOp);
operation.setDescription(managedOp.description());
} else {
debug(
"Operation has no managed annotations, so take the description from the display name.");
operation.setDescription(rhqOperation.displayName());
}
Object[][] paramAnnotations = ctMethod.getParameterAnnotations();
int i = 0;
for (Object[] paramAnnotationsInEach : paramAnnotations) {
boolean hadParameter = false;
for (Object annot : paramAnnotationsInEach) {
debug("Parameter annotation " + annot);
if (annot instanceof Parameter) {
Parameter param = (Parameter) annot;
SimpleProperty prop = new SimpleProperty(param.name());
prop.setDescription(param.description());
operation.getParams().add(prop);
hadParameter = true;
}
}
if (!hadParameter) {
operation.getParams().add(new SimpleProperty("p" + i++));
}
}
CtClass returnType = ctMethod.getReturnType();
if (!returnType.equals(CtClass.voidType)) {
if (!returnType.equals(Void.TYPE)) {
SimpleProperty prop = new SimpleProperty("operationResult");
operation.setResult(prop);
}
}
props.getOperations().add(operation);
}
}
CtField[] ctFields = ctClass.getDeclaredFields();
for (CtField ctField : ctFields) {
debug("Inspecting field " + ctField);
Metric rhqMetric = (Metric) ctField.getAnnotation(Metric.class);
if (rhqMetric != null) {
debug("Field " + ctField + " contains Metric annotation " + rhqMetric);
String property;
if (!rhqMetric.property().isEmpty()) {
property = prefix + rhqMetric.property();
} else {
property = prefix + getPropertyFromBeanConvention(ctField);
}
MetricProps metric = new MetricProps(property);
String displayName =
withNamePrefix
? "[" + mbean.objectName() + "] " + rhqMetric.displayName()
: rhqMetric.displayName();
metric.setDisplayName(displayName);
metric.setDisplayType(rhqMetric.displayType());
metric.setDataType(rhqMetric.dataType());
metric.setUnits(rhqMetric.units());
ManagedAttribute managedAttr =
(ManagedAttribute) ctField.getAnnotation(ManagedAttribute.class);
if (managedAttr != null) {
debug("Metric has ManagedAttribute annotation " + managedAttr);
metric.setDescription(managedAttr.description());
} else {
log.debug(
"Metric has no managed annotations, so take the description from the display name.");
metric.setDescription(rhqMetric.displayName());
}
props.getMetrics().add(metric);
}
}
}
}
private static String getPropertyFromBeanConvention(CtMethod ctMethod) {
String getterOrSetter = ctMethod.getName();
if (getterOrSetter.startsWith("get") || getterOrSetter.startsWith("set")) {
String withoutGet = getterOrSetter.substring(4);
// not specifically Bean convention, but this is what is bound in JMX.
return Character.toUpperCase(getterOrSetter.charAt(3)) + withoutGet;
} else if (getterOrSetter.startsWith("is")) {
String withoutIs = getterOrSetter.substring(3);
return Character.toUpperCase(getterOrSetter.charAt(2)) + withoutIs;
}
return getterOrSetter;
}
private static String getPropertyFromBeanConvention(CtField ctField) {
String fieldName = ctField.getName();
String withoutFirstChar = fieldName.substring(1);
return Character.toUpperCase(fieldName.charAt(0)) + withoutFirstChar;
}
private static void debug(Object o) {
// if (log.isDebugEnabled()) log.debug(o);
// System.out.println(o);
}
}
/** * @author [email protected] * @author Galder Zamarreño * @author Sanne Grinovero <*****@*****.**> (C) 2011 Red Hat Inc. * @since 4.0 */ public class CommandsFactoryImpl implements CommandsFactory { private static final Log log = LogFactory.getLog(CommandsFactoryImpl.class); private static final boolean trace = log.isTraceEnabled(); private DataContainer dataContainer; private CacheNotifier<Object, Object> notifier; private Cache<Object, Object> cache; private String cacheName; private boolean totalOrderProtocol; private InterceptorChain interceptorChain; private DistributionManager distributionManager; private InvocationContextFactory icf; private TransactionTable txTable; private Configuration configuration; private RecoveryManager recoveryManager; private StateProvider stateProvider; private StateConsumer stateConsumer; private LockManager lockManager; private InternalEntryFactory entryFactory; private MapReduceManager mapReduceManager; private StateTransferManager stateTransferManager; private BackupSender backupSender; private CancellationService cancellationService; private XSiteStateProvider xSiteStateProvider; private XSiteStateConsumer xSiteStateConsumer; private XSiteStateTransferManager xSiteStateTransferManager; private EntryRetriever entryRetriever; private GroupManager groupManager; private LocalStreamManager localStreamManager; private ClusterStreamManager clusterStreamManager; private ClusteringDependentLogic clusteringDependentLogic; private TimeService timeService; private Map<Byte, ModuleCommandInitializer> moduleCommandInitializers; private ExternalizerTable externalizerTable; @Inject public void setupDependencies( DataContainer container, CacheNotifier<Object, Object> notifier, Cache<Object, Object> cache, InterceptorChain interceptorChain, DistributionManager distributionManager, InvocationContextFactory icf, TransactionTable txTable, Configuration configuration, @ComponentName(KnownComponentNames.MODULE_COMMAND_INITIALIZERS) Map<Byte, ModuleCommandInitializer> moduleCommandInitializers, RecoveryManager recoveryManager, StateProvider stateProvider, StateConsumer stateConsumer, LockManager lockManager, InternalEntryFactory entryFactory, MapReduceManager mapReduceManager, StateTransferManager stm, BackupSender backupSender, CancellationService cancellationService, TimeService timeService, XSiteStateProvider xSiteStateProvider, XSiteStateConsumer xSiteStateConsumer, XSiteStateTransferManager xSiteStateTransferManager, EntryRetriever entryRetriever, GroupManager groupManager, PartitionHandlingManager partitionHandlingManager, LocalStreamManager localStreamManager, ClusterStreamManager clusterStreamManager, ClusteringDependentLogic clusteringDependentLogic, ExternalizerTable externalizerTable) { this.dataContainer = container; this.notifier = notifier; this.cache = cache; this.interceptorChain = interceptorChain; this.distributionManager = distributionManager; this.icf = icf; this.txTable = txTable; this.configuration = configuration; this.moduleCommandInitializers = moduleCommandInitializers; this.recoveryManager = recoveryManager; this.stateProvider = stateProvider; this.stateConsumer = stateConsumer; this.lockManager = lockManager; this.entryFactory = entryFactory; this.mapReduceManager = mapReduceManager; this.stateTransferManager = stm; this.backupSender = backupSender; this.cancellationService = cancellationService; this.xSiteStateConsumer = xSiteStateConsumer; this.xSiteStateProvider = xSiteStateProvider; this.xSiteStateTransferManager = xSiteStateTransferManager; this.entryRetriever = entryRetriever; this.groupManager = groupManager; this.localStreamManager = localStreamManager; this.clusterStreamManager = clusterStreamManager; this.clusteringDependentLogic = clusteringDependentLogic; this.timeService = timeService; this.externalizerTable = externalizerTable; } @Start(priority = 1) // needs to happen early on public void start() { cacheName = cache.getName(); this.totalOrderProtocol = configuration.transaction().transactionProtocol().isTotalOrder(); } @Override public PutKeyValueCommand buildPutKeyValueCommand( Object key, Object value, Metadata metadata, Set<Flag> flags) { return new PutKeyValueCommand( key, value, false, notifier, metadata, flags, configuration.dataContainer().valueEquivalence(), generateUUID()); } @Override public RemoveCommand buildRemoveCommand(Object key, Object value, Set<Flag> flags) { return new RemoveCommand( key, value, notifier, flags, configuration.dataContainer().valueEquivalence(), generateUUID()); } @Override public InvalidateCommand buildInvalidateCommand(Set<Flag> flags, Object... keys) { return new InvalidateCommand(notifier, flags, generateUUID(), keys); } @Override public InvalidateCommand buildInvalidateFromL1Command(Set<Flag> flags, Collection<Object> keys) { return new InvalidateL1Command( dataContainer, distributionManager, notifier, flags, keys, generateUUID()); } @Override public InvalidateCommand buildInvalidateFromL1Command( Address origin, Set<Flag> flags, Collection<Object> keys) { return new InvalidateL1Command( origin, dataContainer, distributionManager, notifier, flags, keys, generateUUID()); } @Override public RemoveExpiredCommand buildRemoveExpiredCommand(Object key, Object value, Long lifespan) { return new RemoveExpiredCommand( key, value, lifespan, notifier, configuration.dataContainer().valueEquivalence(), timeService, generateUUID()); } @Override public ReplaceCommand buildReplaceCommand( Object key, Object oldValue, Object newValue, Metadata metadata, Set<Flag> flags) { return new ReplaceCommand( key, oldValue, newValue, notifier, metadata, flags, configuration.dataContainer().valueEquivalence(), generateUUID()); } @Override public SizeCommand buildSizeCommand(Set<Flag> flags) { return new SizeCommand(cache, flags); } @Override public KeySetCommand buildKeySetCommand(Set<Flag> flags) { return new KeySetCommand(cache, flags); } @Override public EntrySetCommand buildEntrySetCommand(Set<Flag> flags) { return new EntrySetCommand(cache, flags); } @Override public GetKeyValueCommand buildGetKeyValueCommand(Object key, Set<Flag> flags) { return new GetKeyValueCommand(key, flags); } @Override public GetAllCommand buildGetAllCommand( Collection<?> keys, Set<Flag> flags, boolean returnEntries) { return new GetAllCommand(keys, flags, returnEntries, entryFactory); } @Override public PutMapCommand buildPutMapCommand(Map<?, ?> map, Metadata metadata, Set<Flag> flags) { return new PutMapCommand(map, notifier, metadata, flags, generateUUID()); } @Override public ClearCommand buildClearCommand(Set<Flag> flags) { return new ClearCommand(notifier, dataContainer, flags); } @Override public EvictCommand buildEvictCommand(Object key, Set<Flag> flags) { return new EvictCommand(key, notifier, flags, generateUUID(), entryFactory); } @Override public PrepareCommand buildPrepareCommand( GlobalTransaction gtx, List<WriteCommand> modifications, boolean onePhaseCommit) { return totalOrderProtocol ? new TotalOrderNonVersionedPrepareCommand(cacheName, gtx, modifications) : new PrepareCommand(cacheName, gtx, modifications, onePhaseCommit); } @Override public VersionedPrepareCommand buildVersionedPrepareCommand( GlobalTransaction gtx, List<WriteCommand> modifications, boolean onePhase) { return totalOrderProtocol ? new TotalOrderVersionedPrepareCommand(cacheName, gtx, modifications, onePhase) : new VersionedPrepareCommand(cacheName, gtx, modifications, onePhase); } @Override public CommitCommand buildCommitCommand(GlobalTransaction gtx) { return totalOrderProtocol ? new TotalOrderCommitCommand(cacheName, gtx) : new CommitCommand(cacheName, gtx); } @Override public VersionedCommitCommand buildVersionedCommitCommand(GlobalTransaction gtx) { return totalOrderProtocol ? new TotalOrderVersionedCommitCommand(cacheName, gtx) : new VersionedCommitCommand(cacheName, gtx); } @Override public RollbackCommand buildRollbackCommand(GlobalTransaction gtx) { return totalOrderProtocol ? new TotalOrderRollbackCommand(cacheName, gtx) : new RollbackCommand(cacheName, gtx); } @Override public MultipleRpcCommand buildReplicateCommand(List<ReplicableCommand> toReplicate) { return new MultipleRpcCommand(toReplicate, cacheName); } @Override public SingleRpcCommand buildSingleRpcCommand(ReplicableCommand call) { return new SingleRpcCommand(cacheName, call); } @Override public ClusteredGetCommand buildClusteredGetCommand( Object key, Set<Flag> flags, boolean acquireRemoteLock, GlobalTransaction gtx) { return new ClusteredGetCommand( key, cacheName, flags, acquireRemoteLock, gtx, configuration.dataContainer().keyEquivalence()); } /** @param isRemote true if the command is deserialized and is executed remote. */ @Override public void initializeReplicableCommand(ReplicableCommand c, boolean isRemote) { if (c == null) return; switch (c.getCommandId()) { case PutKeyValueCommand.COMMAND_ID: ((PutKeyValueCommand) c).init(notifier, configuration); break; case ReplaceCommand.COMMAND_ID: ((ReplaceCommand) c).init(notifier, configuration); break; case PutMapCommand.COMMAND_ID: ((PutMapCommand) c).init(notifier); break; case RemoveCommand.COMMAND_ID: ((RemoveCommand) c).init(notifier, configuration); break; case MultipleRpcCommand.COMMAND_ID: MultipleRpcCommand rc = (MultipleRpcCommand) c; rc.init(interceptorChain, icf); if (rc.getCommands() != null) for (ReplicableCommand nested : rc.getCommands()) { initializeReplicableCommand(nested, false); } break; case SingleRpcCommand.COMMAND_ID: SingleRpcCommand src = (SingleRpcCommand) c; src.init(interceptorChain, icf); if (src.getCommand() != null) initializeReplicableCommand(src.getCommand(), false); break; case InvalidateCommand.COMMAND_ID: InvalidateCommand ic = (InvalidateCommand) c; ic.init(notifier, configuration); break; case InvalidateL1Command.COMMAND_ID: InvalidateL1Command ilc = (InvalidateL1Command) c; ilc.init(configuration, distributionManager, notifier, dataContainer); break; case PrepareCommand.COMMAND_ID: case VersionedPrepareCommand.COMMAND_ID: case TotalOrderNonVersionedPrepareCommand.COMMAND_ID: case TotalOrderVersionedPrepareCommand.COMMAND_ID: PrepareCommand pc = (PrepareCommand) c; pc.init(interceptorChain, icf, txTable); pc.initialize(notifier, recoveryManager); if (pc.getModifications() != null) for (ReplicableCommand nested : pc.getModifications()) { initializeReplicableCommand(nested, false); } pc.markTransactionAsRemote(isRemote); if (configuration.deadlockDetection().enabled() && isRemote) { DldGlobalTransaction transaction = (DldGlobalTransaction) pc.getGlobalTransaction(); transaction.setLocksHeldAtOrigin(pc.getAffectedKeys()); } break; case CommitCommand.COMMAND_ID: case VersionedCommitCommand.COMMAND_ID: case TotalOrderCommitCommand.COMMAND_ID: case TotalOrderVersionedCommitCommand.COMMAND_ID: CommitCommand commitCommand = (CommitCommand) c; commitCommand.init(interceptorChain, icf, txTable); commitCommand.markTransactionAsRemote(isRemote); break; case RollbackCommand.COMMAND_ID: case TotalOrderRollbackCommand.COMMAND_ID: RollbackCommand rollbackCommand = (RollbackCommand) c; rollbackCommand.init(interceptorChain, icf, txTable); rollbackCommand.markTransactionAsRemote(isRemote); break; case ClearCommand.COMMAND_ID: ClearCommand cc = (ClearCommand) c; cc.init(notifier, dataContainer); break; case ClusteredGetCommand.COMMAND_ID: ClusteredGetCommand clusteredGetCommand = (ClusteredGetCommand) c; clusteredGetCommand.initialize( icf, this, entryFactory, interceptorChain, distributionManager, txTable, configuration.dataContainer().keyEquivalence()); break; case LockControlCommand.COMMAND_ID: LockControlCommand lcc = (LockControlCommand) c; lcc.init(interceptorChain, icf, txTable); lcc.markTransactionAsRemote(isRemote); if (configuration.deadlockDetection().enabled() && isRemote) { DldGlobalTransaction gtx = (DldGlobalTransaction) lcc.getGlobalTransaction(); RemoteTransaction transaction = txTable.getRemoteTransaction(gtx); if (transaction != null) { if (!configuration.clustering().cacheMode().isDistributed()) { Set<Object> keys = txTable.getLockedKeysForRemoteTransaction(gtx); GlobalTransaction gtx2 = transaction.getGlobalTransaction(); ((DldGlobalTransaction) gtx2).setLocksHeldAtOrigin(keys); gtx.setLocksHeldAtOrigin(keys); } else { GlobalTransaction gtx2 = transaction.getGlobalTransaction(); ((DldGlobalTransaction) gtx2).setLocksHeldAtOrigin(gtx.getLocksHeldAtOrigin()); } } } break; case StateRequestCommand.COMMAND_ID: ((StateRequestCommand) c).init(stateProvider); break; case StateResponseCommand.COMMAND_ID: ((StateResponseCommand) c).init(stateConsumer); break; case GetInDoubtTransactionsCommand.COMMAND_ID: GetInDoubtTransactionsCommand gptx = (GetInDoubtTransactionsCommand) c; gptx.init(recoveryManager); break; case TxCompletionNotificationCommand.COMMAND_ID: TxCompletionNotificationCommand ftx = (TxCompletionNotificationCommand) c; ftx.init(txTable, lockManager, recoveryManager, stateTransferManager); break; case MapCombineCommand.COMMAND_ID: MapCombineCommand mrc = (MapCombineCommand) c; mrc.init(mapReduceManager); break; case ReduceCommand.COMMAND_ID: ReduceCommand reduceCommand = (ReduceCommand) c; reduceCommand.init(mapReduceManager); break; case DistributedExecuteCommand.COMMAND_ID: DistributedExecuteCommand dec = (DistributedExecuteCommand) c; dec.init(cache); break; case GetInDoubtTxInfoCommand.COMMAND_ID: GetInDoubtTxInfoCommand gidTxInfoCommand = (GetInDoubtTxInfoCommand) c; gidTxInfoCommand.init(recoveryManager); break; case CompleteTransactionCommand.COMMAND_ID: CompleteTransactionCommand ccc = (CompleteTransactionCommand) c; ccc.init(recoveryManager); break; case ApplyDeltaCommand.COMMAND_ID: break; case CreateCacheCommand.COMMAND_ID: CreateCacheCommand createCacheCommand = (CreateCacheCommand) c; createCacheCommand.init(cache.getCacheManager()); break; case XSiteAdminCommand.COMMAND_ID: XSiteAdminCommand xSiteAdminCommand = (XSiteAdminCommand) c; xSiteAdminCommand.init(backupSender); break; case CancelCommand.COMMAND_ID: CancelCommand cancelCommand = (CancelCommand) c; cancelCommand.init(cancellationService); break; case XSiteStateTransferControlCommand.COMMAND_ID: XSiteStateTransferControlCommand xSiteStateTransferControlCommand = (XSiteStateTransferControlCommand) c; xSiteStateTransferControlCommand.initialize( xSiteStateProvider, xSiteStateConsumer, xSiteStateTransferManager); break; case XSiteStatePushCommand.COMMAND_ID: XSiteStatePushCommand xSiteStatePushCommand = (XSiteStatePushCommand) c; xSiteStatePushCommand.initialize(xSiteStateConsumer); break; case EntryRequestCommand.COMMAND_ID: EntryRequestCommand entryRequestCommand = (EntryRequestCommand) c; entryRequestCommand.init(entryRetriever); break; case EntryResponseCommand.COMMAND_ID: EntryResponseCommand entryResponseCommand = (EntryResponseCommand) c; entryResponseCommand.init(entryRetriever); break; case GetKeysInGroupCommand.COMMAND_ID: GetKeysInGroupCommand getKeysInGroupCommand = (GetKeysInGroupCommand) c; getKeysInGroupCommand.setGroupManager(groupManager); break; case ClusteredGetAllCommand.COMMAND_ID: ClusteredGetAllCommand clusteredGetAllCommand = (ClusteredGetAllCommand) c; clusteredGetAllCommand.init( icf, this, entryFactory, interceptorChain, txTable, configuration.dataContainer().keyEquivalence()); break; case StreamRequestCommand.COMMAND_ID: StreamRequestCommand streamRequestCommand = (StreamRequestCommand) c; streamRequestCommand.inject(localStreamManager); break; case StreamResponseCommand.COMMAND_ID: StreamResponseCommand streamResponseCommand = (StreamResponseCommand) c; streamResponseCommand.inject(clusterStreamManager); break; case StreamSegmentResponseCommand.COMMAND_ID: StreamSegmentResponseCommand streamSegmentResponseCommand = (StreamSegmentResponseCommand) c; streamSegmentResponseCommand.inject(clusterStreamManager); break; case RemoveExpiredCommand.COMMAND_ID: RemoveExpiredCommand removeExpiredCommand = (RemoveExpiredCommand) c; removeExpiredCommand.init(notifier, configuration); break; default: ModuleCommandInitializer mci = moduleCommandInitializers.get(c.getCommandId()); if (mci != null) { mci.initializeReplicableCommand(c, isRemote); } else { if (trace) log.tracef("Nothing to initialize for command: %s", c); } } } @Override public LockControlCommand buildLockControlCommand( Collection<?> keys, Set<Flag> flags, GlobalTransaction gtx) { return new LockControlCommand(keys, cacheName, flags, gtx); } @Override public LockControlCommand buildLockControlCommand( Object key, Set<Flag> flags, GlobalTransaction gtx) { return new LockControlCommand(key, cacheName, flags, gtx); } @Override public LockControlCommand buildLockControlCommand(Collection<?> keys, Set<Flag> flags) { return new LockControlCommand(keys, cacheName, flags, null); } @Override public StateRequestCommand buildStateRequestCommand( StateRequestCommand.Type subtype, Address sender, int viewId, Set<Integer> segments) { return new StateRequestCommand(cacheName, subtype, sender, viewId, segments); } @Override public StateResponseCommand buildStateResponseCommand( Address sender, int topologyId, Collection<StateChunk> stateChunks) { return new StateResponseCommand(cacheName, sender, topologyId, stateChunks); } @Override public String getCacheName() { return cacheName; } @Override public GetInDoubtTransactionsCommand buildGetInDoubtTransactionsCommand() { return new GetInDoubtTransactionsCommand(cacheName); } @Override public TxCompletionNotificationCommand buildTxCompletionNotificationCommand( Xid xid, GlobalTransaction globalTransaction) { return new TxCompletionNotificationCommand(xid, globalTransaction, cacheName); } @Override public TxCompletionNotificationCommand buildTxCompletionNotificationCommand(long internalId) { return new TxCompletionNotificationCommand(internalId, cacheName); } @Override public <T> DistributedExecuteCommand<T> buildDistributedExecuteCommand( Callable<T> callable, Address sender, Collection keys) { return new DistributedExecuteCommand<T>(cacheName, keys, callable); } @Override public <KIn, VIn, KOut, VOut> MapCombineCommand<KIn, VIn, KOut, VOut> buildMapCombineCommand( String taskId, Mapper<KIn, VIn, KOut, VOut> m, Reducer<KOut, VOut> r, Collection<KIn> keys) { return new MapCombineCommand<KIn, VIn, KOut, VOut>(taskId, m, r, cacheName, keys); } @Override public GetInDoubtTxInfoCommand buildGetInDoubtTxInfoCommand() { return new GetInDoubtTxInfoCommand(cacheName); } @Override public CompleteTransactionCommand buildCompleteTransactionCommand(Xid xid, boolean commit) { return new CompleteTransactionCommand(cacheName, xid, commit); } @Override public ApplyDeltaCommand buildApplyDeltaCommand( Object deltaAwareValueKey, Delta delta, Collection keys) { return new ApplyDeltaCommand(deltaAwareValueKey, delta, keys, generateUUID()); } @Override public CreateCacheCommand buildCreateCacheCommand( String cacheNameToCreate, String cacheConfigurationName) { return new CreateCacheCommand(cacheName, cacheNameToCreate, cacheConfigurationName); } @Override public CreateCacheCommand buildCreateCacheCommand( String cacheNameToCreate, String cacheConfigurationName, int size) { return new CreateCacheCommand(cacheName, cacheNameToCreate, cacheConfigurationName, size); } @Override public <KOut, VOut> ReduceCommand<KOut, VOut> buildReduceCommand( String taskId, String destintationCache, Reducer<KOut, VOut> r, Collection<KOut> keys) { return new ReduceCommand<KOut, VOut>(taskId, r, destintationCache, keys); } @Override public CancelCommand buildCancelCommandCommand(UUID commandUUID) { return new CancelCommand(cacheName, commandUUID); } @Override public XSiteStateTransferControlCommand buildXSiteStateTransferControlCommand( StateTransferControl control, String siteName) { return new XSiteStateTransferControlCommand(cacheName, control, siteName); } @Override public XSiteAdminCommand buildXSiteAdminCommand( String siteName, AdminOperation op, Integer afterFailures, Long minTimeToWait) { return new XSiteAdminCommand(cacheName, siteName, op, afterFailures, minTimeToWait); } @Override public XSiteStatePushCommand buildXSiteStatePushCommand(XSiteState[] chunk, long timeoutMillis) { return new XSiteStatePushCommand(cacheName, chunk, timeoutMillis); } @Override public SingleXSiteRpcCommand buildSingleXSiteRpcCommand(VisitableCommand command) { return new SingleXSiteRpcCommand(cacheName, command); } @Override public <K, V, C> EntryRequestCommand<K, V, C> buildEntryRequestCommand( UUID identifier, Set<Integer> segments, Set<K> keysToFilter, KeyValueFilter<? super K, ? super V> filter, Converter<? super K, ? super V, C> converter, Set<Flag> flags) { return new EntryRequestCommand<K, V, C>( cacheName, identifier, cache.getCacheManager().getAddress(), segments, keysToFilter, filter, converter, flags); } @Override public <K, C> EntryResponseCommand<K, C> buildEntryResponseCommand( UUID identifier, Set<Integer> completedSegments, Set<Integer> inDoubtSegments, Collection<CacheEntry<K, C>> values, CacheException e) { return new EntryResponseCommand<>( cache.getCacheManager().getAddress(), cacheName, identifier, completedSegments, inDoubtSegments, values, e); } @Override public GetKeysInGroupCommand buildGetKeysInGroupCommand(Set<Flag> flags, String groupName) { return new GetKeysInGroupCommand(flags, groupName).setGroupManager(groupManager); } @Override public <K> StreamRequestCommand<K> buildStreamRequestCommand( Object id, boolean parallelStream, StreamRequestCommand.Type type, Set<Integer> segments, Set<K> keys, Set<K> excludedKeys, boolean includeLoader, Object terminalOperation) { return new StreamRequestCommand<>( cacheName, cache.getCacheManager().getAddress(), id, parallelStream, type, segments, keys, excludedKeys, includeLoader, terminalOperation); } @Override public <R> StreamResponseCommand<R> buildStreamResponseCommand( Object identifier, boolean complete, Set<Integer> lostSegments, R response) { if (lostSegments.isEmpty()) { return new StreamResponseCommand<>( cacheName, cache.getCacheManager().getAddress(), identifier, complete, response); } else { return new StreamSegmentResponseCommand<>( cacheName, cache.getCacheManager().getAddress(), identifier, complete, response, lostSegments); } } @Override public GetCacheEntryCommand buildGetCacheEntryCommand(Object key, Set<Flag> explicitFlags) { return new GetCacheEntryCommand(key, explicitFlags, entryFactory); } @Override public ClusteredGetAllCommand buildClusteredGetAllCommand( List<?> keys, Set<Flag> flags, GlobalTransaction gtx) { return new ClusteredGetAllCommand( cacheName, keys, flags, gtx, configuration.dataContainer().keyEquivalence()); } private CommandInvocationId generateUUID() { return CommandInvocationId.generateId(clusteringDependentLogic.getAddress()); } @Override public <K, V, R> ReadOnlyKeyCommand<K, V, R> buildReadOnlyKeyCommand( K key, Function<ReadEntryView<K, V>, R> f) { return new ReadOnlyKeyCommand<>(key, f); } @Override public <K, V, R> ReadOnlyManyCommand<K, V, R> buildReadOnlyManyCommand( Set<? extends K> keys, Function<ReadEntryView<K, V>, R> f) { return new ReadOnlyManyCommand<>(keys, f); } @Override public <K, V, R> ReadWriteKeyValueCommand<K, V, R> buildReadWriteKeyValueCommand( K key, V value, BiFunction<V, ReadWriteEntryView<K, V>, R> f, Params params) { return new ReadWriteKeyValueCommand<>( key, value, f, generateUUID(), getValueMatcher(f), params); } @Override public <K, V, R> ReadWriteKeyCommand<K, V, R> buildReadWriteKeyCommand( K key, Function<ReadWriteEntryView<K, V>, R> f, Params params) { return new ReadWriteKeyCommand<>(key, f, generateUUID(), getValueMatcher(f), params); } @Override public <K, V, R> ReadWriteManyCommand<K, V, R> buildReadWriteManyCommand( Set<? extends K> keys, Function<ReadWriteEntryView<K, V>, R> f) { return new ReadWriteManyCommand<>(keys, f); } @Override public <K, V, R> ReadWriteManyEntriesCommand<K, V, R> buildReadWriteManyEntriesCommand( Map<? extends K, ? extends V> entries, BiFunction<V, ReadWriteEntryView<K, V>, R> f) { return new ReadWriteManyEntriesCommand<>(entries, f); } @Override public <K, V> WriteOnlyKeyCommand<K, V> buildWriteOnlyKeyCommand( K key, Consumer<WriteEntryView<V>> f, Params params) { return new WriteOnlyKeyCommand<>(key, f, generateUUID(), getValueMatcher(f), params); } @Override public <K, V> WriteOnlyKeyValueCommand<K, V> buildWriteOnlyKeyValueCommand( K key, V value, BiConsumer<V, WriteEntryView<V>> f, Params params) { return new WriteOnlyKeyValueCommand<>( key, value, f, generateUUID(), getValueMatcher(f), params); } @Override public <K, V> WriteOnlyManyCommand<K, V> buildWriteOnlyManyCommand( Set<? extends K> keys, Consumer<WriteEntryView<V>> f) { return new WriteOnlyManyCommand<>(keys, f); } @Override public <K, V> WriteOnlyManyEntriesCommand<K, V> buildWriteOnlyManyEntriesCommand( Map<? extends K, ? extends V> entries, BiConsumer<V, WriteEntryView<V>> f, Params params) { WriteOnlyManyEntriesCommand<K, V> cmd = new WriteOnlyManyEntriesCommand<>(entries, f); cmd.setParams(params); return cmd; } private ValueMatcher getValueMatcher(Object o) { SerializeFunctionWith ann = o.getClass().getAnnotation(SerializeFunctionWith.class); if (ann != null) return ValueMatcher.valueOf(ann.valueMatcher().toString()); Externalizer ext = externalizerTable.getExternalizer(o); if (ext != null && ext instanceof LambdaExternalizer) return ValueMatcher.valueOf(((LambdaExternalizer) ext).valueMatcher(o).toString()); return ValueMatcher.MATCH_ALWAYS; } }
/** * Contains all the logic of manipulating the table, including creating it if needed and access * operations like inserting, selecting etc. Used by JDBC based cache persistence. * * @author [email protected] */ public class TableManipulation implements Cloneable { private static final Log log = LogFactory.getLog(TableManipulation.class, Log.class); private static final boolean trace = log.isTraceEnabled(); public static final int DEFAULT_FETCH_SIZE = 100; public static final int DEFAULT_BATCH_SIZE = 128; private String identifierQuoteString; private String cacheName; TableManipulationConfiguration config; /* * following two params manage creation and destruction during start up/shutdown. */ private ConnectionFactory connectionFactory; /* Cache the sql for managing data */ private String insertRowSql; private String updateRowSql; private String selectRowSql; private String selectIdRowSql; private String deleteRowSql; private String loadAllRowsSql; private String countRowsSql; private String loadAllNonExpiredRowsSql; private String deleteAllRows; private String selectExpiredRowsSql; private String deleteExpiredRowsSql; private String loadSomeRowsSql; private DatabaseType databaseType; private String loadAllKeysBinarySql; private String loadAllKeysStringSql; private TableName tableName; public TableManipulation(TableManipulationConfiguration config, DatabaseType databaseType) { this.config = config; this.databaseType = databaseType; } public TableManipulation() {} public boolean tableExists(Connection connection, TableName tableName) throws PersistenceException { if (tableName == null) { throw new NullPointerException("table name is mandatory"); } ResultSet rs = null; try { // we need to make sure, that (even if the user has extended permissions) only the tables in // current schema are checked // explicit set of the schema to the current user one to make sure only tables of the current // users are requested DatabaseMetaData metaData = connection.getMetaData(); String schemaPattern = tableName.getSchema(); if (schemaPattern == null) { switch (getDialect()) { case ORACLE: schemaPattern = metaData.getUserName(); break; default: } } rs = metaData.getTables(null, schemaPattern, tableName.getName(), new String[] {"TABLE"}); return rs.next(); } catch (SQLException e) { if (trace) log.tracef(e, "SQLException occurs while checking the table %s", tableName); return false; } finally { JdbcUtil.safeClose(rs); } } public void createTable(Connection conn) throws PersistenceException { // removed CONSTRAINT clause as this causes problems with some databases, like Informix. assertMandatoryElementsPresent(); String createTableDdl = "CREATE TABLE " + getTableName() + "(" + config.idColumnName() + " " + config.idColumnType() + " NOT NULL, " + config.dataColumnName() + " " + config.dataColumnType() + ", " + config.timestampColumnName() + " " + config.timestampColumnType() + ", PRIMARY KEY (" + config.idColumnName() + "))"; if (trace) { log.tracef("Creating table with following DDL: '%s'.", createTableDdl); } executeUpdateSql(conn, createTableDdl); } private void assertMandatoryElementsPresent() throws PersistenceException { assertNotNull(cacheName, "cacheName needed in order to create table"); } private void assertNotNull(String keyColumnType, String message) throws PersistenceException { if (keyColumnType == null || keyColumnType.trim().length() == 0) { throw new PersistenceException(message); } } private void executeUpdateSql(Connection conn, String sql) throws PersistenceException { Statement statement = null; try { statement = conn.createStatement(); statement.executeUpdate(sql); } catch (SQLException e) { log.errorCreatingTable(sql, e); throw new PersistenceException(e); } finally { JdbcUtil.safeClose(statement); } } public void dropTable(Connection conn) throws PersistenceException { String dropTableDdl = "DROP TABLE " + getTableName(); String clearTable = "DELETE FROM " + getTableName(); executeUpdateSql(conn, clearTable); if (trace) { log.tracef("Dropping table with following DDL '%s'", dropTableDdl); } executeUpdateSql(conn, dropTableDdl); } public void start(ConnectionFactory connectionFactory) throws PersistenceException { this.connectionFactory = connectionFactory; if (config.createOnStart()) { Connection conn = null; try { conn = this.connectionFactory.getConnection(); if (!tableExists(conn, getTableName())) { createTable(conn); } } finally { this.connectionFactory.releaseConnection(conn); } } } public void stop() throws PersistenceException { if (config.dropOnExit()) { Connection conn = null; try { conn = connectionFactory.getConnection(); dropTable(conn); } finally { connectionFactory.releaseConnection(conn); } } } public String getInsertRowSql() { if (insertRowSql == null) { insertRowSql = "INSERT INTO " + getTableName() + " (" + config.dataColumnName() + ", " + config.timestampColumnName() + ", " + config.idColumnName() + ") VALUES(?,?,?)"; } return insertRowSql; } public String getUpdateRowSql() { if (updateRowSql == null) { switch (getDialect()) { case SYBASE: updateRowSql = "UPDATE " + getTableName() + " SET " + config.dataColumnName() + " = ? , " + config.timestampColumnName() + "=? WHERE " + config.idColumnName() + " = convert(" + config.idColumnType() + "," + "?)"; break; case POSTGRES: updateRowSql = "UPDATE " + getTableName() + " SET " + config.dataColumnName() + " = ? , " + config.timestampColumnName() + "=? WHERE " + config.idColumnName() + " = cast(? as " + config.idColumnType() + ")"; break; default: updateRowSql = "UPDATE " + getTableName() + " SET " + config.dataColumnName() + " = ? , " + config.timestampColumnName() + "=? WHERE " + config.idColumnName() + " = ?"; break; } } return updateRowSql; } public String getSelectRowSql() { if (selectRowSql == null) { switch (getDialect()) { case SYBASE: selectRowSql = "SELECT " + config.idColumnName() + ", " + config.dataColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = convert(" + config.idColumnType() + "," + "?)"; break; case POSTGRES: selectRowSql = "SELECT " + config.idColumnName() + ", " + config.dataColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = cast(? as " + config.idColumnType() + ")"; break; default: selectRowSql = "SELECT " + config.idColumnName() + ", " + config.dataColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = ?"; break; } } return selectRowSql; } public String getSelectIdRowSql() { if (selectIdRowSql == null) { switch (getDialect()) { case SYBASE: selectIdRowSql = "SELECT " + config.idColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = convert(" + config.idColumnType() + "," + "?)"; break; case POSTGRES: selectIdRowSql = "SELECT " + config.idColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = cast(? as " + config.idColumnType() + ")"; break; default: selectIdRowSql = "SELECT " + config.idColumnName() + " FROM " + getTableName() + " WHERE " + config.idColumnName() + " = ?"; break; } } return selectIdRowSql; } public String getCountRowsSql() { if (countRowsSql == null) { countRowsSql = "SELECT COUNT(*) FROM " + getTableName(); } return countRowsSql; } public String getDeleteRowSql() { if (deleteRowSql == null) { switch (getDialect()) { case SYBASE: deleteRowSql = "DELETE FROM " + getTableName() + " WHERE " + config.idColumnName() + " = convert(" + config.idColumnType() + "," + "?)"; break; case POSTGRES: deleteRowSql = "DELETE FROM " + getTableName() + " WHERE " + config.idColumnName() + " = cast(? as " + config.idColumnType() + ")"; break; default: deleteRowSql = "DELETE FROM " + getTableName() + " WHERE " + config.idColumnName() + " = ?"; break; } } return deleteRowSql; } public String getLoadNonExpiredAllRowsSql() { if (loadAllNonExpiredRowsSql == null) { loadAllNonExpiredRowsSql = "SELECT " + config.dataColumnName() + "," + config.idColumnName() + ", " + config.timestampColumnName() + " FROM " + getTableName() + " WHERE " + config.timestampColumnName() + " > ? OR " + config.timestampColumnName() + " < 0"; } return loadAllNonExpiredRowsSql; } public String getLoadAllRowsSql() { if (loadAllRowsSql == null) { loadAllRowsSql = "SELECT " + config.dataColumnName() + "," + config.idColumnName() + " FROM " + getTableName(); } return loadAllRowsSql; } public String getDeleteAllRowsSql() { if (deleteAllRows == null) { deleteAllRows = "DELETE FROM " + getTableName(); } return deleteAllRows; } public String getSelectExpiredRowsSql() { if (selectExpiredRowsSql == null) { selectExpiredRowsSql = getLoadAllRowsSql() + " WHERE " + config.timestampColumnName() + "< ?"; } return selectExpiredRowsSql; } public String getDeleteExpiredRowsSql() { if (deleteExpiredRowsSql == null) { deleteExpiredRowsSql = "DELETE FROM " + getTableName() + " WHERE " + config.timestampColumnName() + "< ? AND " + config.timestampColumnName() + "> 0"; } return deleteExpiredRowsSql; } @Override public TableManipulation clone() { try { return (TableManipulation) super.clone(); } catch (CloneNotSupportedException e) { throw new IllegalStateException(e); } } public TableName getTableName() { if (tableName == null) { tableName = new TableName(getIdentifierQuoteString(), config.tableNamePrefix(), cacheName); } return tableName; } public boolean tableExists(Connection connection) throws PersistenceException { return tableExists(connection, getTableName()); } public void setCacheName(String cacheName) { this.cacheName = cacheName; tableName = null; } public boolean isVariableLimitSupported() { DatabaseType type = getDialect(); return !(type == DatabaseType.DB2 || type == DatabaseType.DB2_390 || type == DatabaseType.SYBASE); } public String getLoadSomeRowsSql() { if (loadSomeRowsSql == null) { // this stuff is going to be database specific!! // see // http://stackoverflow.com/questions/595123/is-there-an-ansi-sql-alternative-to-the-mysql-limit-keyword switch (getDialect()) { case ORACLE: loadSomeRowsSql = String.format( "SELECT %s, %s FROM (SELECT %s, %s FROM %s) WHERE ROWNUM <= ?", config.dataColumnName(), config.idColumnName(), config.dataColumnName(), config.idColumnName(), getTableName()); break; case DB2: case DB2_390: case DERBY: loadSomeRowsSql = String.format( "SELECT %s, %s FROM %s FETCH FIRST ? ROWS ONLY", config.dataColumnName(), config.idColumnName(), getTableName()); break; case INFORMIX: case INTERBASE: case FIREBIRD: loadSomeRowsSql = String.format( "SELECT FIRST ? %s, %s FROM %s", config.dataColumnName(), config.idColumnName(), getTableName()); break; case SQL_SERVER: loadSomeRowsSql = String.format( "SELECT TOP (?) %s, %s FROM %s", config.dataColumnName(), config.idColumnName(), getTableName()); break; case ACCESS: case HSQL: case SYBASE: loadSomeRowsSql = String.format( "SELECT TOP ? %s, %s FROM %s", config.dataColumnName(), config.idColumnName(), getTableName()); break; default: // the MySQL-style LIMIT clause (works for PostgreSQL too) loadSomeRowsSql = String.format( "SELECT %s, %s FROM %s LIMIT ?", config.dataColumnName(), config.idColumnName(), getTableName()); break; } } return loadSomeRowsSql; } public String getLoadAllKeysBinarySql() { if (loadAllKeysBinarySql == null) { loadAllKeysBinarySql = String.format("SELECT %s FROM %s", config.dataColumnName(), getTableName()); } return loadAllKeysBinarySql; } public String getLoadAllKeysStringSql() { if (loadAllKeysStringSql == null) { loadAllKeysStringSql = String.format("SELECT %s FROM %s", config.idColumnName(), getTableName()); } return loadAllKeysStringSql; } /** * For DB queries the fetch size will be set on {@link java.sql.ResultSet#setFetchSize(int)}. This * is optional parameter, if not specified will be defaulted to {@link #DEFAULT_FETCH_SIZE}. */ public int getFetchSize() { return getDialect() == DatabaseType.MYSQL ? Integer.MIN_VALUE : config.fetchSize(); } /** * When doing repetitive DB inserts this will be batched according to this parameter. This is an * optional parameter, and if it is not specified it will be defaulted to {@link * #DEFAULT_BATCH_SIZE}. Guaranteed to be a power of two. */ public int getBatchSize() { return config.batchSize(); } private DatabaseType getDialect() { if (databaseType == null) { // need to guess from the database type! Connection connection = null; try { connection = connectionFactory.getConnection(); String dbProduct = connection.getMetaData().getDatabaseProductName(); databaseType = guessDialect(dbProduct); } catch (Exception e) { log.debug("Unable to guess dialect from JDBC metadata.", e); } finally { connectionFactory.releaseConnection(connection); } if (databaseType == null) { log.debug( "Unable to detect database dialect using connection metadata. Attempting to guess on driver name."); try { connection = connectionFactory.getConnection(); String dbProduct = connectionFactory.getConnection().getMetaData().getDriverName(); databaseType = guessDialect(dbProduct); } catch (Exception e) { log.debug("Unable to guess database dialect from JDBC driver name.", e); } finally { connectionFactory.releaseConnection(connection); } } if (databaseType == null) { throw new CacheConfigurationException( "Unable to detect database dialect from JDBC driver name or connection metadata. Please provide this manually using the 'dialect' property in your configuration. Supported database dialect strings are " + Arrays.toString(DatabaseType.values())); } else { log.debugf( "Guessing database dialect as '%s'. If this is incorrect, please specify the correct dialect using the 'dialect' attribute in your configuration. Supported database dialect strings are %s", databaseType, Arrays.toString(DatabaseType.values())); } } return databaseType; } private DatabaseType guessDialect(String name) { DatabaseType type = null; if (name != null) { if (name.toLowerCase().contains("mysql")) { type = DatabaseType.MYSQL; } else if (name.toLowerCase().contains("postgres")) { type = DatabaseType.POSTGRES; } else if (name.toLowerCase().contains("derby")) { type = DatabaseType.DERBY; } else if (name.toLowerCase().contains("hsql") || name.toLowerCase().contains("hypersonic")) { type = DatabaseType.HSQL; } else if (name.toLowerCase().contains("h2")) { type = DatabaseType.H2; } else if (name.toLowerCase().contains("sqlite")) { type = DatabaseType.SQLITE; } else if (name.toLowerCase().contains("db2")) { type = DatabaseType.DB2; } else if (name.toLowerCase().contains("informix")) { type = DatabaseType.INFORMIX; } else if (name.toLowerCase().contains("interbase")) { type = DatabaseType.INTERBASE; } else if (name.toLowerCase().contains("firebird")) { type = DatabaseType.FIREBIRD; } else if (name.toLowerCase().contains("sqlserver") || name.toLowerCase().contains("microsoft")) { type = DatabaseType.SQL_SERVER; } else if (name.toLowerCase().contains("access")) { type = DatabaseType.ACCESS; } else if (name.toLowerCase().contains("oracle")) { type = DatabaseType.ORACLE; } else if (name.toLowerCase().contains("adaptive")) { type = DatabaseType.SYBASE; } } return type; } public String getIdentifierQuoteString() { if (identifierQuoteString == null) { switch (getDialect()) { case MYSQL: identifierQuoteString = "`"; break; default: identifierQuoteString = "\""; break; } } return identifierQuoteString; } }
/** * Helper class that handles all notifications to registered listeners. * * @author Manik Surtani (manik AT infinispan DOT org) * @author [email protected] * @since 4.0 */ public final class CacheNotifierImpl extends AbstractListenerImpl implements CacheNotifier, ClassLoaderAwareListenable { private static final Log log = LogFactory.getLog(CacheNotifierImpl.class); private static final Map<Class<? extends Annotation>, Class<?>> allowedListeners = new HashMap<Class<? extends Annotation>, Class<?>>(16); static { allowedListeners.put(CacheEntryCreated.class, CacheEntryCreatedEvent.class); allowedListeners.put(CacheEntryRemoved.class, CacheEntryRemovedEvent.class); allowedListeners.put(CacheEntryVisited.class, CacheEntryVisitedEvent.class); allowedListeners.put(CacheEntryModified.class, CacheEntryModifiedEvent.class); allowedListeners.put(CacheEntryActivated.class, CacheEntryActivatedEvent.class); allowedListeners.put(CacheEntryPassivated.class, CacheEntryPassivatedEvent.class); allowedListeners.put(CacheEntryLoaded.class, CacheEntryLoadedEvent.class); allowedListeners.put(CacheEntriesEvicted.class, CacheEntriesEvictedEvent.class); allowedListeners.put(TransactionRegistered.class, TransactionRegisteredEvent.class); allowedListeners.put(TransactionCompleted.class, TransactionCompletedEvent.class); allowedListeners.put(CacheEntryInvalidated.class, CacheEntryInvalidatedEvent.class); allowedListeners.put(DataRehashed.class, DataRehashedEvent.class); allowedListeners.put(TopologyChanged.class, TopologyChangedEvent.class); // For backward compat allowedListeners.put(CacheEntryEvicted.class, CacheEntryEvictedEvent.class); } final List<ListenerInvocation> cacheEntryCreatedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryRemovedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryVisitedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryModifiedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryActivatedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryPassivatedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryLoadedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntryInvalidatedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> cacheEntriesEvictedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> transactionRegisteredListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> transactionCompletedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> dataRehashedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); final List<ListenerInvocation> topologyChangedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); // For backward compat final List<ListenerInvocation> cacheEntryEvictedListeners = new CopyOnWriteArrayList<ListenerInvocation>(); private Cache<Object, Object> cache; public CacheNotifierImpl() { listenersMap.put(CacheEntryCreated.class, cacheEntryCreatedListeners); listenersMap.put(CacheEntryRemoved.class, cacheEntryRemovedListeners); listenersMap.put(CacheEntryVisited.class, cacheEntryVisitedListeners); listenersMap.put(CacheEntryModified.class, cacheEntryModifiedListeners); listenersMap.put(CacheEntryActivated.class, cacheEntryActivatedListeners); listenersMap.put(CacheEntryPassivated.class, cacheEntryPassivatedListeners); listenersMap.put(CacheEntryLoaded.class, cacheEntryLoadedListeners); listenersMap.put(CacheEntriesEvicted.class, cacheEntriesEvictedListeners); listenersMap.put(TransactionRegistered.class, transactionRegisteredListeners); listenersMap.put(TransactionCompleted.class, transactionCompletedListeners); listenersMap.put(CacheEntryInvalidated.class, cacheEntryInvalidatedListeners); listenersMap.put(DataRehashed.class, dataRehashedListeners); listenersMap.put(TopologyChanged.class, topologyChangedListeners); // For backward compat listenersMap.put(CacheEntryEvicted.class, cacheEntryEvictedListeners); } @Inject void injectDependencies(Cache<Object, Object> cache) { this.cache = cache; } @Override protected Log getLog() { return log; } @Override protected Map<Class<? extends Annotation>, Class<?>> getAllowedMethodAnnotations() { return allowedListeners; } @Override public void notifyCacheEntryCreated( Object key, Object value, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (!cacheEntryCreatedListeners.isEmpty()) { boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_CREATED); e.setOriginLocal(originLocal); // Added capability to set cache entry created value in order // to avoid breaking behaviour of CacheEntryModifiedEvent.getValue() // when isPre=false. e.setValue(value); e.setPre(pre); e.setKey(key); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryCreatedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryModified( Object key, Object value, boolean created, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (!cacheEntryModifiedListeners.isEmpty()) { boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_MODIFIED); e.setOriginLocal(originLocal); e.setValue(value); e.setPre(pre); e.setKey(key); // Even if CacheEntryCreatedEvent.getValue() has been added, to // avoid breaking old behaviour and make it easy to comply with // JSR-107 specification TCK, it's necessary to find out whether a // modification is the result of a cache entry being created or not. // This is needed because on JSR-107, a modification is only fired // when the entry is updated, and only one event is fired, so you // want to fire it when isPre=false. e.setCreated(created); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryModifiedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryRemoved( Object key, Object value, Object oldValue, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryRemovedListeners)) { boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_REMOVED); e.setOriginLocal(originLocal); e.setValue(value); e.setOldValue(oldValue); e.setPre(pre); e.setKey(key); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryRemovedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryVisited( Object key, Object value, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryVisitedListeners)) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_VISITED); e.setPre(pre); e.setKey(key); e.setValue(value); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryVisitedListeners) listener.invoke(e); } } @Override public void notifyCacheEntriesEvicted( Collection<InternalCacheEntry> entries, InvocationContext ctx, FlagAffectedCommand command) { if (!entries.isEmpty()) { if (isNotificationAllowed(command, cacheEntriesEvictedListeners)) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_EVICTED); Map<Object, Object> evictedKeysAndValues = transformCollectionToMap( entries, new InfinispanCollections.MapMakerFunction<Object, Object, InternalCacheEntry>() { @Override public Map.Entry<Object, Object> transform(final InternalCacheEntry input) { return new Map.Entry<Object, Object>() { @Override public Object getKey() { return input.getKey(); } @Override public Object getValue() { return input.getValue(); } @Override public Object setValue(Object value) { throw new UnsupportedOperationException(); } }; } }); e.setEntries(evictedKeysAndValues); for (ListenerInvocation listener : cacheEntriesEvictedListeners) listener.invoke(e); } // For backward compat if (isNotificationAllowed(command, cacheEntryEvictedListeners)) { for (InternalCacheEntry ice : entries) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_EVICTED); e.setKey(ice.getKey()); e.setValue(ice.getValue()); for (ListenerInvocation listener : cacheEntryEvictedListeners) listener.invoke(e); } } } } @Override public void notifyCacheEntryEvicted( Object key, Object value, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntriesEvictedListeners)) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_EVICTED); e.setEntries(Collections.singletonMap(key, value)); for (ListenerInvocation listener : cacheEntriesEvictedListeners) listener.invoke(e); } // For backward compat if (isNotificationAllowed(command, cacheEntryEvictedListeners)) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_EVICTED); e.setKey(key); e.setValue(value); for (ListenerInvocation listener : cacheEntryEvictedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryInvalidated( final Object key, Object value, final boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryInvalidatedListeners)) { final boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_INVALIDATED); e.setOriginLocal(originLocal); e.setPre(pre); e.setKey(key); e.setValue(value); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryInvalidatedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryLoaded( Object key, Object value, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryLoadedListeners)) { boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_LOADED); e.setOriginLocal(originLocal); e.setPre(pre); e.setKey(key); e.setValue(value); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryLoadedListeners) listener.invoke(e); } } @Override public void notifyCacheEntryActivated( Object key, Object value, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryActivatedListeners)) { boolean originLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_ACTIVATED); e.setOriginLocal(originLocal); e.setPre(pre); e.setKey(key); e.setValue(value); setTx(ctx, e); for (ListenerInvocation listener : cacheEntryActivatedListeners) listener.invoke(e); } } private void setTx(InvocationContext ctx, EventImpl<Object, Object> e) { if (ctx != null && ctx.isInTxScope()) { GlobalTransaction tx = ((TxInvocationContext) ctx).getGlobalTransaction(); e.setTransactionId(tx); } } @Override public void notifyCacheEntryPassivated( Object key, Object value, boolean pre, InvocationContext ctx, FlagAffectedCommand command) { if (isNotificationAllowed(command, cacheEntryPassivatedListeners)) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, CACHE_ENTRY_PASSIVATED); e.setPre(pre); e.setKey(key); e.setValue(value); for (ListenerInvocation listener : cacheEntryPassivatedListeners) listener.invoke(e); } } @Override public void notifyTransactionCompleted( GlobalTransaction transaction, boolean successful, InvocationContext ctx) { if (!transactionCompletedListeners.isEmpty()) { boolean isOriginLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, TRANSACTION_COMPLETED); e.setOriginLocal(isOriginLocal); e.setTransactionId(transaction); e.setTransactionSuccessful(successful); for (ListenerInvocation listener : transactionCompletedListeners) listener.invoke(e); } } @Override public void notifyTransactionRegistered( GlobalTransaction globalTransaction, InvocationContext ctx) { if (!transactionRegisteredListeners.isEmpty()) { boolean isOriginLocal = ctx.isOriginLocal(); EventImpl<Object, Object> e = EventImpl.createEvent(cache, TRANSACTION_REGISTERED); e.setOriginLocal(isOriginLocal); e.setTransactionId(globalTransaction); for (ListenerInvocation listener : transactionRegisteredListeners) listener.invoke(e); } } @Override public void notifyDataRehashed( ConsistentHash oldCH, ConsistentHash newCH, int newTopologyId, boolean pre) { if (!dataRehashedListeners.isEmpty()) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, DATA_REHASHED); e.setPre(pre); e.setConsistentHashAtStart(oldCH); e.setConsistentHashAtEnd(newCH); e.setNewTopologyId(newTopologyId); for (ListenerInvocation listener : dataRehashedListeners) listener.invoke(e); } } @Override public void notifyTopologyChanged( ConsistentHash oldConsistentHash, ConsistentHash newConsistentHash, int newTopologyId, boolean pre) { if (!topologyChangedListeners.isEmpty()) { EventImpl<Object, Object> e = EventImpl.createEvent(cache, TOPOLOGY_CHANGED); e.setPre(pre); e.setConsistentHashAtStart(oldConsistentHash); e.setConsistentHashAtEnd(newConsistentHash); e.setNewTopologyId(newTopologyId); for (ListenerInvocation listener : topologyChangedListeners) listener.invoke(e); } } public boolean isNotificationAllowed( FlagAffectedCommand cmd, List<ListenerInvocation> listeners) { return (cmd == null || !cmd.hasFlag(Flag.SKIP_LISTENER_NOTIFICATION)) && !listeners.isEmpty(); } }
/**
* Repository for {@link RemoteTransaction} and {@link
* org.infinispan.transaction.xa.TransactionXaAdapter}s (locally originated transactions).
*
* @author [email protected]
* @author Galder Zamarreño
* @since 4.0
*/
@Listener(sync = false)
public class TransactionTable {
public static final int CACHE_STOPPED_VIEW_ID = -1;
private static final Log log = LogFactory.getLog(TransactionTable.class);
private ConcurrentMap<Transaction, LocalTransaction> localTransactions;
private ConcurrentMap<GlobalTransaction, RemoteTransaction> remoteTransactions;
private final StaleTransactionCleanupService cleanupService =
new StaleTransactionCleanupService(this);
protected Configuration configuration;
protected InvocationContextContainer icc;
protected TransactionCoordinator txCoordinator;
protected TransactionFactory txFactory;
protected RpcManager rpcManager;
protected CommandsFactory commandsFactory;
private InterceptorChain invoker;
private CacheNotifier notifier;
private EmbeddedCacheManager cm;
private TransactionSynchronizationRegistry transactionSynchronizationRegistry;
protected ClusteringDependentLogic clusteringLogic;
protected boolean clustered = false;
private Lock minViewRecalculationLock;
/**
* minTxViewId is the minimum view ID across all ongoing local and remote transactions. It doesn't
* update on transaction creation, but only on removal. That's because it is not possible for a
* newly created transaction to have an bigger view ID than the current one.
*/
private volatile int minTxViewId = CACHE_STOPPED_VIEW_ID;
private volatile int currentViewId = CACHE_STOPPED_VIEW_ID;
private String cacheName;
@Inject
public void initialize(
RpcManager rpcManager,
Configuration configuration,
InvocationContextContainer icc,
InterceptorChain invoker,
CacheNotifier notifier,
TransactionFactory gtf,
EmbeddedCacheManager cm,
TransactionCoordinator txCoordinator,
TransactionSynchronizationRegistry transactionSynchronizationRegistry,
CommandsFactory commandsFactory,
ClusteringDependentLogic clusteringDependentLogic,
Cache cache) {
this.rpcManager = rpcManager;
this.configuration = configuration;
this.icc = icc;
this.invoker = invoker;
this.notifier = notifier;
this.txFactory = gtf;
this.cm = cm;
this.txCoordinator = txCoordinator;
this.transactionSynchronizationRegistry = transactionSynchronizationRegistry;
this.commandsFactory = commandsFactory;
this.clusteringLogic = clusteringDependentLogic;
this.cacheName = cache.getName();
}
@Start
private void start() {
final int concurrencyLevel = configuration.locking().concurrencyLevel();
localTransactions =
ConcurrentMapFactory.makeConcurrentMap(concurrencyLevel, 0.75f, concurrencyLevel);
if (configuration.clustering().cacheMode().isClustered()) {
minViewRecalculationLock = new ReentrantLock();
// Only initialize this if we are clustered.
remoteTransactions =
ConcurrentMapFactory.makeConcurrentMap(concurrencyLevel, 0.75f, concurrencyLevel);
cleanupService.start(cacheName, rpcManager, invoker);
cm.addListener(cleanupService);
cm.addListener(this);
notifier.addListener(cleanupService);
minTxViewId = rpcManager.getTransport().getViewId();
currentViewId = minTxViewId;
log.debugf("Min view id set to %s", minTxViewId);
clustered = true;
}
}
@Stop
private void stop() {
if (clustered) {
notifier.removeListener(cleanupService);
cm.removeListener(cleanupService);
cleanupService.stop();
cm.removeListener(this);
currentViewId = CACHE_STOPPED_VIEW_ID; // indicate that the cache has stopped
}
shutDownGracefully();
}
public Set<Object> getLockedKeysForRemoteTransaction(GlobalTransaction gtx) {
RemoteTransaction transaction = remoteTransactions.get(gtx);
if (transaction == null) return emptySet();
return transaction.getLockedKeys();
}
public void remoteTransactionPrepared(GlobalTransaction gtx) {
// do nothing
}
public void localTransactionPrepared(LocalTransaction localTransaction) {
// nothing, only used by recovery
}
public void enlist(Transaction transaction, LocalTransaction localTransaction) {
if (!localTransaction.isEnlisted()) {
SynchronizationAdapter sync =
new SynchronizationAdapter(
localTransaction,
txCoordinator,
commandsFactory,
rpcManager,
this,
clusteringLogic,
configuration);
if (transactionSynchronizationRegistry != null) {
try {
transactionSynchronizationRegistry.registerInterposedSynchronization(sync);
} catch (Exception e) {
log.failedSynchronizationRegistration(e);
throw new CacheException(e);
}
} else {
try {
transaction.registerSynchronization(sync);
} catch (Exception e) {
log.failedSynchronizationRegistration(e);
throw new CacheException(e);
}
}
((SyncLocalTransaction) localTransaction).setEnlisted(true);
}
}
public void failureCompletingTransaction(Transaction tx) {
final LocalTransaction localTransaction = localTransactions.get(tx);
if (localTransaction != null) {
removeLocalTransaction(localTransaction);
}
}
/**
* Returns true if the given transaction is already registered with the transaction table.
*
* @param tx if null false is returned
*/
public boolean containsLocalTx(Transaction tx) {
return tx != null && localTransactions.containsKey(tx);
}
public int getMinViewId() {
return minTxViewId;
}
protected void updateStateOnNodesLeaving(Collection<Address> leavers) {
Set<GlobalTransaction> toKill = new HashSet<GlobalTransaction>();
for (GlobalTransaction gt : remoteTransactions.keySet()) {
if (leavers.contains(gt.getAddress())) toKill.add(gt);
}
if (toKill.isEmpty())
log.tracef(
"No global transactions pertain to originator(s) %s who have left the cluster.", leavers);
else
log.tracef(
"%s global transactions pertain to leavers list %s and need to be killed",
toKill.size(), leavers);
for (GlobalTransaction gtx : toKill) {
log.tracef("Killing remote transaction originating on leaver %s", gtx);
RollbackCommand rc = new RollbackCommand(cacheName, gtx);
rc.init(invoker, icc, TransactionTable.this);
try {
rc.perform(null);
log.tracef("Rollback of transaction %s complete.", gtx);
} catch (Throwable e) {
log.unableToRollbackGlobalTx(gtx, e);
}
}
log.trace("Completed cleaning transactions originating on leavers");
}
/**
* Returns the {@link RemoteTransaction} associated with the supplied transaction id. Returns null
* if no such association exists.
*/
public RemoteTransaction getRemoteTransaction(GlobalTransaction txId) {
return remoteTransactions.get(txId);
}
public void remoteTransactionRollback(GlobalTransaction gtx) {
final RemoteTransaction remove = removeRemoteTransaction(gtx);
log.tracef("Removed local transaction %s? %b", gtx, remove);
}
/**
* Creates and register a {@link RemoteTransaction} with no modifications. Returns the created
* transaction.
*
* @throws IllegalStateException if an attempt to create a {@link RemoteTransaction} for an
* already registered id is made.
*/
public RemoteTransaction createRemoteTransaction(
GlobalTransaction globalTx, WriteCommand[] modifications) {
RemoteTransaction remoteTransaction =
modifications == null
? txFactory.newRemoteTransaction(globalTx, currentViewId)
: txFactory.newRemoteTransaction(modifications, globalTx, currentViewId);
registerRemoteTransaction(globalTx, remoteTransaction);
return remoteTransaction;
}
private void registerRemoteTransaction(GlobalTransaction gtx, RemoteTransaction rtx) {
RemoteTransaction transaction = remoteTransactions.put(gtx, rtx);
if (transaction != null) {
log.remoteTxAlreadyRegistered();
throw new IllegalStateException(
"A remote transaction with the given id was already registered!!!");
}
log.tracef("Created and registered remote transaction %s", rtx);
}
/**
* Returns the {@link org.infinispan.transaction.xa.TransactionXaAdapter} corresponding to the
* supplied transaction. If none exists, will be created first.
*/
public LocalTransaction getOrCreateLocalTransaction(
Transaction transaction, TxInvocationContext ctx) {
LocalTransaction current = localTransactions.get(transaction);
if (current == null) {
Address localAddress = rpcManager != null ? rpcManager.getTransport().getAddress() : null;
GlobalTransaction tx = txFactory.newGlobalTransaction(localAddress, false);
current =
txFactory.newLocalTransaction(
transaction, tx, ctx.isImplicitTransaction(), currentViewId);
log.tracef("Created a new local transaction: %s", current);
localTransactions.put(transaction, current);
notifier.notifyTransactionRegistered(tx, ctx);
}
return current;
}
/**
* Removes the {@link org.infinispan.transaction.xa.TransactionXaAdapter} corresponding to the
* given tx. Returns true if such an tx exists.
*/
public boolean removeLocalTransaction(LocalTransaction localTransaction) {
return localTransaction != null
&& (removeLocalTransactionInternal(localTransaction.getTransaction()) != null);
}
public LocalTransaction removeLocalTransaction(Transaction tx) {
return removeLocalTransactionInternal(tx);
}
protected final LocalTransaction removeLocalTransactionInternal(Transaction tx) {
LocalTransaction removed;
removed = localTransactions.remove(tx);
releaseResources(removed);
return removed;
}
private void releaseResources(CacheTransaction cacheTransaction) {
if (cacheTransaction != null) {
if (clustered) {
recalculateMinViewIdIfNeeded(cacheTransaction);
}
log.tracef("Removed %s from transaction table.", cacheTransaction);
cacheTransaction.notifyOnTransactionFinished();
}
}
/** Removes the {@link RemoteTransaction} corresponding to the given tx. */
public void remoteTransactionCommitted(GlobalTransaction gtx) {
if (Configurations.isSecondPhaseAsync(configuration)) {
removeRemoteTransaction(gtx);
}
}
public final RemoteTransaction removeRemoteTransaction(GlobalTransaction txId) {
RemoteTransaction removed;
removed = remoteTransactions.remove(txId);
releaseResources(removed);
return removed;
}
public int getRemoteTxCount() {
return remoteTransactions.size();
}
public int getLocalTxCount() {
return localTransactions.size();
}
public LocalTransaction getLocalTransaction(Transaction tx) {
return localTransactions.get(tx);
}
public boolean containRemoteTx(GlobalTransaction globalTransaction) {
return remoteTransactions.containsKey(globalTransaction);
}
public Collection<RemoteTransaction> getRemoteTransactions() {
return remoteTransactions.values();
}
protected final LocalTransaction getLocalTx(Transaction tx) {
return localTransactions.get(tx);
}
public final Collection<LocalTransaction> getLocalTransactions() {
return localTransactions.values();
}
protected final void recalculateMinViewIdIfNeeded(CacheTransaction removedTransaction) {
if (removedTransaction == null)
throw new IllegalArgumentException("Transaction cannot be null!");
if (currentViewId != CACHE_STOPPED_VIEW_ID) {
// Assume that we only get here if we are clustered.
int removedTransactionViewId = removedTransaction.getViewId();
if (removedTransactionViewId < minTxViewId) {
log.tracef(
"A transaction has a view ID (%s) that is smaller than the smallest transaction view ID (%s) this node knows about! This can happen if a concurrent thread recalculates the minimum view ID after the current transaction has been removed from the transaction table.",
removedTransactionViewId, minTxViewId);
} else if (removedTransactionViewId == minTxViewId
&& removedTransactionViewId < currentViewId) {
// We should only need to re-calculate the minimum view ID if the transaction being
// completed
// has the same ID as the smallest known transaction ID, to check what the new smallest is,
// and this is
// not the current view ID.
calculateMinViewId(removedTransactionViewId);
}
}
}
@ViewChanged
public void recalculateMinViewIdOnTopologyChange(ViewChangedEvent vce) {
// don't do anything if this cache is not clustered - view changes are global
if (clustered) {
log.debugf("View changed, recalculating minViewId");
currentViewId = vce.getViewId();
calculateMinViewId(-1);
}
}
/**
* This method calculates the minimum view ID known by the current node. This method is only used
* in a clustered cache, and only invoked when either a view change is detected, or a transaction
* whose view ID is not the same as the current view ID.
*
* <p>This method is guarded by minViewRecalculationLock to prevent concurrent updates to the
* minimum view ID field.
*
* @param idOfRemovedTransaction the view ID associated with the transaction that triggered this
* recalculation, or -1 if triggered by a view change event.
*/
@GuardedBy("minViewRecalculationLock")
private void calculateMinViewId(int idOfRemovedTransaction) {
minViewRecalculationLock.lock();
try {
// We should only need to re-calculate the minimum view ID if the transaction being completed
// has the same ID as the smallest known transaction ID, to check what the new smallest is.
// We do this check
// again here, since this is now within a synchronized method.
if (idOfRemovedTransaction == -1
|| (idOfRemovedTransaction == minTxViewId && idOfRemovedTransaction < currentViewId)) {
int minViewIdFound = currentViewId;
for (CacheTransaction ct : localTransactions.values()) {
int viewId = ct.getViewId();
if (viewId < minViewIdFound) minViewIdFound = viewId;
}
for (CacheTransaction ct : remoteTransactions.values()) {
int viewId = ct.getViewId();
if (viewId < minViewIdFound) minViewIdFound = viewId;
}
if (minViewIdFound > minTxViewId) {
log.tracef("Changing minimum view ID from %s to %s", minTxViewId, minViewIdFound);
minTxViewId = minViewIdFound;
} else {
log.tracef("Minimum view ID still is %s; nothing to change", minViewIdFound);
}
}
} finally {
minViewRecalculationLock.unlock();
}
}
private boolean areTxsOnGoing() {
return !localTransactions.isEmpty()
|| (remoteTransactions != null && !remoteTransactions.isEmpty());
}
private void shutDownGracefully() {
if (log.isDebugEnabled())
log.debugf(
"Wait for on-going transactions to finish for %s.",
Util.prettyPrintTime(
configuration.transaction().cacheStopTimeout(), TimeUnit.MILLISECONDS));
long failTime = currentMillisFromNanotime() + configuration.transaction().cacheStopTimeout();
boolean txsOnGoing = areTxsOnGoing();
while (txsOnGoing && currentMillisFromNanotime() < failTime) {
try {
Thread.sleep(30);
txsOnGoing = areTxsOnGoing();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
if (clustered) {
log.debugf(
"Interrupted waiting for on-going transactions to finish. %s local transactions and %s remote transactions",
localTransactions.size(), remoteTransactions.size());
} else {
log.debugf(
"Interrupted waiting for %s on-going transactions to finish.",
localTransactions.size());
}
}
}
if (txsOnGoing) {
log.unfinishedTransactionsRemain(
localTransactions == null ? 0 : localTransactions.size(),
remoteTransactions == null ? 0 : remoteTransactions.size());
} else {
log.debug("All transactions terminated");
}
}
}
public class L1ManagerImpl implements L1Manager {
private static final Log log = LogFactory.getLog(L1ManagerImpl.class);
private final boolean trace = log.isTraceEnabled();
private RpcManager rpcManager;
private CommandsFactory commandsFactory;
private int threshold;
private long rpcTimeout;
private final ConcurrentMap<Object, Collection<Address>> requestors;
public L1ManagerImpl() {
requestors = ConcurrentMapFactory.makeConcurrentMap();
}
@Inject
public void init(
Configuration configuration, RpcManager rpcManager, CommandsFactory commandsFactory) {
this.rpcManager = rpcManager;
this.commandsFactory = commandsFactory;
this.threshold = configuration.getL1InvalidationThreshold();
this.rpcTimeout = configuration.getSyncReplTimeout();
}
public void addRequestor(Object key, Address origin) {
// we do a plain get first as that's likely to be enough
Collection<Address> as = requestors.get(key);
if (as == null) {
// only if needed we create a new HashSet, but make sure we don't replace another one being
// created
as = new ConcurrentHashSet<Address>();
as.add(origin);
Collection<Address> previousAs = requestors.putIfAbsent(key, as);
if (previousAs != null) {
// another thread added it already, so use his copy and discard our proposed instance
previousAs.add(origin);
}
} else {
as.add(origin);
}
}
public NotifyingNotifiableFuture<Object> flushCache(
Collection<Object> keys, Object retval, Address origin) {
if (trace) log.tracef("Invalidating L1 caches for keys %s", keys);
NotifyingNotifiableFuture<Object> future = new AggregatingNotifyingFutureImpl(retval, 2);
Collection<Address> invalidationAddresses = buildInvalidationAddressList(keys, origin);
int nodes = invalidationAddresses.size();
if (nodes > 0) {
// No need to invalidate at all if there is no one to invalidate!
boolean multicast = isUseMulticast(nodes);
if (trace)
log.tracef(
"There are %s nodes involved in invalidation. Threshold is: %s; using multicast: %s",
nodes, threshold, multicast);
if (multicast) {
if (trace) log.tracef("Invalidating keys %s via multicast", keys);
InvalidateCommand ic = commandsFactory.buildInvalidateFromL1Command(origin, false, keys);
rpcManager.broadcastRpcCommandInFuture(ic, future);
} else {
InvalidateCommand ic = commandsFactory.buildInvalidateFromL1Command(origin, false, keys);
// Ask the caches who have requested from us to remove
if (trace) log.tracef("Keys %s needs invalidation on %s", keys, invalidationAddresses);
rpcManager.invokeRemotelyInFuture(
invalidationAddresses, ic, true, future, rpcTimeout, true);
return future;
}
} else if (trace) log.trace("No L1 caches to invalidate");
return future;
}
private Collection<Address> buildInvalidationAddressList(
Collection<Object> keys, Address origin) {
Collection<Address> addresses = new HashSet<Address>(2);
for (Object key : keys) {
Collection<Address> as = requestors.remove(key);
if (as != null) {
addresses.addAll(as);
if (origin != null && as.contains(origin)) addRequestor(key, origin);
}
}
if (origin != null) addresses.remove(origin);
return addresses;
}
private boolean isUseMulticast(int nodes) {
// User has requested unicast only
if (threshold == -1) return false;
// Underlying transport is not multicast capable
if (!rpcManager.getTransport().isMulticastCapable()) return false;
// User has requested multicast only
if (threshold == 0) return true;
// we decide:
return nodes > threshold;
}
}
/**
* @author Pedro Ruivo
* @since 5.2
*/
public class GMUEntryWrappingInterceptor extends EntryWrappingInterceptor {
private static final Log log = LogFactory.getLog(GMUEntryWrappingInterceptor.class);
protected GMUVersionGenerator versionGenerator;
private TransactionCommitManager transactionCommitManager;
@Inject
public void inject(
TransactionCommitManager transactionCommitManager,
DataContainer dataContainer,
CommitLog commitLog,
VersionGenerator versionGenerator) {
this.transactionCommitManager = transactionCommitManager;
this.versionGenerator = toGMUVersionGenerator(versionGenerator);
}
@Override
public Object visitPrepareCommand(TxInvocationContext ctx, PrepareCommand command)
throws Throwable {
GMUPrepareCommand spc = convert(command, GMUPrepareCommand.class);
if (ctx.isOriginLocal()) {
spc.setVersion(ctx.getTransactionVersion());
spc.setReadSet(ctx.getReadSet());
} else {
ctx.setTransactionVersion(spc.getPrepareVersion());
}
wrapEntriesForPrepare(ctx, command);
performValidation(ctx, spc);
Object retVal = invokeNextInterceptor(ctx, command);
if (ctx.isOriginLocal() && command.getModifications().length > 0) {
EntryVersion commitVersion =
calculateCommitVersion(
ctx.getTransactionVersion(),
versionGenerator,
cll.getWriteOwners(ctx.getCacheTransaction()));
ctx.setTransactionVersion(commitVersion);
} else {
retVal = ctx.getTransactionVersion();
}
if (command.isOnePhaseCommit()) {
commitContextEntries.commitContextEntries(ctx);
}
return retVal;
}
@Override
public Object visitCommitCommand(TxInvocationContext ctx, CommitCommand command)
throws Throwable {
GMUCommitCommand gmuCommitCommand = convert(command, GMUCommitCommand.class);
if (ctx.isOriginLocal()) {
gmuCommitCommand.setCommitVersion(ctx.getTransactionVersion());
} else {
ctx.setTransactionVersion(gmuCommitCommand.getCommitVersion());
}
transactionCommitManager.commitTransaction(
ctx.getCacheTransaction(), gmuCommitCommand.getCommitVersion());
Object retVal = null;
try {
retVal = invokeNextInterceptor(ctx, command);
} catch (Throwable throwable) {
// let ignore the exception. we cannot have some nodes applying the write set and another not
// another one
// receives the rollback and don't applies the write set
} finally {
transactionCommitManager.awaitUntilCommitted(
ctx.getCacheTransaction(), ctx.isOriginLocal() ? null : gmuCommitCommand);
}
return ctx.isOriginLocal() ? retVal : RequestHandler.DO_NOT_REPLY;
}
@Override
public Object visitRollbackCommand(TxInvocationContext ctx, RollbackCommand command)
throws Throwable {
try {
return invokeNextInterceptor(ctx, command);
} finally {
transactionCommitManager.rollbackTransaction(ctx.getCacheTransaction());
}
}
/*
* NOTE: these are the only commands that passes values to the application and these keys needs to be validated
* and added to the transaction read set.
*/
@Override
public Object visitGetKeyValueCommand(InvocationContext ctx, GetKeyValueCommand command)
throws Throwable {
ctx.clearKeyReadInCommand();
Object retVal = super.visitGetKeyValueCommand(ctx, command);
updateTransactionVersion(ctx);
return retVal;
}
@Override
public Object visitPutKeyValueCommand(InvocationContext ctx, PutKeyValueCommand command)
throws Throwable {
ctx.clearKeyReadInCommand();
Object retVal = super.visitPutKeyValueCommand(ctx, command);
updateTransactionVersion(ctx);
return retVal;
}
@Override
public Object visitRemoveCommand(InvocationContext ctx, RemoveCommand command) throws Throwable {
ctx.clearKeyReadInCommand();
Object retVal = super.visitRemoveCommand(ctx, command);
updateTransactionVersion(ctx);
return retVal;
}
@Override
public Object visitReplaceCommand(InvocationContext ctx, ReplaceCommand command)
throws Throwable {
ctx.clearKeyReadInCommand();
Object retVal = super.visitReplaceCommand(ctx, command);
updateTransactionVersion(ctx);
return retVal;
}
/**
* validates the read set and returns the prepare version from the commit queue
*
* @param ctx the context
* @param command the prepare command
* @throws InterruptedException if interrupted
*/
protected void performValidation(TxInvocationContext ctx, GMUPrepareCommand command)
throws InterruptedException {
boolean hasToUpdateLocalKeys = hasLocalKeysToUpdate(command.getModifications());
boolean isReadOnly = command.getModifications().length == 0;
if (!isReadOnly) {
cll.performReadSetValidation(ctx, command);
if (hasToUpdateLocalKeys) {
transactionCommitManager.prepareTransaction(ctx.getCacheTransaction());
} else {
transactionCommitManager.prepareReadOnlyTransaction(ctx.getCacheTransaction());
}
}
if (log.isDebugEnabled()) {
log.debugf(
"Transaction %s can commit on this node. Prepare Version is %s",
command.getGlobalTransaction().prettyPrint(), ctx.getTransactionVersion());
}
}
private void updateTransactionVersion(InvocationContext context) {
if (!context.isInTxScope() && !context.isOriginLocal()) {
return;
}
if (context instanceof SingleKeyNonTxInvocationContext) {
if (log.isDebugEnabled()) {
log.debugf(
"Received a SingleKeyNonTxInvocationContext... This should be a single read operation");
}
return;
}
TxInvocationContext txInvocationContext = (TxInvocationContext) context;
List<EntryVersion> entryVersionList = new LinkedList<EntryVersion>();
entryVersionList.add(txInvocationContext.getTransactionVersion());
if (log.isTraceEnabled()) {
log.tracef(
"[%s] Keys read in this command: %s",
txInvocationContext.getGlobalTransaction().prettyPrint(),
txInvocationContext.getKeysReadInCommand());
}
for (InternalGMUCacheEntry internalGMUCacheEntry :
txInvocationContext.getKeysReadInCommand().values()) {
Object key = internalGMUCacheEntry.getKey();
boolean local = cll.localNodeIsOwner(key);
if (log.isTraceEnabled()) {
log.tracef(
"[%s] Analyze entry [%s]: local?=%s",
txInvocationContext.getGlobalTransaction().prettyPrint(), internalGMUCacheEntry, local);
}
if (txInvocationContext.hasModifications() && !internalGMUCacheEntry.isMostRecent()) {
throw new CacheException("Read-Write transaction read an old value and should rollback");
}
if (internalGMUCacheEntry.getMaximumTransactionVersion() != null) {
entryVersionList.add(internalGMUCacheEntry.getMaximumTransactionVersion());
}
txInvocationContext.getCacheTransaction().addReadKey(key);
if (local) {
txInvocationContext.setAlreadyReadOnThisNode(true);
txInvocationContext.addReadFrom(cll.getAddress());
}
}
if (entryVersionList.size() > 1) {
EntryVersion[] txVersionArray = new EntryVersion[entryVersionList.size()];
txInvocationContext.setTransactionVersion(
versionGenerator.mergeAndMax(entryVersionList.toArray(txVersionArray)));
}
}
private boolean hasLocalKeysToUpdate(WriteCommand[] modifications) {
for (WriteCommand writeCommand : modifications) {
if (writeCommand instanceof ClearCommand) {
return true;
} else if (writeCommand instanceof ApplyDeltaCommand) {
if (cll.localNodeIsOwner(((ApplyDeltaCommand) writeCommand).getDeltaAwareKey())) {
return true;
}
} else {
for (Object key : writeCommand.getAffectedKeys()) {
if (cll.localNodeIsOwner(key)) {
return true;
}
}
}
}
return false;
}
}
/** * @author [email protected] * @author Galder Zamarreño * @author Sanne Grinovero <*****@*****.**> (C) 2011 Red Hat Inc. * @since 4.0 */ public class CommandsFactoryImpl implements CommandsFactory { private static final Log log = LogFactory.getLog(CommandsFactoryImpl.class); private static final boolean trace = log.isTraceEnabled(); private DataContainer dataContainer; private CacheNotifier notifier; private Cache cache; private String cacheName; // some stateless commands can be reused so that they aren't constructed again all the time. SizeCommand cachedSizeCommand; KeySetCommand cachedKeySetCommand; ValuesCommand cachedValuesCommand; EntrySetCommand cachedEntrySetCommand; private InterceptorChain interceptorChain; private DistributionManager distributionManager; private InvocationContextContainer icc; private TransactionTable txTable; private Configuration configuration; private RecoveryManager recoveryManager; private Map<Byte, ModuleCommandInitializer> moduleCommandInitializers; @Inject public void setupDependencies( DataContainer container, CacheNotifier notifier, Cache cache, InterceptorChain interceptorChain, DistributionManager distributionManager, InvocationContextContainer icc, TransactionTable txTable, Configuration configuration, @ComponentName(KnownComponentNames.MODULE_COMMAND_INITIALIZERS) Map<Byte, ModuleCommandInitializer> moduleCommandInitializers, RecoveryManager recoveryManager) { this.dataContainer = container; this.notifier = notifier; this.cache = cache; this.interceptorChain = interceptorChain; this.distributionManager = distributionManager; this.icc = icc; this.txTable = txTable; this.configuration = configuration; this.moduleCommandInitializers = moduleCommandInitializers; this.recoveryManager = recoveryManager; } @Start(priority = 1) // needs to happen early on public void start() { cacheName = cache.getName(); } public PutKeyValueCommand buildPutKeyValueCommand( Object key, Object value, long lifespanMillis, long maxIdleTimeMillis, Set<Flag> flags) { return new PutKeyValueCommand( key, value, false, notifier, lifespanMillis, maxIdleTimeMillis, flags); } public RemoveCommand buildRemoveCommand(Object key, Object value, Set<Flag> flags) { return new RemoveCommand(key, value, notifier, flags); } public InvalidateCommand buildInvalidateCommand(Object... keys) { return new InvalidateCommand(notifier, keys); } public InvalidateCommand buildInvalidateFromL1Command(boolean forRehash, Object... keys) { return new InvalidateL1Command( forRehash, dataContainer, configuration, distributionManager, notifier, keys); } public InvalidateCommand buildInvalidateFromL1Command( boolean forRehash, Collection<Object> keys) { return new InvalidateL1Command( forRehash, dataContainer, configuration, distributionManager, notifier, keys); } public ReplaceCommand buildReplaceCommand( Object key, Object oldValue, Object newValue, long lifespan, long maxIdleTimeMillis, Set<Flag> flags) { return new ReplaceCommand(key, oldValue, newValue, lifespan, maxIdleTimeMillis, flags); } public SizeCommand buildSizeCommand() { if (cachedSizeCommand == null) { cachedSizeCommand = new SizeCommand(dataContainer); } return cachedSizeCommand; } public KeySetCommand buildKeySetCommand() { if (cachedKeySetCommand == null) { cachedKeySetCommand = new KeySetCommand(dataContainer); } return cachedKeySetCommand; } public ValuesCommand buildValuesCommand() { if (cachedValuesCommand == null) { cachedValuesCommand = new ValuesCommand(dataContainer); } return cachedValuesCommand; } public EntrySetCommand buildEntrySetCommand() { if (cachedEntrySetCommand == null) { cachedEntrySetCommand = new EntrySetCommand(dataContainer); } return cachedEntrySetCommand; } public GetKeyValueCommand buildGetKeyValueCommand(Object key, Set<Flag> flags) { return new GetKeyValueCommand(key, notifier, flags); } public PutMapCommand buildPutMapCommand( Map map, long lifespan, long maxIdleTimeMillis, Set<Flag> flags) { return new PutMapCommand(map, notifier, lifespan, maxIdleTimeMillis, flags); } public ClearCommand buildClearCommand(Set<Flag> flags) { return new ClearCommand(notifier, flags); } public EvictCommand buildEvictCommand(Object key) { return new EvictCommand(key, notifier); } public PrepareCommand buildPrepareCommand( GlobalTransaction gtx, List<WriteCommand> modifications, boolean onePhaseCommit) { PrepareCommand command = new PrepareCommand(gtx, modifications, onePhaseCommit); command.setCacheName(cacheName); return command; } public CommitCommand buildCommitCommand(GlobalTransaction gtx) { CommitCommand commitCommand = new CommitCommand(gtx); commitCommand.setCacheName(cacheName); return commitCommand; } public RollbackCommand buildRollbackCommand(GlobalTransaction gtx) { RollbackCommand rollbackCommand = new RollbackCommand(gtx); rollbackCommand.setCacheName(cacheName); return rollbackCommand; } public MultipleRpcCommand buildReplicateCommand(List<ReplicableCommand> toReplicate) { return new MultipleRpcCommand(toReplicate, cacheName); } public SingleRpcCommand buildSingleRpcCommand(ReplicableCommand call) { return new SingleRpcCommand(cacheName, call); } public StateTransferControlCommand buildStateTransferControlCommand(boolean block) { return new StateTransferControlCommand(block); } public ClusteredGetCommand buildClusteredGetCommand(Object key, Set<Flag> flags) { return new ClusteredGetCommand(key, cacheName, flags); } /** @param isRemote true if the command is deserialized and is executed remote. */ public void initializeReplicableCommand(ReplicableCommand c, boolean isRemote) { if (c == null) return; switch (c.getCommandId()) { case PutKeyValueCommand.COMMAND_ID: ((PutKeyValueCommand) c).init(notifier); break; case PutMapCommand.COMMAND_ID: ((PutMapCommand) c).init(notifier); break; case RemoveCommand.COMMAND_ID: ((RemoveCommand) c).init(notifier); break; case MultipleRpcCommand.COMMAND_ID: MultipleRpcCommand rc = (MultipleRpcCommand) c; rc.init(interceptorChain, icc); if (rc.getCommands() != null) for (ReplicableCommand nested : rc.getCommands()) { initializeReplicableCommand(nested, false); } break; case SingleRpcCommand.COMMAND_ID: SingleRpcCommand src = (SingleRpcCommand) c; src.init(interceptorChain, icc); if (src.getCommand() != null) initializeReplicableCommand(src.getCommand(), false); break; case InvalidateCommand.COMMAND_ID: InvalidateCommand ic = (InvalidateCommand) c; ic.init(notifier); break; case InvalidateL1Command.COMMAND_ID: InvalidateL1Command ilc = (InvalidateL1Command) c; ilc.init(configuration, distributionManager, notifier, dataContainer); break; case PrepareCommand.COMMAND_ID: PrepareCommand pc = (PrepareCommand) c; pc.init(interceptorChain, icc, txTable); pc.initialize(notifier, recoveryManager); if (pc.getModifications() != null) for (ReplicableCommand nested : pc.getModifications()) { initializeReplicableCommand(nested, false); } pc.markTransactionAsRemote(isRemote); if (configuration.isEnableDeadlockDetection() && isRemote) { DldGlobalTransaction transaction = (DldGlobalTransaction) pc.getGlobalTransaction(); transaction.setLocksHeldAtOrigin(pc.getAffectedKeys()); } break; case CommitCommand.COMMAND_ID: CommitCommand commitCommand = (CommitCommand) c; commitCommand.init(interceptorChain, icc, txTable); commitCommand.markTransactionAsRemote(isRemote); break; case RollbackCommand.COMMAND_ID: RollbackCommand rollbackCommand = (RollbackCommand) c; rollbackCommand.init(interceptorChain, icc, txTable); rollbackCommand.markTransactionAsRemote(isRemote); break; case ClearCommand.COMMAND_ID: ClearCommand cc = (ClearCommand) c; cc.init(notifier); break; case ClusteredGetCommand.COMMAND_ID: ClusteredGetCommand clusteredGetCommand = (ClusteredGetCommand) c; clusteredGetCommand.initialize(icc, this, interceptorChain, distributionManager); break; case LockControlCommand.COMMAND_ID: LockControlCommand lcc = (LockControlCommand) c; lcc.init(interceptorChain, icc, txTable); lcc.markTransactionAsRemote(isRemote); if (configuration.isEnableDeadlockDetection() && isRemote) { DldGlobalTransaction gtx = (DldGlobalTransaction) lcc.getGlobalTransaction(); RemoteTransaction transaction = txTable.getRemoteTransaction(gtx); if (transaction != null) { if (!configuration.getCacheMode().isDistributed()) { Set<Object> keys = txTable.getLockedKeysForRemoteTransaction(gtx); GlobalTransaction gtx2 = transaction.getGlobalTransaction(); ((DldGlobalTransaction) gtx2).setLocksHeldAtOrigin(keys); gtx.setLocksHeldAtOrigin(keys); } else { GlobalTransaction gtx2 = transaction.getGlobalTransaction(); ((DldGlobalTransaction) gtx2).setLocksHeldAtOrigin(gtx.getLocksHeldAtOrigin()); } } } break; case RehashControlCommand.COMMAND_ID: RehashControlCommand rcc = (RehashControlCommand) c; rcc.init(distributionManager, configuration, dataContainer, this); break; case GetInDoubtTransactionsCommand.COMMAND_ID: GetInDoubtTransactionsCommand gptx = (GetInDoubtTransactionsCommand) c; gptx.init(recoveryManager); break; case RemoveRecoveryInfoCommand.COMMAND_ID: RemoveRecoveryInfoCommand ftx = (RemoveRecoveryInfoCommand) c; ftx.init(recoveryManager); break; case MapReduceCommand.COMMAND_ID: MapReduceCommand mrc = (MapReduceCommand) c; mrc.init( this, interceptorChain, icc, distributionManager, cache.getAdvancedCache().getRpcManager().getAddress()); break; case DistributedExecuteCommand.COMMAND_ID: DistributedExecuteCommand dec = (DistributedExecuteCommand) c; dec.init(cache); break; case GetInDoubtTxInfoCommand.COMMAND_ID: GetInDoubtTxInfoCommand gidTxInfoCommand = (GetInDoubtTxInfoCommand) c; gidTxInfoCommand.init(recoveryManager); break; case CompleteTransactionCommand.COMMAND_ID: CompleteTransactionCommand ccc = (CompleteTransactionCommand) c; ccc.init(recoveryManager); break; default: ModuleCommandInitializer mci = moduleCommandInitializers.get(c.getCommandId()); if (mci != null) { mci.initializeReplicableCommand(c, isRemote); } else { if (trace) log.tracef("Nothing to initialize for command: %s", c); } } } public LockControlCommand buildLockControlCommand( Collection keys, boolean implicit, Set<Flag> flags) { return new LockControlCommand(keys, cacheName, flags, implicit); } public RehashControlCommand buildRehashControlCommand( RehashControlCommand.Type type, Address sender, int viewId) { return new RehashControlCommand(cacheName, type, sender, viewId); } public RehashControlCommand buildRehashControlCommand( RehashControlCommand.Type type, Address sender, int viewId, Map<Object, InternalCacheValue> state, ConsistentHash oldCH, ConsistentHash newCH) { return new RehashControlCommand(cacheName, type, sender, viewId, state, oldCH, newCH); } public String getCacheName() { return cacheName; } @Override public GetInDoubtTransactionsCommand buildGetInDoubtTransactionsCommand() { return new GetInDoubtTransactionsCommand(cacheName); } @Override public RemoveRecoveryInfoCommand buildRemoveRecoveryInfoCommand(Xid xid) { return new RemoveRecoveryInfoCommand(xid, cacheName); } @Override public <T> DistributedExecuteCommand<T> buildDistributedExecuteCommand( Callable<T> callable, Address sender, Collection keys) { return new DistributedExecuteCommand(keys, callable); } @Override public MapReduceCommand buildMapReduceCommand( Mapper m, Reducer r, Address sender, Collection keys) { return new MapReduceCommand(m, r, cacheName, keys); } @Override public GetInDoubtTxInfoCommand buildGetInDoubtTxInfoCommand() { return new GetInDoubtTxInfoCommand(cacheName); } @Override public CompleteTransactionCommand buildCompleteTransactionCommand(Xid xid, boolean commit) { return new CompleteTransactionCommand(cacheName, xid, commit); } @Override public RemoveRecoveryInfoCommand buildRemoveRecoveryInfoCommand(long internalId) { return new RemoveRecoveryInfoCommand(internalId, cacheName); } }
/**
* {@link SyncConsistentHashFactory} adapted for replicated caches, so that the primary owner of a
* key is the same in replicated and distributed caches.
*
* @author Dan Berindei
* @since 8.2
*/
public class SyncReplicatedConsistentHashFactory
implements ConsistentHashFactory<ReplicatedConsistentHash> {
private static final Log log = LogFactory.getLog(SyncReplicatedConsistentHashFactory.class);
public static final float OWNED_SEGMENTS_ALLOWED_VARIATION = 1.10f;
public static final float PRIMARY_SEGMENTS_ALLOWED_VARIATION = 1.20f;
private static final SyncConsistentHashFactory syncCHF = new SyncConsistentHashFactory();
@Override
public ReplicatedConsistentHash create(
Hash hashFunction,
int numOwners,
int numSegments,
List<Address> members,
Map<Address, Float> capacityFactors) {
DefaultConsistentHash dch = syncCHF.create(hashFunction, 1, numSegments, members, null);
return replicatedFromDefault(dch);
}
@Override
public ReplicatedConsistentHash fromPersistentState(ScopedPersistentState state) {
String consistentHashClass = state.getProperty("consistentHash");
if (!ReplicatedConsistentHash.class.getName().equals(consistentHashClass))
throw log.persistentConsistentHashMismatch(this.getClass().getName(), consistentHashClass);
return new ReplicatedConsistentHash(state);
}
private ReplicatedConsistentHash replicatedFromDefault(DefaultConsistentHash dch) {
int numSegments = dch.getNumSegments();
List<Address> members = dch.getMembers();
int[] primaryOwners = new int[numSegments];
for (int segment = 0; segment < numSegments; segment++) {
primaryOwners[segment] = members.indexOf(dch.locatePrimaryOwnerForSegment(segment));
}
return new ReplicatedConsistentHash(dch.getHashFunction(), members, primaryOwners);
}
@Override
public ReplicatedConsistentHash updateMembers(
ReplicatedConsistentHash baseCH,
List<Address> newMembers,
Map<Address, Float> actualCapacityFactors) {
DefaultConsistentHash baseDCH = defaultFromReplicated(baseCH);
DefaultConsistentHash dch = syncCHF.updateMembers(baseDCH, newMembers, null);
return replicatedFromDefault(dch);
}
private DefaultConsistentHash defaultFromReplicated(ReplicatedConsistentHash baseCH) {
int numSegments = baseCH.getNumSegments();
List<Address>[] baseSegmentOwners = new List[numSegments];
for (int segment = 0; segment < numSegments; segment++) {
baseSegmentOwners[segment] =
Collections.singletonList(baseCH.locatePrimaryOwnerForSegment(segment));
}
return new DefaultConsistentHash(
baseCH.getHashFunction(), 1, numSegments, baseCH.getMembers(), null, baseSegmentOwners);
}
@Override
public ReplicatedConsistentHash rebalance(ReplicatedConsistentHash baseCH) {
DefaultConsistentHash baseDCH = defaultFromReplicated(baseCH);
DefaultConsistentHash dch = syncCHF.rebalance(baseDCH);
return replicatedFromDefault(dch);
}
@Override
public ReplicatedConsistentHash union(
ReplicatedConsistentHash ch1, ReplicatedConsistentHash ch2) {
return ch1.union(ch2);
}
public static class Externalizer
extends AbstractExternalizer<SyncReplicatedConsistentHashFactory> {
@Override
public void writeObject(ObjectOutput output, SyncReplicatedConsistentHashFactory chf) {}
@Override
@SuppressWarnings("unchecked")
public SyncReplicatedConsistentHashFactory readObject(ObjectInput unmarshaller) {
return new SyncReplicatedConsistentHashFactory();
}
@Override
public Integer getId() {
return Ids.SYNC_REPLICATED_CONSISTENT_HASH_FACTORY;
}
@Override
public Set<Class<? extends SyncReplicatedConsistentHashFactory>> getTypeClasses() {
return Collections.<Class<? extends SyncReplicatedConsistentHashFactory>>singleton(
SyncReplicatedConsistentHashFactory.class);
}
}
}