[Jboss-cvs] JBossAS SVN: r55155 - branches/Branch_4_0/tomcat/src/main/org/jboss/web/tomcat/tc5/sso

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

Author: bstansberry at jboss.com
Date: 2006-08-03 22:48:16 -0400 (Thu, 03 Aug 2006)
New Revision: 55155

Modified:
   branches/Branch_4_0/tomcat/src/main/org/jboss/web/tomcat/tc5/sso/TreeCacheSSOClusterManager.java
Log:
Ensure data gravitation occurs if buddy repl is enabled

Modified: branches/Branch_4_0/tomcat/src/main/org/jboss/web/tomcat/tc5/sso/TreeCacheSSOClusterManager.java
===================================================================
--- branches/Branch_4_0/tomcat/src/main/org/jboss/web/tomcat/tc5/sso/TreeCacheSSOClusterManager.java	2006-08-04 01:37:13 UTC (rev 55154)
+++ branches/Branch_4_0/tomcat/src/main/org/jboss/web/tomcat/tc5/sso/TreeCacheSSOClusterManager.java	2006-08-04 02:48:16 UTC (rev 55155)
@@ -40,6 +40,7 @@
 import org.jboss.cache.Fqn;
 import org.jboss.cache.TreeCache;
 import org.jboss.cache.TreeCacheListener;
+import org.jboss.cache.config.Option;
 import org.jboss.logging.Logger;
 import org.jboss.mx.util.MBeanServerLocator;
 import org.jboss.util.NestedRuntimeException;
@@ -91,19 +92,28 @@
     * Parameter signature used for TreeCache.get calls over JMX
     */
    private static final String[] GET_SIGNATURE =
-      {Fqn.class.getName(), Object.class.getName()};
+      {Fqn.class.getName(), Object.class.getName(), Option.class.getName()};
 
    /**
     * Parameter signature used for TreeCache.put calls over JMX
     */
    private static final String[] PUT_SIGNATURE =
-      {Fqn.class.getName(), Object.class.getName(), Object.class.getName()};
+      {Fqn.class.getName(), Object.class.getName(), 
+       Object.class.getName(), Option.class.getName()};
 
    /**
     * Parameter signature used for TreeCache.remove calls over JMX
     */
-   private static final String[] REMOVE_SIGNATURE = {Fqn.class.getName()};
+   private static final String[] REMOVE_SIGNATURE = 
+         {Fqn.class.getName(), Option.class.getName()};
    
+   private static final Option GRAVITATE_OPTION = new Option();
+
+   static
+   {
+      GRAVITATE_OPTION.setForceDataGravitation(true);
+   }
+   
    // -------------------------------------------------------  Instance Fields
    
    /**
@@ -888,7 +898,7 @@
 
    private Object getFromTreeCache(Fqn fqn) throws Exception
    {
-      Object[] args = new Object[]{fqn, KEY};
+      Object[] args = new Object[]{fqn, KEY, GRAVITATE_OPTION};
       return server.invoke(getCacheObjectName(), "get", args, GET_SIGNATURE);
    }
 
@@ -1060,7 +1070,7 @@
 
    private void putInTreeCache(Fqn fqn, Object data) throws Exception
    {
-      Object[] args = new Object[]{fqn, KEY, data};
+      Object[] args = new Object[]{fqn, KEY, data, GRAVITATE_OPTION};
       server.invoke(getCacheObjectName(), "put", args, PUT_SIGNATURE);
    }
 
@@ -1159,7 +1169,7 @@
    private void removeFromTreeCache(Fqn fqn) throws Exception
    {
       server.invoke(getCacheObjectName(), "remove",
-         new Object[]{fqn},
+         new Object[]{fqn, GRAVITATE_OPTION},
          REMOVE_SIGNATURE);
    }