/*
* 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.container;
import java.util.UUID;
import javax.jms.BytesMessage;
import javax.jms.Destination;
import javax.jms.InvalidDestinationException;
import javax.jms.MapMessage;
import javax.jms.Message;
import javax.jms.MessageFormatException;
import javax.jms.ObjectMessage;
import javax.jms.StreamMessage;
import javax.jms.TextMessage;
import org.jboss.aop.joinpoint.Invocation;
import org.jboss.aop.joinpoint.MethodInvocation;
import org.jboss.jms.client.delegate.DelegateSupport;
import org.jboss.jms.client.state.ConnectionState;
import org.jboss.jms.client.state.ProducerState;
import org.jboss.jms.client.state.SessionState;
import org.jboss.jms.delegate.SessionDelegate;
import org.jboss.jms.destination.JBossDestination;
import org.jboss.jms.destination.JBossTemporaryQueue;
import org.jboss.jms.destination.JBossTemporaryTopic;
import org.jboss.jms.message.JBossBytesMessage;
import org.jboss.jms.message.JBossMapMessage;
import org.jboss.jms.message.JBossMessage;
import org.jboss.jms.message.JBossObjectMessage;
import org.jboss.jms.message.JBossStreamMessage;
import org.jboss.jms.message.JBossTextMessage;
import org.jboss.jms.message.MessageProxy;
import org.jboss.logging.Logger;
/**
* Handles sending of messages plus handles get and set methods for Producer returning state from
* local cache.
*
* This aspect is PER_VM.
*
* Remember! PER_INSTANCE aspects are very expensive so we avoid them.
*
* @author <a href="mailto:tim.fox@jboss.com>Tim Fox</a>
* @author <a href="mailto:hgao@redhat.com>Howard Gao</a>
* @version <tt>$Revision: 6821 $</tt>
*
* $Id: ProducerAspect.java 6821 2009-05-16 13:23:49Z gaohoward $
*/
public class ProducerAspect
{
// Constants ------------------------------------------------------------------------------------
private static final Logger log = Logger.getLogger(ProducerAspect.class);
// Attributes -----------------------------------------------------------------------------------
private boolean trace = log.isTraceEnabled();
// Static ---------------------------------------------------------------------------------------
// Constructors ---------------------------------------------------------------------------------
// Public ---------------------------------------------------------------------------------------
public Object handleSend(Invocation invocation) throws Throwable
{
MethodInvocation mi = (MethodInvocation)invocation;
Object[] args = mi.getArguments();
Destination destination = (Destination)args[0];
Message m = (Message)args[1];
int deliveryMode = ((Integer)args[2]).intValue();
int priority = ((Integer)args[3]).intValue();
long timeToLive = ((Long)args[4]).longValue();
boolean keepID = args.length>5? ((Boolean)args[5]).booleanValue() : false;
// configure the message for sending, using attributes stored as metadata
ProducerState producerState = getProducerState(mi);
if (deliveryMode == -1)
{
// Use the delivery mode of the producer
deliveryMode = producerState.getDeliveryMode();
if (trace) { log.trace("Using producer's default delivery mode: " + deliveryMode); }
}
m.setJMSDeliveryMode(deliveryMode);
if (priority == -1)
{
// Use the priority of the producer
priority = producerState.getPriority();
if (trace) { log.trace("Using producer's default priority: " + priority); }
}
if (priority < 0 || priority > 9)
{
throw new MessageFormatException("Invalid message priority (" + priority + "). " +
"Valid priorities are 0-9");
}
m.setJMSPriority(priority);
if (producerState.isDisableMessageTimestamp())
{
m.setJMSTimestamp(0l);
}
else
{
m.setJMSTimestamp(System.currentTimeMillis());
}
if (timeToLive == Long.MIN_VALUE)
{
// Use time to live value from producer
timeToLive = producerState.getTimeToLive();
if (trace) { log.trace("Using producer's default timeToLive: " + timeToLive); }
}
if (timeToLive == 0)
{
// Zero implies never expires
m.setJMSExpiration(0);
}
else
{
m.setJMSExpiration(System.currentTimeMillis() + timeToLive);
}
if (destination == null)
{
// use destination from producer
destination = producerState.getDestination();
if (destination == null)
{
throw new UnsupportedOperationException("Destination not specified");
}
if (trace) { log.trace("Using producer's default destination: " + destination); }
}
else
{
// if a default destination was already specified then this must be same destination as
// that specified in the arguments
if (producerState.getDestination() != null &&
!producerState.getDestination().equals(destination))
{
throw new UnsupportedOperationException("Where a default destination is specified " +
"for the sender and a destination is " +
"specified in the arguments to the send, " +
"these destinations must be equal");
}
}
// destination should aways be a JBossDestination as this is tested on the MessageProducer creation
// and on the send, so it should be safe to just do this cast
if (((JBossDestination)destination).isTemporary())
{
validateTemporaryDestination(destination);
}
SessionState sessionState = (SessionState)producerState.getParent();
// Generate the message id
ConnectionState connectionState = (ConnectionState)sessionState.getParent();
JBossMessage messageToSend;
boolean foreign = false;
if (!(m instanceof MessageProxy))
{
// it's a foreign message
foreign = true;
// JMS 1.1 Sect. 3.11.4: A provider must be prepared to accept, from a client,
// a message whose implementation is not one of its own.
// create a matching JBossMessage Type from JMS Type
if(m instanceof BytesMessage)
{
messageToSend = new JBossBytesMessage((BytesMessage)m,0);
}
else if(m instanceof MapMessage)
{
messageToSend = new JBossMapMessage((MapMessage)m,0);
}
else if(m instanceof ObjectMessage)
{
messageToSend = new JBossObjectMessage((ObjectMessage)m,0);
}
else if(m instanceof StreamMessage)
{
messageToSend = new JBossStreamMessage((StreamMessage)m,0);
}
else if(m instanceof TextMessage)
{
messageToSend = new JBossTextMessage((TextMessage)m,0);
}
else
{
messageToSend = new JBossMessage(m, 0);
}
messageToSend.setJMSMessageID(null);
//We must set the destination *after* converting from foreign message
messageToSend.setJMSDestination(destination);
if(connectionState.getRemotingConnection().isStrictTck())
{
m.setJMSDestination(destination);
}
}
else
{
// get the actual message
MessageProxy proxy = (MessageProxy)m;
m.setJMSDestination(destination);
//Get the underlying message
messageToSend = proxy.getMessage();
proxy.beforeSend();
}
// Set the new id
if (!keepID)
{
String id = "ID:JBM-" + UUID.randomUUID().toString();
messageToSend.setJMSMessageID(id);
}
// This only really used for BytesMessages and StreamMessages to reset their state
messageToSend.doBeforeSend();
// now that we know the messageID, set it also on the foreign message, if is the case
if (foreign)
{
m.setJMSMessageID(messageToSend.getJMSMessageID());
}
// Processing the ordering group message here.
if (producerState.isOrderingGroupEnabled())
{
String grpName = producerState.getOrderingGroupName();
messageToSend.setJBMOrderingGroupName(grpName);
}
// we now invoke the send(Message) method on the session, which will eventually be fielded
// by connection endpoint
((SessionDelegate)sessionState.getDelegate()).send(messageToSend, false);
return null;
}
private void validateTemporaryDestination(Destination destination)
throws InvalidDestinationException
{
if (destination instanceof JBossTemporaryTopic)
{
if (((JBossTemporaryTopic)destination).isDeleted())
{
throw new InvalidDestinationException("Temporary topic was deleted");
}
}
else
if (destination instanceof JBossTemporaryQueue)
{
if (((JBossTemporaryQueue)destination).isDeleted())
{
throw new InvalidDestinationException("Temporary queue was deleted");
}
}
}
public Object handleSetDisableMessageID(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setDisableMessageID(((Boolean)args[0]).booleanValue());
return null;
}
public Object handleGetDisableMessageID(Invocation invocation) throws Throwable
{
return getProducerState(invocation).isDisableMessageID() ? Boolean.TRUE : Boolean.FALSE;
}
public Object handleSetDisableMessageTimestamp(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setDisableMessageTimestamp(((Boolean)args[0]).booleanValue());
return null;
}
public Object handleGetDisableMessageTimestamp(Invocation invocation) throws Throwable
{
return getProducerState(invocation).isDisableMessageTimestamp() ? Boolean.TRUE : Boolean.FALSE;
}
public Object handleSetDeliveryMode(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setDeliveryMode(((Integer)args[0]).intValue());
return null;
}
public Object handleGetDeliveryMode(Invocation invocation) throws Throwable
{
return new Integer(getProducerState(invocation).getDeliveryMode());
}
public Object handleSetPriority(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setPriority(((Integer)args[0]).intValue());
return null;
}
public Object handleGetPriority(Invocation invocation) throws Throwable
{
return new Integer(getProducerState(invocation).getPriority());
}
public Object handleSetTimeToLive(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setTimeToLive(((Long)args[0]).longValue());
return null;
}
public Object handleGetTimeToLive(Invocation invocation) throws Throwable
{
return new Long(getProducerState(invocation).getTimeToLive());
}
public Object handleGetDestination(Invocation invocation) throws Throwable
{
return getProducerState(invocation).getDestination();
}
public Object handleSetDestination(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
getProducerState(invocation).setDestination((Destination)args[0]);
return null;
}
public Object handleClosing(Invocation invocation) throws Throwable
{
return new Long(-1);
}
public Object handleClose(Invocation invocation) throws Throwable
{
return null;
}
public Object handleEnableOrderingGroup(Invocation invocation) throws Throwable
{
Object[] args = ((MethodInvocation)invocation).getArguments();
String orderingGroupName = (String)args[0];
ProducerState state = getProducerState(invocation);
state.setOrderingGroupName(orderingGroupName);
state.setOrderingGroupEnabled(true);
return null;
}
public Object handleDisableOrderingGroup(Invocation invocation) throws Throwable
{
ProducerState pState = getProducerState(invocation);
pState.disableOrderingGroup();
return null;
}
// Class YYY overrides --------------------------------------------------------------------------
// Protected ------------------------------------------------------------------------------------
// Package Private ------------------------------------------------------------------------------
// Private --------------------------------------------------------------------------------------
private ProducerState getProducerState(Invocation inv)
{
return (ProducerState)((DelegateSupport)inv.getTargetObject()).getState();
}
// Inner Classes --------------------------------------------------------------------------------
}