/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.jms.client.remoting;
import java.util.Map;
import org.jboss.jms.client.container.ClientConsumer;
import org.jboss.jms.client.delegate.ClientConnectionDelegate;
import org.jboss.jms.message.JBossMessage;
import org.jboss.jms.message.MessageProxy;
import org.jboss.jms.wireformat.ClientDelivery;
import org.jboss.jms.wireformat.ConnectionFactoryUpdate;
import org.jboss.logging.Logger;
import org.jboss.messaging.core.contract.Message;
import org.jboss.remoting.callback.Callback;
import org.jboss.remoting.callback.HandleCallbackException;
import org.jboss.remoting.callback.InvokerCallbackHandler;
import EDU.oswego.cs.dl.util.concurrent.ConcurrentReaderHashMap;
/**
* The CallbackManager is an InvocationHandler used for handling callbacks to message consumers.
* The callback is received and dispatched off to the relevant consumer.
*
* There is one instance of this class per remoting connection - which is to a unique server -
* therefore there is no need to add the server id to the key when doing look ups.
*
* @author <a href="tim.fox@jboss.com">Tim Fox</a>
* @author <a href="ovidiu@feodorov.com">Ovidiu Feodorov</a>
* @version <tt>$Revision: 8333 $</tt>
*
* $Id: CallbackManager.java 8333 2011-06-14 02:20:56Z gaohoward $
*/
public class CallbackManager implements InvokerCallbackHandler
{
// Constants ------------------------------------------------------------------------------------
private static final Logger log = Logger.getLogger(CallbackManager.class);
public static final String JMS_CALLBACK_SUBSYSTEM = "CALLBACK";
// Static ---------------------------------------------------------------------------------------
private static boolean trace = log.isTraceEnabled();
// Attributes -----------------------------------------------------------------------------------
protected Map<String, ClientConsumer> callbackHandlers;
protected ConnectionFactoryCallbackHandler connectionfactoryCallbackHandler;
// Constructors ---------------------------------------------------------------------------------
public CallbackManager()
{
callbackHandlers = new ConcurrentReaderHashMap();
}
// InvokerCallbackHandler implementation --------------------------------------------------------
public void handleCallback(Callback callback) throws HandleCallbackException
{
Object parameter = callback.getParameter();
if (parameter instanceof ClientDelivery)
{
ClientDelivery dr = (ClientDelivery)parameter;
ClientConsumer handler =
(ClientConsumer)callbackHandlers.get(dr.getConsumerId());
if (handler == null)
{
// This should never happen since we wait for all deliveries to arrive before closing
// the consumer
throw new IllegalStateException(this + " callback handler not found, message arrived after consumer is closed.");
}
try
{
handler.handleMessage(dr, this);
}
catch (Exception e)
{
log.error("Failed to handle message", e);
throw new HandleCallbackException(e.getMessage(), e);
}
}
else if (parameter instanceof ConnectionFactoryUpdate)
{
if (connectionfactoryCallbackHandler == null)
{
log.warn("ConnectionFactoryUpdate was received but there is no callbackHandler set");
}
else
{
ConnectionFactoryUpdate viewChange = (ConnectionFactoryUpdate)parameter;
if (trace) { log.trace(this + " receiving cluster view change " + viewChange); }
connectionfactoryCallbackHandler.handleMessage(viewChange);
}
}
else
{
throw new HandleCallbackException("Unknow callback type: " + callback);
}
}
// Public ---------------------------------------------------------------------------------------
public void registerHandler(String consumerID, ClientConsumer handler)
{
callbackHandlers.put(consumerID, handler);
}
public ClientConsumer unregisterHandler(String consumerID)
{
return (ClientConsumer)callbackHandlers.remove(consumerID);
}
public ConnectionFactoryCallbackHandler getConnectionfactoryCallbackHandler()
{
return connectionfactoryCallbackHandler;
}
public void setConnectionfactoryCallbackHandler(ConnectionFactoryCallbackHandler connectionfactoryCallbackHandler)
{
this.connectionfactoryCallbackHandler = connectionfactoryCallbackHandler;
}
public String toString()
{
return "CallbackManager[" + Integer.toHexString(hashCode()) + "]";
}
// Package protected ----------------------------------------------------------------------------
// Protected ------------------------------------------------------------------------------------
// Private --------------------------------------------------------------------------------------
// Inner classes --------------------------------------------------------------------------------
}