package org.jboss.cache.invalidation;
import java.io.Serializable;
import java.util.HashSet;
import javax.management.MBeanParameterInfo;
import javax.management.MBeanOperationInfo;
import org.jboss.cache.invalidation.InvalidationManager.BridgeInvalidationSubscriptionImpl;
public class InvalidationManager
extends org.jboss.system.ServiceMBeanSupport
implements InvalidationManagerMBean
{
public static final String DEFAULT_JMX_SERVICE_NAME = "jboss.cache:service=InvalidationManager";
public static final String DEFAULT_INVALIDERS_JMX_NAME = "jboss.cache:service=InvalidationGroup";
protected java.util.Hashtable groups = new java.util.Hashtable ();
protected java.util.Vector bridgeSubscribers = new java.util.Vector ();
protected int hashcode = 0;
protected boolean DEFAULT_TO_ASYNCHRONOUS_MODE = false;
public InvalidationManager () { super(); }
public void startService () throws Exception
{
log.debug ("Starting Invalidation Manager " + this.getServiceName ().toString ());
org.jboss.system.Registry.bind (this.getServiceName ().toString (), this);
this.hashcode = this.getServiceName ().hashCode ();
}
public void stopService ()
{
log.debug ("Stoping Invalidation Manager " + this.getServiceName ().toString ());
org.jboss.system.Registry.unbind (this.getServiceName ().toString ());
}
public boolean getIsAsynchByDefault()
{
return DEFAULT_TO_ASYNCHRONOUS_MODE;
}
public void setIsAsynchByDefault(boolean flag)
{
this.DEFAULT_TO_ASYNCHRONOUS_MODE = flag;
}
public java.util.Collection getInvalidationGroups ()
{
return this.groups.values ();
}
public InvalidationGroup getInvalidationGroup (String groupName)
{
synchronized (this.groups)
{
InvalidationGroup group = (InvalidationGroup)this.groups.get (groupName);
if (group == null)
{
group = createGroup (groupName);
}
group.addReference ();
return group;
}
}
public synchronized BridgeInvalidationSubscription registerBridgeListener (InvalidationBridgeListener listener)
{
log.debug ("Subscribing a new cache-invalidation bridge");
BridgeInvalidationSubscription subs = new BridgeInvalidationSubscriptionImpl(listener);
java.util.Vector newVector = new java.util.Vector (this.bridgeSubscribers);
newVector.add (subs);
this.bridgeSubscribers = newVector;
return subs;
}
public void batchInvalidate (BatchInvalidation[] invalidations)
{
this.batchInvalidate (invalidations, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
public void batchInvalidate (BatchInvalidation[] invalidations, boolean asynchronous)
{
if (log.isTraceEnabled ())
log.trace ("Batch cache invalidation. Caches concerned: " + invalidations.length);
this.crossDomainBatchInvalidate (null, invalidations, asynchronous);
}
public void invalidateAll(String groupName)
{
invalidateAll(groupName, DEFAULT_TO_ASYNCHRONOUS_MODE);
}
public void invalidateAll(String groupName, boolean async)
{
if (log.isTraceEnabled ())
log.trace ("Invalidate all for group: " + groupName);
crossDomainInvalidateAll(null, groupName, async);
}
public int hashCode ()
{
return hashcode;
}
protected InvalidationGroup createGroup (String groupName)
{
InvalidationGroup group = new org.jboss.cache.invalidation.InvalidationManager.InvalidationGroupImpl (groupName);
this.groups.put (groupName, group);
try
{
log.debug ("Creating and registering a new InvalidationGroup: " + groupName);
javax.management.ObjectName groupObjectName = new javax.management.ObjectName (DEFAULT_INVALIDERS_JMX_NAME + ",GroupName="+groupName);
this.getServer ().registerMBean (group, groupObjectName);
}
catch (Exception e)
{
log.debug ("Problem while trying to register a new invalidation group in JMX", e);
}
log.debug ("Informing bridges about new group creation ...");
for (int i=0; i<bridgeSubscribers.size (); i++)
((BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i))).groupCreated (groupName);
return group;
}
protected void removeGroup (String groupName)
{
synchronized (this.groups)
{
this.groups.remove (groupName);
try
{
log.debug ("Removing and JMX-unregistering an InvalidationGroup: " + groupName);
javax.management.ObjectName groupObjectName = new javax.management.ObjectName (DEFAULT_INVALIDERS_JMX_NAME + ",GroupName="+groupName);
this.getServer ().unregisterMBean (groupObjectName);
}
catch (Exception e)
{
log.debug ("Problem while trying to un-register a new invalidation group in JMX", e);
}
for (int i=0; i<bridgeSubscribers.size (); i++)
((BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i))).groupDropped (groupName);
}
}
protected synchronized void unregisterBridgeListener (BridgeInvalidationSubscription bridgeSubscriber)
{
log.debug ("Unsubscription of a cache-invalidation bridge");
java.util.Vector newVector = new java.util.Vector (this.bridgeSubscribers);
newVector.remove (bridgeSubscriber);
this.bridgeSubscribers = newVector;
}
protected void doLocalOnlyInvalidation (String groupName, Serializable key, boolean asynchronous)
{
InvalidationGroupImpl group = (InvalidationGroupImpl)this.groups.get (groupName);
if (group != null)
group.localOnlyInvalidate (key, asynchronous);
}
protected void doLocalOnlyInvalidations (String groupName, Serializable[] keys, boolean asynchronous)
{
InvalidationGroupImpl group = (InvalidationGroupImpl)this.groups.get (groupName);
if (group != null)
group.localOnlyInvalidate (keys, asynchronous);
}
protected void doLocalOnlyInvalidateAll (String groupName, boolean asynchronous)
{
InvalidationGroupImpl group = (InvalidationGroupImpl)this.groups.get (groupName);
if (group != null)
group.localOnlyInvalidateAll();
}
protected void doBridgedOnlyInvalidation (BridgeInvalidationSubscriptionImpl exceptSource, String groupName, Serializable key)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
{
BridgeInvalidationSubscriptionImpl bridge = (BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i));
if (bridge != exceptSource)
bridge.bridgedInvalidate (groupName, key, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
}
protected void doBridgedOnlyInvalidation (BridgeInvalidationSubscriptionImpl exceptSource, String groupName, Serializable[] keys)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
{
BridgeInvalidationSubscriptionImpl bridge = (BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i));
if (bridge != exceptSource)
bridge.bridgedInvalidate (groupName, keys, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
}
protected void doBridgedOnlyInvalidateAll (BridgeInvalidationSubscriptionImpl exceptSource, String groupName)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
{
BridgeInvalidationSubscriptionImpl bridge = (BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i));
if (bridge != exceptSource)
bridge.bridgedInvalidateAll (groupName, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
}
protected void localGroupInvalidationEvent (String groupName, Serializable key, boolean asynchronous)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
((BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i))).bridgedInvalidate (groupName, key, asynchronous);
}
protected void localGroupInvalidationsEvent (String groupName, Serializable[] keys, boolean asynchronous)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
((BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i))).bridgedInvalidate (groupName, keys, asynchronous);
}
protected void localGroupInvalidateAllEvent (String groupName, boolean asynchronous)
{
for (int i=0; i<bridgeSubscribers.size (); i++)
((BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i))).bridgedInvalidateAll (groupName, asynchronous);
}
protected void bridgeGroupInvalidationEvent (BridgeInvalidationSubscriptionImpl source, String groupName, Serializable key)
{
doBridgedOnlyInvalidation (source, groupName, key);
doLocalOnlyInvalidation (groupName, key, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
protected void bridgeGroupInvalidationEvent (BridgeInvalidationSubscriptionImpl source, String groupName, Serializable[] keys)
{
doBridgedOnlyInvalidation (source, groupName, keys);
doLocalOnlyInvalidation (groupName, keys, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
protected void bridgeGroupInvalidateAllEvent (BridgeInvalidationSubscriptionImpl source, String groupName)
{
doBridgedOnlyInvalidateAll (source, groupName);
doLocalOnlyInvalidateAll (groupName, this.DEFAULT_TO_ASYNCHRONOUS_MODE);
}
protected void crossDomainBatchInvalidate (BridgeInvalidationSubscriptionImpl source, BatchInvalidation[] invalidations, boolean asynchronous)
{
if (invalidations == null)
return;
for (int i=0; i<invalidations.length; i++)
{
BatchInvalidation currInvalid = invalidations[i];
doLocalOnlyInvalidations (currInvalid.getInvalidationGroupName (),
currInvalid.getIds (),
asynchronous);
}
for (int i=0; i<bridgeSubscribers.size (); i++)
{
BridgeInvalidationSubscriptionImpl bridge = (BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i));
if (bridge != source)
bridge.bridgedBatchInvalidations (invalidations, asynchronous);
}
}
protected void crossDomainInvalidateAll(BridgeInvalidationSubscriptionImpl source, String groupName, boolean asynchronous)
{
doLocalOnlyInvalidateAll(groupName, asynchronous);
for (int i=0; i<bridgeSubscribers.size (); i++)
{
BridgeInvalidationSubscriptionImpl bridge = (BridgeInvalidationSubscriptionImpl)(bridgeSubscribers.elementAt (i));
if (bridge != source)
bridge.bridgedInvalidateAll(groupName, asynchronous);
}
}
class InvalidationGroupImpl implements InvalidationGroup, javax.management.DynamicMBean
{
protected org.jboss.logging.Logger igLog = null;
protected String groupName = null;
protected boolean asynchronous = DEFAULT_TO_ASYNCHRONOUS_MODE;
protected HashSet registered = new HashSet();
protected int counter = 0;
public int hashCode ()
{
return groupName.hashCode ();
}
public String getGroupName ()
{
return this.groupName;
}
public InvalidationGroupImpl (String groupName)
{
this.groupName = groupName;
this.igLog = org.jboss.logging.Logger.getLogger(getClass() + "." + groupName);
}
public InvalidationManagerMBean getInvalidationManager ()
{
return InvalidationManager.this;
}
public void invalidate (Serializable key)
{
this.invalidate (key, this.asynchronous);
}
public void invalidate (Serializable key, boolean asynchronous)
{
localOnlyInvalidate (key, asynchronous);
localGroupInvalidationEvent (this.groupName, key, asynchronous);
}
public void invalidate (Serializable[] keys)
{
this.invalidate (keys, this.asynchronous);
}
public void invalidate (Serializable[] keys, boolean asynchronous)
{
localOnlyInvalidate (keys, asynchronous);
localGroupInvalidationsEvent (this.groupName, keys, asynchronous);
}
public void invalidateAll()
{
invalidateAll(asynchronous);
}
public void invalidateAll(boolean asynchronous)
{
localOnlyInvalidateAll();
localGroupInvalidateAllEvent(groupName, asynchronous);
}
public synchronized void register (Invalidatable newRegistered)
{
HashSet newlyRegistered = new HashSet (this.registered);
newlyRegistered.add (newRegistered);
this.registered = newlyRegistered;
}
public synchronized void unregister (Invalidatable oldRegistered)
{
HashSet newlyRegistered = new HashSet(this.registered);
newlyRegistered.remove (oldRegistered);
this.registered = newlyRegistered;
this.removeReference ();
}
public void setAsynchronousInvalidation (boolean async)
{
this.asynchronous = async;
}
public boolean getAsynchronousInvalidation ()
{
return this.asynchronous;
}
public void addReference ()
{
counter++;
igLog.debug ("Counter reference value (++): " + counter);
}
public int getReferenceCount ()
{
return this.counter;
}
public void removeReference ()
{
counter--;
igLog.debug ("Counter reference value (--): " + counter);
if (counter<=0)
{
removeGroup (this.groupName);
}
}
public Object getAttribute (String attribute) throws javax.management.AttributeNotFoundException, javax.management.MBeanException, javax.management.ReflectionException
{
if (attribute == null || attribute.equals (""))
throw new IllegalArgumentException ("null or empty attribute name");
if (attribute.equals ("AsynchronousInvalidation"))
return new Boolean(this.asynchronous);
else
throw new javax.management.AttributeNotFoundException(attribute + " is not a known attribute");
}
public javax.management.AttributeList getAttributes (java.lang.String[] attributes)
{
return null;
}
public javax.management.MBeanInfo getMBeanInfo ()
{
MBeanParameterInfo serSimpleParam = new MBeanParameterInfo (
"key",
Serializable.class.getName (),
"Primary key to be invalidated"
);
MBeanParameterInfo serArrayParam = new MBeanParameterInfo (
"keys",
Serializable[].class .getName (),
"Primary keys to be invalidated"
);
MBeanParameterInfo asynchParam = new MBeanParameterInfo (
"asynchronous",
Boolean.class.getName (),
"Indicates if the invalidation should be asynchronous or must be synchronous"
);
javax.management.MBeanAttributeInfo[] attrInfo = new javax.management.MBeanAttributeInfo[] {
new javax.management.MBeanAttributeInfo("AsynchronousInvalidation",
Boolean.class.getName(),
"Indicates if invalidation, by default, should be done asynchronously",
true,
true,
false)};
MBeanOperationInfo[] opInfo = {
new MBeanOperationInfo("invalidate",
"invalidate a single key using default (a)synchronous behaviour",
new MBeanParameterInfo[] {serSimpleParam},
void.class.getName(),
MBeanOperationInfo.ACTION),
new MBeanOperationInfo("invalidate",
"invalidate a single key indicating the (a)synchronous behaviour",
new MBeanParameterInfo[] {serSimpleParam, asynchParam},
void.class.getName(),
MBeanOperationInfo.ACTION),
new MBeanOperationInfo("invalidate",
"invalidate multiple keys using default (a)synchronous behaviour",
new MBeanParameterInfo[] {serArrayParam},
void.class.getName(),
MBeanOperationInfo.ACTION),
new MBeanOperationInfo("invalidate",
"invalidate multiple keys indicating the (a)synchronous behaviour",
new MBeanParameterInfo[] {serArrayParam, asynchParam},
void.class.getName(),
MBeanOperationInfo.ACTION),
new MBeanOperationInfo("invalidateAll",
"invalidate all keys using default (a)synchronous behaviour",
new MBeanParameterInfo[] {},
void.class.getName(),
MBeanOperationInfo.ACTION),
new MBeanOperationInfo("invalidateAll",
"invalidate all keys with specified (a)synchronous behaviour",
new MBeanParameterInfo[] {asynchParam},
void.class.getName(),
MBeanOperationInfo.ACTION)
};
javax.management.MBeanNotificationInfo[] notifyInfo = null;
javax.management.MBeanConstructorInfo[] ctorInfo = new javax.management.MBeanConstructorInfo[] {};
return new javax.management.MBeanInfo(getClass().getName(),
"Cache invalidation for group named " + this.groupName,
attrInfo,
ctorInfo,
opInfo,
notifyInfo);
}
public java.lang.Object invoke (java.lang.String actionName, java.lang.Object[] params, java.lang.String[] signature) throws javax.management.MBeanException, javax.management.ReflectionException
{
if ("invalidate".equals (actionName))
{
if (params.length == 1)
{
if (params[0] instanceof Serializable[])
this.invalidate ((Serializable[])params[0]);
else if (params[0] instanceof Serializable)
this.invalidate ((Serializable)params[0]);
else
throw new IllegalArgumentException ("First argument must be Serializable (or array of)");
}
else if (params.length == 2)
{
if (params[0] instanceof Serializable[])
this.invalidate ((Serializable[])params[0], ((Boolean)params[1]).booleanValue ());
else if (params[0] instanceof Serializable)
this.invalidate ((Serializable)params[0], ((Boolean)params[1]).booleanValue ());
else
throw new IllegalArgumentException ("First argument must be Serializable (or array of)");
}
else
{
throw new IllegalArgumentException ("Unknown operation with these parameters: " + actionName);
}
}
else if("invalidateAll".equals(actionName))
{
if(params == null || params.length == 0)
{
this.invalidateAll();
}
else if (params.length == 1)
{
this.invalidateAll (((Boolean)params[1]).booleanValue ());
}
else
{
throw new IllegalArgumentException ("invalidateAll can take zero or one parameter but got " + params.length);
}
}
else
{
throw new IllegalArgumentException ("Unknown operation: " + actionName);
}
return null;
}
public void setAttribute (javax.management.Attribute attribute) throws javax.management.AttributeNotFoundException, javax.management.InvalidAttributeValueException, javax.management.MBeanException, javax.management.ReflectionException
{
String attrName = attribute.getName();
if (attrName == null || attrName.equals (""))
throw new IllegalArgumentException ("null or empty attribute name");
if (attrName.equals ("AsynchronousInvalidation"))
{
Object value = attribute.getValue ();
if (value instanceof Boolean)
this.asynchronous = ((Boolean)value).booleanValue ();
else
throw new javax.management.InvalidAttributeValueException("Attribute is of boolean type");
}
else
throw new javax.management.AttributeNotFoundException(attrName + " is not a known attribute");
}
public javax.management.AttributeList setAttributes (javax.management.AttributeList attributes)
{
return null;
}
protected void localOnlyInvalidate (Serializable[] keys, boolean asynchronous)
{
java.util.Iterator iter = this.registered.iterator ();
while (iter.hasNext ())
{
Invalidatable inv = (Invalidatable)iter.next ();
inv.areInvalid (keys);
}
}
protected void localOnlyInvalidate (Serializable key, boolean asynchronous)
{
java.util.Iterator iter = this.registered.iterator ();
while (iter.hasNext ())
{
Invalidatable inv = (Invalidatable)iter.next ();
inv.isInvalid (key);
}
}
protected void localOnlyInvalidateAll()
{
java.util.Iterator iter = this.registered.iterator ();
while (iter.hasNext ())
{
Invalidatable inv = (Invalidatable)iter.next ();
inv.invalidateAll();
}
}
}
class BridgeInvalidationSubscriptionImpl
implements BridgeInvalidationSubscription
{
protected InvalidationBridgeListener listener = null;
public BridgeInvalidationSubscriptionImpl (InvalidationBridgeListener listener)
{
this.listener = listener;
}
public void invalidate (String invalidationGroupName, Serializable key)
{
bridgeGroupInvalidationEvent (this, invalidationGroupName, key);
}
public void invalidate (String invalidationGroupName, Serializable[] keys)
{
bridgeGroupInvalidationEvent (this, invalidationGroupName, keys);
}
public void invalidateAll(String groupName)
{
bridgeGroupInvalidateAllEvent(this, groupName);
}
public void batchInvalidate (BatchInvalidation[] invalidations)
{
crossDomainBatchInvalidate (this, invalidations, DEFAULT_TO_ASYNCHRONOUS_MODE);
}
public void unregister ()
{
unregisterBridgeListener (this);
}
protected void bridgedInvalidate (String invalidationGroupName, Serializable key, boolean asynchronous)
{
this.listener.invalidate (invalidationGroupName, key, asynchronous);
}
protected void bridgedInvalidate (String invalidationGroupName, Serializable[] keys, boolean asynchronous)
{
this.listener.invalidate (invalidationGroupName, keys, asynchronous);
}
protected void bridgedInvalidateAll (String invalidationGroupName, boolean asynchronous)
{
this.listener.invalidateAll (invalidationGroupName, asynchronous);
}
protected void bridgedBatchInvalidations (BatchInvalidation[] invalidations, boolean asynchronous)
{
this.listener.batchInvalidate (invalidations, asynchronous);
}
protected void groupCreated (String invalidationGroupName)
{
this.listener.newGroupCreated (invalidationGroupName);
}
protected void groupDropped (String invalidationGroupName)
{
this.listener.groupIsDropped (invalidationGroupName);
}
}
}