package org.infinispan.distribution;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.infinispan.Cache;
import org.infinispan.container.DataContainer;
import org.infinispan.container.entries.ImmortalCacheEntry;
import org.infinispan.container.entries.InternalCacheEntry;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.remoting.transport.Address;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;
/**
* Test helper class
*
* @author Manik Surtani
* @since 4.2.1
*/
public class DistributionTestHelper {
public static String safeType(Object o) {
if (o == null) return "null";
return o.getClass().getSimpleName();
}
public static void assertIsInL1(Cache<?, ?> cache, Object key) {
DataContainer dc = cache.getAdvancedCache().getDataContainer();
InternalCacheEntry ice = dc.get(key);
assert ice != null : "Entry for key [" + key + "] should be in L1 on cache at [" + addressOf(cache) + "]!";
assert !(ice instanceof ImmortalCacheEntry) : "Entry for key [" + key + "] should have a lifespan on cache at [" + addressOf(cache) + "]!";
}
public static void assertIsNotInL1(Cache<?, ?> cache, Object key) {
DataContainer dc = cache.getAdvancedCache().getDataContainer();
InternalCacheEntry ice = dc.get(key);
assert ice == null : "Entry for key [" + key + "] should not be in data container at all on cache at [" + addressOf(cache) + "]!";
}
public static void assertIsInContainerImmortal(Cache<?, ?> cache, Object key) {
Log log = LogFactory.getLog(BaseDistFunctionalTest.class);
DataContainer dc = cache.getAdvancedCache().getDataContainer();
InternalCacheEntry ice = dc.get(key);
if (ice == null) {
String msg = "Entry for key [" + key + "] should be in data container on cache at [" + addressOf(cache) + "]!";
log.fatal(msg);
assert false : msg;
}
if (!(ice instanceof ImmortalCacheEntry)) {
String msg = "Entry for key [" + key + "] on cache at [" + addressOf(cache) + "] should be immortal but was [" + ice + "]!";
log.fatal(msg);
assert false : msg;
}
}
public static void assertIsInL1OrNull(Cache<?, ?> cache, Object key) {
Log log = LogFactory.getLog(BaseDistFunctionalTest.class);
DataContainer dc = cache.getAdvancedCache().getDataContainer();
InternalCacheEntry ice = dc.get(key);
if (ice instanceof ImmortalCacheEntry) {
String msg = "Entry for key [" + key + "] on cache at [" + addressOf(cache) + "] should be mortal or null but was [" + ice + "]!";
log.fatal(msg);
assert false : msg;
}
}
public static boolean isOwner(Cache<?, ?> c, Object key) {
DistributionManager dm = c.getAdvancedCache().getDistributionManager();
List<Address> ownerAddresses = dm.locate(key);
return ownerAddresses.contains(addressOf(c));
}
public static boolean isFirstOwner(Cache<?, ?> c, Object key) {
DistributionManager dm = c.getAdvancedCache().getDistributionManager();
Address primaryOwnerAddress = dm.getPrimaryLocation(key);
return addressOf(c).equals(primaryOwnerAddress);
}
public static boolean hasOwners(Object key, Cache<?, ?> primaryOwner, Cache<?, ?>... backupOwners) {
DistributionManager dm = primaryOwner.getAdvancedCache().getDistributionManager();
List<Address> ownerAddresses = dm.locate(key);
if (!addressOf(primaryOwner).equals(ownerAddresses.get(0)))
return false;
for (int i = 0; i < backupOwners.length; i++) {
if (!ownerAddresses.contains(addressOf(backupOwners[i])))
return false;
}
return true;
}
public static <K, V> Collection<Cache<K, V>> getOwners(Object key, List<Cache<K, V>> caches) {
List<Cache<K, V>> owners = new ArrayList<Cache<K, V>>();
for (Cache<K, V> c : caches) {
if (isFirstOwner(c, key)) {
owners.add(c);
break;
}
}
for (Cache<K, V> c : caches)
if (isOwner(c, key) && !isFirstOwner(c, key)) owners.add(c);
return owners;
}
public static <K, V> Cache<K, V> getFirstOwner(Object key, List<Cache<K, V>> caches) {
return getOwners(key, caches).iterator().next();
}
public static <K, V> Collection<Cache<K, V>> getNonOwners(Object key, List<Cache<K, V>> caches) {
List<Cache<K, V>> nonOwners = new ArrayList<Cache<K, V>>();
for (Cache<K, V> c : caches)
if (!isOwner(c, key)) nonOwners.add(c);
return nonOwners;
}
public static <K, V> Cache<K, V> getFirstNonOwner(Object key, List<Cache<K, V>> caches) {
return getNonOwners(key, caches).iterator().next();
}
public static Address addressOf(Cache<?, ?> cache) {
EmbeddedCacheManager cacheManager = cache.getCacheManager();
return cacheManager.getAddress();
}
}