Skip to content

Commit

Permalink
Merge pull request #3986 from mdogan/fix/ea/jcache-listner
Browse files Browse the repository at this point in the history 
Fixed jcache sync listener test
  • Loading branch information
@mdogan
mdogan committed Nov 3, 2014
2 parents 78d65d7 + de21eee commit c0e3d4a
Show file tree
Hide file tree
Showing 2 changed files with 149 additions and 120 deletions.
139 changes: 19 additions & 120 deletions hazelcast-client/src/test/java/com/hazelcast/client/cache/JCacheClientListenerTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,144 +16,43 @@

package com.hazelcast.client.cache;

import com.hazelcast.cache.JCacheListenerTest;
import com.hazelcast.client.HazelcastClient;
import com.hazelcast.client.cache.impl.HazelcastClientCachingProvider;
import com.hazelcast.client.config.ClientConfig;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.instance.HazelcastInstanceFactory;
import com.hazelcast.test.HazelcastSerialClassRunner;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.Test;
import org.junit.After;
import org.junit.Before;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;

import javax.cache.Cache;
import javax.cache.CacheManager;
import javax.cache.configuration.CompleteConfiguration;
import javax.cache.configuration.FactoryBuilder;
import javax.cache.configuration.MutableCacheEntryListenerConfiguration;
import javax.cache.configuration.MutableConfiguration;
import javax.cache.event.CacheEntryCreatedListener;
import javax.cache.event.CacheEntryEvent;
import javax.cache.event.CacheEntryExpiredListener;
import javax.cache.event.CacheEntryListenerException;
import javax.cache.event.CacheEntryRemovedListener;
import javax.cache.event.CacheEntryUpdatedListener;
import java.io.Serializable;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicInteger;
import javax.cache.spi.CachingProvider;

@RunWith(HazelcastSerialClassRunner.class)
@Category(QuickTest.class)
public class JCacheClientListenerTest {
public class JCacheClientListenerTest extends JCacheListenerTest {

@Test
public void testSyncListenerFromClient() throws Exception {
try {
HazelcastInstance hz1 = Hazelcast.newHazelcastInstance();
HazelcastInstance hz2 = Hazelcast.newHazelcastInstance();

ClientConfig clientConfig = new ClientConfig();
clientConfig.getNetworkConfig().addAddress("127.0.0.1");

HazelcastInstance client = HazelcastClient.newHazelcastClient(clientConfig);

HazelcastClientCachingProvider cachingProvider = HazelcastClientCachingProvider.createCachingProvider(client);
CacheManager cacheManager = cachingProvider.getCacheManager();

AtomicInteger counter = new AtomicInteger();

CompleteConfiguration<String, String> config = new MutableConfiguration<String, String>()
.setTypes(String.class, String.class)
.addCacheEntryListenerConfiguration(
new MutableCacheEntryListenerConfiguration<String, String>(
FactoryBuilder.factoryOf(new TestListener(counter)), null, true, true
)
);

final Cache<String, String> cache = cacheManager.createCache("test", config);

final CountDownLatch[] latches = new CountDownLatch[10];
for (int o = 0; o < latches.length; o++) {
final int index = o;
latches[o] = new CountDownLatch(1);
new Thread() {
public void run() {
for (int i = 0; i < 100000; i++) {
String key = String.valueOf(i % 10);
String value = UUID.randomUUID().toString();
System.out.println("key=" + key + ", value=" + value);
cache.put(key, value);
}
latches[index].countDown();
}
}.start();
}

for (int o = 0; o < latches.length; o++) {
latches[o].await();
}

System.out.println("Listener invoked locally: " + counter.get());

} finally {
HazelcastClient.shutdownAll();
Hazelcast.shutdownAll();
}
@Before
@After
public void cleanup() {
HazelcastClient.shutdownAll();
HazelcastInstanceFactory.terminateAll();
}

public static class TestListener
implements CacheEntryCreatedListener<String, String>,
CacheEntryUpdatedListener<String, String>,
CacheEntryRemovedListener<String, String>,
CacheEntryExpiredListener<String, String>,
Serializable {

private final AtomicInteger counter;

public TestListener(AtomicInteger counter) {
this.counter = counter;
}

@Override
public void onCreated(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {

for (CacheEntryEvent cacheEntryEvent : cacheEntryEvents) {
System.out.println(cacheEntryEvent);
counter.incrementAndGet();
}
}

@Override
public void onExpired(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {

for (CacheEntryEvent cacheEntryEvent : cacheEntryEvents) {
System.out.println(cacheEntryEvent);
counter.incrementAndGet();
}
}

@Override
public void onRemoved(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {
@Override
protected CachingProvider getCachingProvider() {
Hazelcast.newHazelcastInstance();
Hazelcast.newHazelcastInstance();

for (CacheEntryEvent cacheEntryEvent : cacheEntryEvents) {
System.out.println(cacheEntryEvent);
counter.incrementAndGet();
}
}
ClientConfig clientConfig = new ClientConfig();
clientConfig.getNetworkConfig().addAddress("127.0.0.1");

@Override
public void onUpdated(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {
HazelcastInstance client = HazelcastClient.newHazelcastClient(clientConfig);

for (CacheEntryEvent cacheEntryEvent : cacheEntryEvents) {
System.out.println(cacheEntryEvent);
counter.incrementAndGet();
}
}
return HazelcastClientCachingProvider.createCachingProvider(client);
}
}
130 changes: 130 additions & 0 deletions hazelcast/src/test/java/com/hazelcast/cache/JCacheListenerTest.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,130 @@
/*
* Copyright (c) 2008-2014, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package com.hazelcast.cache;

import com.hazelcast.cache.impl.HazelcastServerCachingProvider;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.test.HazelcastSerialClassRunner;
import com.hazelcast.test.HazelcastTestSupport;
import com.hazelcast.test.TestHazelcastInstanceFactory;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;

import javax.cache.Cache;
import javax.cache.CacheManager;
import javax.cache.configuration.CompleteConfiguration;
import javax.cache.configuration.FactoryBuilder;
import javax.cache.configuration.MutableCacheEntryListenerConfiguration;
import javax.cache.configuration.MutableConfiguration;
import javax.cache.event.CacheEntryCreatedListener;
import javax.cache.event.CacheEntryEvent;
import javax.cache.event.CacheEntryListenerException;
import javax.cache.event.CacheEntryUpdatedListener;
import javax.cache.spi.CachingProvider;
import java.io.Serializable;
import java.util.Random;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.locks.LockSupport;

@RunWith(HazelcastSerialClassRunner.class)
@Category(QuickTest.class)
public class JCacheListenerTest extends HazelcastTestSupport {

protected CachingProvider getCachingProvider() {
TestHazelcastInstanceFactory factory = createHazelcastInstanceFactory(2);
HazelcastInstance hz1 = factory.newHazelcastInstance();
HazelcastInstance hz2 = factory.newHazelcastInstance();

return HazelcastServerCachingProvider.createCachingProvider(hz2);
}

@Test
public void testSyncListener() throws Exception {
CachingProvider cachingProvider = getCachingProvider();
CacheManager cacheManager = cachingProvider.getCacheManager();

final AtomicInteger counter = new AtomicInteger();

CompleteConfiguration<String, String> config = new MutableConfiguration<String, String>()
.setTypes(String.class, String.class).addCacheEntryListenerConfiguration(
new MutableCacheEntryListenerConfiguration<String, String>(
FactoryBuilder.factoryOf(new TestListener(counter)), null, true, true));

final Cache<String, String> cache = cacheManager.createCache("test", config);

final int threadCount = 10;
final int putCount = 1000;
final CountDownLatch latch = new CountDownLatch(threadCount);
for (int i = 0; i < threadCount; i++) {
new Thread() {
public void run() {
Random rand = new Random();
for (int i = 0; i < putCount; i++) {
String key = String.valueOf(rand.nextInt(putCount));
String value = UUID.randomUUID().toString();
cache.put(key, value);
}
latch.countDown();
}
}.start();
}

HazelcastTestSupport.assertOpenEventually(latch);
Assert.assertEquals(threadCount * putCount, counter.get());
}

public static class TestListener
implements CacheEntryCreatedListener<String, String>,
CacheEntryUpdatedListener<String, String>,
Serializable {

private final AtomicInteger counter;

public TestListener(AtomicInteger counter) {
this.counter = counter;
}

@Override
public void onCreated(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {

onEvent(cacheEntryEvents);
}

@Override
public void onUpdated(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents)
throws CacheEntryListenerException {

onEvent(cacheEntryEvents);
}

private void onEvent(Iterable<CacheEntryEvent<? extends String, ? extends String>> cacheEntryEvents) {
int count = 0;
for (CacheEntryEvent cacheEntryEvent : cacheEntryEvents) {
count++;
}
// add some random delay to simulate sync listener
LockSupport.parkNanos((long) (Math.random() * 10 * count));
counter.addAndGet(count);
}
}
}

0 comments on commit c0e3d4a

Please sign in to comment.