@Test(timeout = 5000L)
public void testCacheLoaderAsyncLoadAll() throws InterruptedException {
final AtomicInteger loads = new AtomicInteger();
final CacheLoader<String, Integer> cacheLoader =
new CacheLoader<String, Integer>() {
@Override
public Integer load(String key) throws CacheLoaderException {
loads.incrementAndGet();
return Integer.valueOf(key);
}
@Override
public Map<String, Integer> loadAll(Iterable<? extends String> keys)
throws CacheLoaderException {
throw new UnsupportedOperationException("Not supported yet.");
}
};
final AtomicBoolean completed = new AtomicBoolean(false);
final CompletionListener completionListener =
new CompletionListener() {
@Override
public void onCompletion() {
completed.set(true);
}
@Override
public void onException(Exception e) {
System.err.println(e);
}
};
try (CachingProvider cachingProvider =
Caching.getCachingProvider(GuavaCachingProvider.class.getName())) {
CacheManager cacheManager = cachingProvider.getCacheManager();
MutableConfiguration<String, Integer> custom = new MutableConfiguration<>();
custom.setStoreByValue(false);
custom.setTypes(String.class, Integer.class);
custom.setReadThrough(true);
custom.setCacheLoaderFactory(
new Factory<CacheLoader<String, Integer>>() {
@Override
public CacheLoader<String, Integer> create() {
return cacheLoader;
}
});
Cache<String, Integer> loadingCache = cacheManager.createCache("loadingCache", custom);
loadingCache.put("1", 1);
loadingCache.put("2", 2);
loadingCache.put("3", 3);
Set<String> keys = Sets.newHashSet("1", "2", "3", "4", "5", "6");
loadingCache.loadAll(keys, false, completionListener);
while (!completed.get()) {
Thread.sleep(250);
}
assertEquals(3, loads.getAndSet(0));
completed.set(false);
loadingCache.loadAll(keys, true, completionListener);
while (!completed.get()) {
Thread.sleep(250);
}
assertEquals(6, loads.get());
assertEquals(Integer.valueOf(1), loadingCache.getAndRemove("1"));
assertEquals(Integer.valueOf(2), loadingCache.getAndRemove("2"));
assertEquals(Integer.valueOf(3), loadingCache.getAndRemove("3"));
assertEquals(Integer.valueOf(4), loadingCache.getAndRemove("4"));
assertEquals(Integer.valueOf(5), loadingCache.getAndRemove("5"));
assertEquals(Integer.valueOf(6), loadingCache.getAndRemove("6"));
}
}
@SuppressWarnings("rawtypes")
@Override
public void loadAll(Collection keys) {
cache.loadAll(keys);
}
