com.vmware.bdd.manager.RuntimeConnectionManager.java Source code

Introduction

Here is the source code for com.vmware.bdd.manager.RuntimeConnectionManager.java

Source

/***************************************************************************
 * Copyright (c) 2012-2013 VMware, Inc. All Rights Reserved.
 * Licensed 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 com.vmware.bdd.manager;

import java.io.IOException;

import org.apache.log4j.Logger;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.vmware.bdd.utils.ConfigInfo;

public class RuntimeConnectionManager {
    private static final Logger logger = Logger.getLogger(RuntimeConnectionManager.class);
    private Channel runtimeChannel;
    private Connection conn = null;
    private String exchangeName = ConfigInfo.getRuntimeMqExchangeName();

    /**
     * Setup rabbitMQ exchange and channel to notify VHMs
     * @throws IOException 
     */
    public void init() throws IOException {
        String serverHost = ConfigInfo.getMqServerHost();
        int serverPort = ConfigInfo.getMqServerPort();
        String serverUsername = ConfigInfo.getMqServerUsername();
        String serverPassword = ConfigInfo.getMqServerPassword();

        logger.info("init runtime exchange");

        ConnectionFactory factory = new ConnectionFactory();
        if (serverUsername != null && !serverUsername.equals("")) {
            factory.setUsername(serverUsername);
            factory.setPassword(serverPassword);
        }
        factory.setVirtualHost("/");
        factory.setHost(serverHost);
        factory.setPort(serverPort);
        try {
            conn = factory.newConnection();
            runtimeChannel = conn.createChannel();

            logger.info("creating exchange: " + exchangeName);
            runtimeChannel.exchangeDeclare(exchangeName, "direct", true, // durable
                    false, // auto-delete
                    null); // arguments map
        } catch (IOException e) {
            logger.error(e.getMessage());
            destroy();
            throw e;
        }
    }

    /**
     * Send a message to VHMs (on runtime rabbit exchange)
     * 
     * @throws IOException
     */
    public void sendMessage(String routeKey, String message) throws IOException {
        logger.info("sending message \"" + message + "\" on exchange " + exchangeName + ".");
        runtimeChannel.basicPublish(exchangeName, routeKey, null, message.getBytes());
    }

    public Channel getRuntimeChannel() {
        return runtimeChannel;
    }

    public void destroy() throws IOException {
        if (runtimeChannel != null) {
            runtimeChannel.close();
        }
        if (conn != null) {
            conn.close();
        }
    }

}