src/main/activemq/core/ActiveMQConnectionFactory.cpp

/*
* 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.
*/
#include "ActiveMQConnectionFactory.h"

#include <activemq/util/Guid.h>
#include <activemq/util/Properties.h>
#include <activemq/connector/ConnectorFactoryMap.h>
#include <activemq/transport/TransportBuilder.h>
#include <activemq/exceptions/NullPointerException.h>
#include <activemq/core/ActiveMQConnection.h>
#include <activemq/core/ActiveMQConstants.h>
#include <activemq/support/LibraryInit.h>

using namespace std;
using namespace activemq;
using namespace activemq::core;
using namespace activemq::util;
using namespace activemq::connector;
using namespace activemq::exceptions;
using namespace activemq::transport;

////////////////////////////////////////////////////////////////////////////////
cms::ConnectionFactory* cms::ConnectionFactory::createCMSConnectionFactory( const std::string& brokerURI )
    throw ( cms::CMSException ) {

    return new ActiveMQConnectionFactory( brokerURI );
}

////////////////////////////////////////////////////////////////////////////////
ActiveMQConnectionFactory::ActiveMQConnectionFactory() {

    brokerURL = "tcp://localhost:61616";
    this->username = "";
    this->password = "";
}

////////////////////////////////////////////////////////////////////////////////
ActiveMQConnectionFactory::ActiveMQConnectionFactory(
    const std::string& url,
    const std::string& username,
    const std::string& password ) {

    brokerURL = url;
    this->username = username;
    this->password = password;
}

////////////////////////////////////////////////////////////////////////////////
cms::Connection* ActiveMQConnectionFactory::createConnection()
    throw ( cms::CMSException ) {

    return createConnection( brokerURL, username, password, Guid::createGUIDString() );
}

////////////////////////////////////////////////////////////////////////////////
cms::Connection* ActiveMQConnectionFactory::createConnection(
    const std::string& username,
    const std::string& password )
        throw ( cms::CMSException ) {

    return createConnection( brokerURL, username, password, Guid::createGUIDString() );
}

////////////////////////////////////////////////////////////////////////////////
cms::Connection* ActiveMQConnectionFactory::createConnection(
    const std::string& username,
    const std::string& password,
    const std::string& clientId )
        throw ( cms::CMSException ) {

    return createConnection( brokerURL, username, password, clientId );
}

////////////////////////////////////////////////////////////////////////////////
cms::Connection* ActiveMQConnectionFactory::createConnection(
    const std::string& url,
    const std::string& username,
    const std::string& password,
    const std::string& clientId )
       throw ( cms::CMSException ) {

    // Declared here so that they can be deleted in the catch block
    Properties* properties = NULL;
    Transport* transport = NULL;
    Connector* connector = NULL;
    ActiveMQConnectionData* connectionData = NULL;
    ActiveMQConnection* connection = NULL;
    std::string clientIdLocal = clientId;
    TransportBuilder transportBuilder;

    try{

        properties = new Properties;

        // if no Client Id specified, create one
        if( clientIdLocal == "" ) {
            clientIdLocal = Guid::createGUIDString();
        }

        // Store login data in the properties
        properties->setProperty(
            ActiveMQConstants::toString(
                ActiveMQConstants::PARAM_USERNAME ), username );
        properties->setProperty(
            ActiveMQConstants::toString(
                ActiveMQConstants::PARAM_PASSWORD ), password );
        properties->setProperty(
            ActiveMQConstants::toString(
                ActiveMQConstants::PARAM_CLIENTID ), clientIdLocal );

        // Use the TransportBuilder to get our Transport
        transport = transportBuilder.buildTransport( url, *properties );

        if( transport == NULL ){
            throw ActiveMQException(
                __FILE__, __LINE__,
                "ActiveMQConnectionFactory::createConnection - "
                "failed creating new Transport" );
        }

        // What wire format are we using, defaults to Stomp
        std::string wireFormat =
            properties->getProperty( "wireFormat", "openwire" );

        // Now try and find a factory to create the Connector
        ConnectorFactory* connectorfactory =
            ConnectorFactoryMap::getInstance()->lookup( wireFormat );

        if( connectorfactory == NULL ) {
            throw NullPointerException(
                __FILE__, __LINE__,
                "ActiveMQConnectionFactory::createConnection - "
                "Connector for Wire Format not registered in Map" );
        }

        // Create the Connector.
        connector = connectorfactory->createConnector( *properties, transport );

        if( connector == NULL ) {
            throw NullPointerException(
                __FILE__, __LINE__,
                "ActiveMQConnectionFactory::createConnection - "
                "Failed to Create the Connector" );
        }

        // Start the Connector
        connector->start();

        // Create Holder and store the data for the Connection
        connectionData = new ActiveMQConnectionData(
            connector, transport, properties );

        // Create and Return the new connection object.
        connection = new ActiveMQConnection( connectionData );

        return connection;

    } catch( exceptions::ActiveMQException& ex ) {
        ex.setMark( __FILE__, __LINE__ );

        delete connection;
        delete connector;
        delete transport;
        delete properties;

        throw ex;

    } catch( ... ) {
        exceptions::ActiveMQException ex(
            __FILE__, __LINE__,
            "ActiveMQConnectionFactory::create - "
            "caught unknown exception" );

        delete connection;
        delete connector;
        delete transport;
        delete properties;

        throw ex;
    }
}