[jboss-cvs] JBossAS SVN: r108061 - branches/infinispan-int/testsuite/src/main/org/jboss/test/cluster/clusteredentity.

[jboss-cvs-commits at lists.jboss.org]

Author: pferraro
Date: 2010-09-08 20:18:14 -0400 (Wed, 08 Sep 2010)
New Revision: 108061

Modified:
   branches/infinispan-int/testsuite/src/main/org/jboss/test/cluster/clusteredentity/EntityTestBean.java
Log:
Add cache listeners in getCache(...)

Modified: branches/infinispan-int/testsuite/src/main/org/jboss/test/cluster/clusteredentity/EntityTestBean.java
===================================================================
--- branches/infinispan-int/testsuite/src/main/org/jboss/test/cluster/clusteredentity/EntityTestBean.java	2010-09-08 21:37:33 UTC (rev 108060)
+++ branches/infinispan-int/testsuite/src/main/org/jboss/test/cluster/clusteredentity/EntityTestBean.java	2010-09-09 00:18:14 UTC (rev 108061)
@@ -32,11 +32,10 @@
 import javax.persistence.EntityManager;
 import javax.persistence.PersistenceContext;
 
+import org.infinispan.manager.EmbeddedCacheManager;
 import org.infinispan.notifications.Listener;
 import org.infinispan.notifications.cachelistener.annotation.CacheEntryVisited;
 import org.infinispan.notifications.cachelistener.event.CacheEntryVisitedEvent;
-import org.infinispan.notifications.cachemanagerlistener.annotation.CacheStarted;
-import org.infinispan.notifications.cachemanagerlistener.event.CacheStartedEvent;
 import org.jboss.ejb3.annotation.RemoteBinding;
 import org.jboss.ha.ispn.DefaultCacheContainerRegistry;
 import org.jboss.logging.Logger;
@@ -68,7 +67,14 @@
       if (listener == null)
       {
          listener = new MyListener();
-         DefaultCacheContainerRegistry.getInstance().getCacheContainer("entity").addListener(listener);
+         EmbeddedCacheManager container = DefaultCacheContainerRegistry.getInstance().getCacheContainer("entity");
+         for (String cacheName: container.getCacheNames())
+         {
+            if (cacheName.startsWith("persistence.unit"))
+            {
+               container.getCache(cacheName).addListener(listener);
+            }
+         }
       }
    }
    
@@ -205,12 +211,6 @@
          visited.clear();
       }
       
-      @CacheStarted
-      public void cacheStarted(CacheStartedEvent event)
-      {
-         event.getCacheManager().getCache(event.getCacheName()).addListener(this);
-      }
-      
       @CacheEntryVisited
       public void entryVisited(CacheEntryVisitedEvent event)
       {