Package com.omniti.reconnoiter.broker

Source Code of com.omniti.reconnoiter.broker.RabbitMQ

/*
* Copyright (c) 2013, Circonus, Inc. All rights reserved.
* Copyright (c) 2010, OmniTI Computer Consulting, Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
*     * Redistributions of source code must retain the above copyright
*       notice, this list of conditions and the following disclaimer.
*     * Redistributions in binary form must reproduce the above
*       copyright notice, this list of conditions and the following
*       disclaimer in the documentation and/or other materials provided
*       with the distribution.
*     * Neither the name OmniTI Computer Consulting, Inc. nor the names
*       of its contributors may be used to endorse or promote products
*       derived from this software without specific prior written
*       permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

package com.omniti.reconnoiter.broker;

import java.io.IOException;

import org.apache.log4j.Logger;
import com.omniti.reconnoiter.IEventHandler;
import com.omniti.reconnoiter.StratconConfig;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.MessageProperties;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.QueueingConsumer;


public class RabbitMQ implements IMQMQ  {
  static Logger logger = Logger.getLogger(RabbitMQ.class.getName());
  private int cidx;
  private Connection conn;
  private Channel channel;
  private boolean noAck = true;
  private String userName;
  private String password;
  private String virtualHost;
  private String hostName[];
  private ConnectionFactory factory[];
  private int portNumber;
  private String exchangeName;
  private String exchangeType;
  private String declaredQueueName;
  private String queueName;
  private String returnedQueueName;
  private String routingKey;
  private String alertRoutingKey;
  private String alertExchangeName;
  private Integer heartBeat;
  private Integer connectTimeout;
  private boolean exclusiveQueue;
  private boolean durableQueue;
  private boolean durableExchange;

  public RabbitMQ(StratconConfig config) {
    this.conn = null;
    this.cidx = 0;
    this.userName = config.getMQTypeParameter("rabbitmq", "username", "guest");
    this.password = config.getMQTypeParameter("rabbitmq", "password", "guest");
    this.virtualHost = config.getMQTypeParameter("rabbitmq", "virtualhost", "/");
    this.hostName = config.getMQTypeParameter("rabbitmq", "hostname", "127.0.0.1").split(",");
    this.portNumber = Integer.parseInt(config.getMQTypeParameter("rabbitmq", "port", "5672"));
    this.heartBeat = Integer.parseInt(config.getMQTypeParameter("rabbitmq", "heartbeat", "5000"));
    this.heartBeat = (this.heartBeat + 999) / 1000; // (ms -> seconds, rounding up)
    this.connectTimeout = Integer.parseInt(config.getMQTypeParameter("rabbitmq", "connect_timeout", "5000"));

    this.exchangeType = config.getMQTypeParameter("rabbitmq", "exchangetype", "fanout");
    this.durableExchange = config.getMQTypeParameter("rabbitmq", "durableexchange", "false").equals("true");
    this.exchangeName = config.getMQTypeParameter("rabbitmq", "exchange", "noit.firehose");
    this.exclusiveQueue = config.getMQTypeParameter("rabbitmq", "exclusivequeue", "false").equals("true");
    this.durableQueue = config.getMQTypeParameter("rabbitmq", "durablequeue", "false").equals("true");
    this.declaredQueueName = config.getMQTypeParameter("rabbitmq", "queue", "reconnoiter-{node}-{pid}");
    this.routingKey = config.getMQTypeParameter("rabbitmq", "routingkey", "");
 
    try {
      this.queueName = this.declaredQueueName;
      this.queueName = this.queueName.replace("{node}",
        java.net.InetAddress.getLocalHost().getHostName());
      this.queueName = this.queueName.replace("{pid}",
        java.lang.management.ManagementFactory.getRuntimeMXBean()
                                              .getName().replaceAll("@.+$", ""));
    } catch(java.net.UnknownHostException uhe) {
      System.err.println("Cannot self identify for queuename!");
      System.exit(-1);
    }    

    this.factory = new ConnectionFactory[this.hostName.length];
    for(int i = 0; i<hostName.length; i++) {
      this.factory[i] = new ConnectionFactory();
      this.factory[i].setUsername(this.userName);
      this.factory[i].setPassword(this.password);
      this.factory[i].setVirtualHost(this.virtualHost);
      this.factory[i].setRequestedHeartbeat(this.heartBeat);
      this.factory[i].setConnectionTimeout(this.connectTimeout);
      this.factory[i].setPort(this.portNumber);
      this.factory[i].setHost(this.hostName[i]);
    }
  }
 
  //
  public void disconnect() {
    logger.info("AMQP disconnect.");
    try {
      channel.abort();
    }
    catch (Exception e) { }
    channel = null;
    try {
      conn.abort();
    }
    catch (Exception e) { }
    conn = null;
  }
  public void connect() throws Exception {
    if(conn != null) disconnect();

    int offset = ++cidx % factory.length;
    logger.info("AMQP connect to " + this.hostName[offset]);
    conn = factory[offset].newConnection();

    if(conn == null) throw new Exception("connection failed");

    channel = conn.createChannel();

    boolean durable = durableExchange, exclusive = false,
            internal = false, autoDelete = false;
    channel.exchangeDeclare(exchangeName, exchangeType,
                            durable, autoDelete, internal, null);

    autoDelete = true;
    returnedQueueName = channel.queueDeclare(queueName, durableQueue,
                                             exclusiveQueue, autoDelete, null).getQueue();
    for (String rk : routingKey.split(",")) {
        if ( rk.equalsIgnoreCase("null") ) rk = "";
        channel.queueBind(returnedQueueName, exchangeName, rk);
    }
  }
  public Channel getChannel() { return channel; }
 
  public void consume(IEventHandler eh) throws IOException {
    QueueingConsumer consumer = new QueueingConsumer(channel);

    channel.basicConsume(returnedQueueName, noAck, consumer);
   
    while (true) {
      QueueingConsumer.Delivery delivery;
      try {
        delivery = consumer.nextDelivery();
      } catch (InterruptedException ie) {
        continue;
      }
      // (process the message components ...)
   
      String xml = new String(delivery.getBody());
      try {
        eh.processMessage(xml);
      } catch (Exception e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
      }
      if(!noAck)
        channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
    }
  }
}
TOP

Related Classes of com.omniti.reconnoiter.broker.RabbitMQ

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.