org.apache.axis2.transport.amqp.out.AMQPMessageSender.java Source code

Introduction

Here is the source code for org.apache.axis2.transport.amqp.out.AMQPMessageSender.java

Source

/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you under the Apache License, Version 2.0 (the
 *  "License"); you may not use this file except in compliance
 *  with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing,
 *  software distributed under the License is distributed on an
 *   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 *  KIND, either express or implied.  See the License for the
 *  specific language governing permissions and limitations
 *  under the License.
 */

package org.apache.axis2.transport.amqp.out;

import java.io.IOException;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.transport.amqp.common.AMQPConstants;
import org.apache.axis2.transport.amqp.common.AMQPMessage;
import org.apache.axis2.transport.amqp.common.Destination;
import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.client.Channel;

/**
 * Performs the actual sending of a AMQP message, and the subsequent committing of a JTA transaction
 * (if requested) or the local session transaction, if used. An instance of this class is unique
 * to a single message send out operation and will not be shared.
 */
public class AMQPMessageSender {

    private static final Log log = LogFactory.getLog(AMQPMessageSender.class);
    private Channel chan;
    private Destination destination;

    /**
     * This is a low-end method to support the one-time sends using AMQP
     * @param chan AMQP Channel
     * @param destination the AMQP Destination
     */
    public AMQPMessageSender(Channel chan, Destination destination) {
        this.chan = chan;
        this.destination = destination;
    }

    /**
     * Perform actual send of AMQP message to the Destination selected
     *
     * @param message the AMQP message
     * @param msgCtx the Axis2 MessageContext
     * @throws IOException 
     */
    public void send(AMQPMessage message, MessageContext msgCtx) throws IOException {

        Boolean persistent = getBooleanProperty(msgCtx, AMQPConstants.DELIVERY_MODE);
        Integer priority = getIntegerProperty(msgCtx, AMQPConstants.PRIORITY);
        Integer timeToLive = getIntegerProperty(msgCtx, AMQPConstants.TIME_TO_LIVE);
        BasicProperties msg_prop = null;

        msg_prop = message.getProperties();
        if (persistent != null) {
            msg_prop = msg_prop.builder().deliveryMode(2).build();

        }
        if (priority != null) {
            msg_prop = msg_prop.builder().priority(1).build();
        }
        if (timeToLive != null) {
            msg_prop = msg_prop.builder().expiration(timeToLive.toString()).build();
        }

        // perform actual message sending

        if (destination.getType() == AMQPConstants.QUEUE) {
            chan.basicPublish("", destination.getName(), message.getProperties(), message.getBody());
        } else {
            chan.basicPublish(destination.getName(), destination.getRoutingKey(), message.getProperties(),
                    message.getBody());
        }

        // set the actual MessageID to the message context for use by any others down the line
        String msgId = null;
        msgId = msg_prop.getMessageId();
        if (msgId != null) {
            msgCtx.setProperty(AMQPConstants.AMQP_MESSAGE_ID, msgId);
        }

        log.debug("Sent Message Context ID : " + msgCtx.getMessageID() + " with Message ID : " + msgId
                + " to destination : " + destination);
    }

    /**
     * Close non-shared producer, session and connection if any
     */
    public void close() {
        try {
            chan.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /*
    private void handleException(String message, Exception e) {
    log.error(message, e);
    throw new AxisAMQPException(message, e);
    }
    */

    private Boolean getBooleanProperty(MessageContext msgCtx, String name) {
        Object o = msgCtx.getProperty(name);
        if (o != null) {
            if (o instanceof Boolean) {
                return (Boolean) o;
            } else if (o instanceof String) {
                return Boolean.valueOf((String) o);
            }
        }
        return null;
    }

    private Integer getIntegerProperty(MessageContext msgCtx, String name) {
        Object o = msgCtx.getProperty(name);
        if (o != null) {
            if (o instanceof Integer) {
                return (Integer) o;
            } else if (o instanceof String) {
                return Integer.parseInt((String) o);
            }
        }
        return null;
    }

    public Channel getChannel() {
        return chan;
    }

    public void setChannel(Channel chan) {
        this.chan = chan;
    }

    public Destination getDestination() {
        return destination;
    }

    public void setDestination(Destination destination) {
        this.destination = destination;
    }

}