return "ClusterChooserInterceptor";
}
public Object invoke(org.jboss.aop.joinpoint.Invocation invocation) throws Throwable
{
LoadBalancePolicy lb = (LoadBalancePolicy) invocation.getMetaData(CLUSTERED_REMOTING, LOADBALANCE_POLICY);
FamilyWrapper family = (FamilyWrapper) invocation.getMetaData(CLUSTERED_REMOTING, CLUSTER_FAMILY_WRAPPER);
// we give the opportunity, to any server interceptor, to know if this a
// first invocation to a node or if it is a failovered call
//
int failoverCounter = 0;
String familyName = family.get().getFamilyName();
invocation.getMetaData().addMetaData(CLUSTERED_REMOTING, CLUSTER_FAMILY, familyName, PayloadKey.AS_IS);
InvokerLocator target = null;
InvokerLocator txStickyTarget = null;
boolean supportsTxStickiness = family.isTxStickinessSupported();
if (supportsTxStickiness) {
Object tpc = this.getTransactionPropagationContext();
if (tpc != null)
{
txStickyTarget = txStickyTargets.get(tpc);
if (txStickyTarget != null)
{
if (family.get().getTargets().contains(txStickyTarget))
{
if (trace)
{
log.trace("Using transaction sticky target: " + txStickyTarget);
}
target = txStickyTarget;
}
else
{
throw new RuntimeException("Current transaction is stuck to " + txStickyTarget + " which is no longer available. Halting invocation.");
}
}
}
}
if (target == null)
{
if (lb instanceof AopLoadBalancePolicy)
{
target = (InvokerLocator) ((AopLoadBalancePolicy) lb).chooseTarget(family.get(), invocation);
}
else
{
target = (InvokerLocator) lb.chooseTarget(family.get());
}
}
Throwable lastException = null;
boolean failoverAuthorized = true;
while ((target != null) && failoverAuthorized)
{
invocation.getMetaData().addMetaData(CLUSTERED_REMOTING, FAILOVER_COUNTER, new Integer(failoverCounter), PayloadKey.AS_IS);
invocation.getMetaData().addMetaData(InvokeRemoteInterceptor.REMOTING, InvokeRemoteInterceptor.INVOKER_LOCATOR, target, PayloadKey.AS_IS);
invocation.getMetaData().addMetaData(CLUSTERED_REMOTING, CLUSTER_VIEW_ID, new Long(family.get().getCurrentViewId()), PayloadKey.AS_IS);
boolean definitivlyRemoveNodeOnFailure = true;
lastException = null;
try
{
Object rsp = invocation.invokeNext();
List<?> newReplicants = (List<?>) invocation.getResponseAttachment("replicants");
if (newReplicants != null)
{
long newViewId = ((Long) invocation.getResponseAttachment("viewId")).longValue();
family.get().updateClusterInfo(newReplicants, newViewId);
}
if (txStickyTarget == null)
{
if (invocationHasReachedAServer(target))
{
if (trace)
{
log.trace("Setting " + target + " as the new sticky target");
}
txStickyTarget = target;
}
}
return rsp;
}
catch (DispatcherConnectException dce)
{
//In case of graceful shutdown, the target object will no longer exist in the Dispatcher,
//fail over to another server if we can...
//For now remove the node "definitely"
lastException = dce;
}
catch (CannotConnectException ex)
{
lastException = ex;
}
catch (GenericClusteringException gce)
{
lastException = gce;
// this is a generic clustering exception that contain the
// completion status: usefull to determine if we are authorized
// to re-issue a query to another node
//
if (gce.getCompletionStatus() == GenericClusteringException.COMPLETED_NO)
{
// we don't want to remove the node from the list of failed
// node UNLESS there is a risk to indefinitively loop
//
if (family.get().getTargets().size() >= failoverCounter)
{
if (!gce.isDefinitive())
definitivlyRemoveNodeOnFailure = false;
}
}
else
{
this.invocationHasReachedAServer(target);
throw new RuntimeException("Clustering exception thrown", gce);
}
}
catch (Throwable t)
{
// Just in case this get wrapped in a Throwable. This can happen when
// the exception is generated inside the container and it is wrapped in
// a ForwardId exception.
if (t.getCause() instanceof GenericClusteringException)
{
GenericClusteringException gce = (GenericClusteringException)t.getCause();
lastException = gce;
// this is a generic clustering exception that contain the
// completion status: usefull to determine if we are authorized
// to re-issue a query to another node
//
if (gce.getCompletionStatus() == GenericClusteringException.COMPLETED_NO)
{
// we don't want to remove the node from the list of failed
// node UNLESS there is a risk to indefinitively loop
//
if (family.get().getTargets().size() >= failoverCounter)
{
if (!gce.isDefinitive())
definitivlyRemoveNodeOnFailure = false;
}
}
else
{
this.invocationHasReachedAServer(target);
throw new RuntimeException("Clustering exception thrown", gce);
}
}
else
{
throw t;
}
}
// If we reach here, this means that we must fail-over
family.get().removeDeadTarget(target);
if (!definitivlyRemoveNodeOnFailure)
{
family.get().resetView();
}
target = (InvokerLocator) lb.chooseTarget(family.get());
if (target == null)
{
if (lastException != null)
{
throw new RuntimeException("cluster invocation failed, last exception was: ", lastException);