package org.infinispan.xsite;
import static org.testng.AssertJUnit.assertTrue;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.BackupConfiguration;
import org.infinispan.configuration.cache.BackupConfigurationBuilder;
import org.infinispan.configuration.cache.BackupFailurePolicy;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.configuration.cache.Configuration;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.global.GlobalConfigurationBuilder;
import org.infinispan.manager.CacheContainer;
/**
* @author Mircea Markus
* @since 5.2
*/
public abstract class AbstractTwoSitesTest extends AbstractXSiteTest {
protected static final String LON = "LON";
protected static final String NYC = "NYC";
protected BackupFailurePolicy lonBackupFailurePolicy = BackupFailurePolicy.WARN;
protected boolean isLonBackupTransactional = false;
protected BackupConfiguration.BackupStrategy lonBackupStrategy = BackupConfiguration.BackupStrategy.SYNC;
protected String lonCustomFailurePolicyClass = null;
protected boolean use2Pc = false;
protected int initialClusterSize = 2;
/**
* If true, the caches from one site will backup to a cache having the same name remotely (mirror) and the backupFor
* config element won't be used.
*/
protected boolean implicitBackupCache = false;
@Override
protected void createSites() {
ConfigurationBuilder lon = lonConfigurationBuilder();
ConfigurationBuilder nyc = getNycActiveConfig();
nyc.sites().addBackup()
.site(LON)
.strategy(BackupConfiguration.BackupStrategy.SYNC)
.sites().addInUseBackupSite(LON);
createSite(LON, initialClusterSize, globalConfigurationBuilderForSite(LON), lon);
createSite(NYC, initialClusterSize, globalConfigurationBuilderForSite(NYC), nyc);
if (!implicitBackupCache) {
ConfigurationBuilder nycBackup = getDefaultClusteredCacheConfig(CacheMode.DIST_SYNC, false);
nycBackup.sites().backupFor().remoteSite(NYC).defaultRemoteCache();
startCache(LON, "nycBackup", nycBackup);
ConfigurationBuilder lonBackup = getDefaultClusteredCacheConfig(CacheMode.DIST_SYNC, isLonBackupTransactional);
lonBackup.sites().backupFor().remoteSite(LON).defaultRemoteCache();
startCache(NYC, "lonBackup", lonBackup);
Configuration lonBackupConfig = cache(NYC, "lonBackup", 0).getCacheConfiguration();
assertTrue(lonBackupConfig.sites().backupFor().isBackupFor(LON, CacheContainer.DEFAULT_CACHE_NAME));
}
}
protected ConfigurationBuilder lonConfigurationBuilder() {
ConfigurationBuilder lon = getLonActiveConfig();
BackupConfigurationBuilder lonBackupConfigurationBuilder = lon.sites().addBackup();
lonBackupConfigurationBuilder
.site(NYC)
.backupFailurePolicy(lonBackupFailurePolicy)
.strategy(lonBackupStrategy)
.failurePolicyClass(lonCustomFailurePolicyClass)
.useTwoPhaseCommit(use2Pc)
.sites().addInUseBackupSite(NYC);
adaptLONConfiguration(lonBackupConfigurationBuilder);
return lon;
}
protected GlobalConfigurationBuilder globalConfigurationBuilderForSite(String siteName) {
GlobalConfigurationBuilder builder = GlobalConfigurationBuilder.defaultClusteredBuilder();
builder.site().localSite(siteName);
return builder;
}
protected void adaptLONConfiguration(BackupConfigurationBuilder builder) {
//no-op
}
protected Cache<Object, Object> backup(String site) {
if (site.equals(LON)) return implicitBackupCache ? cache(NYC, 0) : cache(NYC, "lonBackup", 0);
if (site.equals(NYC)) return implicitBackupCache ? cache(LON, 0) : cache(LON, "nycBackup", 0);
throw new IllegalArgumentException("No such site: " + site);
}
protected String val(String site) {
return "v_" + site;
}
protected String key(String site) {
return "k_" + site;
}
protected abstract ConfigurationBuilder getNycActiveConfig();
protected abstract ConfigurationBuilder getLonActiveConfig();
}