package org.infinispan.query.statetransfer;
import org.apache.lucene.queryparser.classic.ParseException;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.configuration.cache.Index;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.query.CacheQuery;
import org.infinispan.query.test.Person;
import org.infinispan.statetransfer.JoiningNode;
import org.infinispan.test.CacheManagerCallable;
import org.infinispan.test.MultipleCacheManagersTest;
import org.infinispan.test.fwk.TransportFlags;
import java.util.List;
import static org.infinispan.query.helper.TestQueryHelperFactory.createCacheQuery;
import static org.infinispan.test.TestingUtil.withCacheManager;
import static org.testng.AssertJUnit.assertEquals;
/**
* Base class for state transfer and query related tests
*
* @author Galder Zamarreño
* @since 5.2
*/
public abstract class BaseReIndexingTest extends MultipleCacheManagersTest {
protected Person[] persons;
protected ConfigurationBuilder builder;
abstract protected void configureCache(ConfigurationBuilder builder);
@Override
protected void createCacheManagers() throws Throwable {
builder = getDefaultClusteredCacheConfig(CacheMode.REPL_SYNC, false);
builder.indexing().index(Index.ALL)
.addProperty("default.directory_provider", "ram")
.addProperty("lucene_version", "LUCENE_CURRENT");
configureCache(builder);
createClusteredCaches(2, builder);
}
private EmbeddedCacheManager createCacheManager() {
return addClusterEnabledCacheManager(
builder, new TransportFlags().withMerge(true));
}
protected void executeSimpleQuery(Cache<String, Person> cache) throws ParseException {
CacheQuery cacheQuery = createCacheQuery(cache, "blurb", "playing");
List<Object> found = cacheQuery.list();
int elems = found.size();
assertEquals(1, elems);
Object val = found.get(0);
Person expectedPerson = persons[0];
assertEquals(expectedPerson, val);
}
protected void loadCacheEntries(Cache<String, Person> cache) {
Person person1 = new Person();
person1.setName("NavinSurtani");
person1.setBlurb("Likes playing WoW");
person1.setAge(45);
Person person2 = new Person();
person2.setName("BigGoat");
person2.setBlurb("Eats grass");
person2.setAge(30);
Person person3 = new Person();
person3.setName("MiniGoat");
person3.setBlurb("Eats cheese");
person3.setAge(35);
Person person4 = new Person();
person4.setName("MightyGoat");
person4.setBlurb("Also eats grass");
person4.setAge(66);
persons = new Person[]{person1, person2, person3, person4};
// Put the 3 created objects in the cache
cache.put(person1.getName(), person1);
cache.put(person2.getName(), person2);
cache.put(person3.getName(), person3);
cache.put(person4.getName(), person4);
}
protected void addNodeCheckingContentsAndQuery() {
withCacheManager(new CacheManagerCallable(createCacheManager()) {
@Override
public void call() {
try {
// New node joining
JoiningNode newNode = new JoiningNode(cm);
Cache<String, Person> newCache = newNode.getCache();
newNode.waitForJoin(120000, caches().get(0), caches().get(1), newCache);
// Verify state transfer
int size = newCache.size();
assertEquals(4, size);
for (int i = 0; i < size; i++)
assertEquals(persons[i], newCache.get(persons[i].getName()));
// Repeat query on new node
executeSimpleQuery(newCache);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
});
}
}