be.yildizgames.module.messaging.activemq.ActivemqBroker Maven / Gradle / Ivy

Go to download
/*
 * This file is part of the Yildiz-Engine project, licenced under the MIT License  (MIT)
 *
 * Copyright (c) 2018 Grégory Van den Borre
 *
 * More infos available: https://www.yildiz-games.be
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this
 * software and associated documentation files (the "Software"), to deal in the Software
 * without restriction, including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons
 * to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or
 * substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS  OR COPYRIGHT  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE  SOFTWARE.
 */

package be.yildizgames.module.messaging.activemq;


import be.yildizgames.common.exception.technical.InitializationException;
import be.yildizgames.module.messaging.Broker;
import be.yildizgames.module.messaging.BrokerProperties;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.activemq.broker.BrokerService;

import java.io.File;
import java.nio.file.Path;

/**
 * @author Grégory Van den Borre
 */
public class ActivemqBroker extends Broker {

    private final BrokerService brokerService = new BrokerService();

    private ActivemqBroker() {
        super();
    }

    public static ActivemqBroker initialize(BrokerProperties properties) {
        return ActivemqBroker.initialize(properties.getBrokerHost(), properties.getBrokerPort());
    }

    public static ActivemqBroker initialize(String host, int port) {
        try {
            ActivemqBroker broker = new ActivemqBroker();
            String address = "failover:tcp://" + host + ":" + port;
            ActiveMQConnectionFactory connectionFactory = new ActiveMQConnectionFactory(address);
            broker.initializeConnection(connectionFactory.createConnection());
            broker.start();
            return broker;
        } catch (Exception e) {
            throw new InitializationException(e);
        }
    }

    public static ActivemqBroker initializeInternal(String name, BrokerProperties properties) {
        return ActivemqBroker.initializeInternal(name, new File(properties.getBrokerDataFolder()), properties.getBrokerHost(), properties.getBrokerPort());
    }

    public static ActivemqBroker initializeInternal(String name, Path dataDirectory, String host, int port) {
        return ActivemqBroker.initializeInternal(name, dataDirectory.toFile(), host, port);
    }

    public static ActivemqBroker initializeInternal(String name, File dataDirectory, String host, int port) {
        try {
            ActivemqBroker broker = new ActivemqBroker();
            broker.brokerService.setBrokerName(name);
            broker.brokerService.setDataDirectoryFile(dataDirectory);
            String address = "tcp://" + host + ":" + port;
            broker.brokerService.addConnector(address);
            broker.brokerService.start();
            ActiveMQConnectionFactory connectionFactory = new ActiveMQConnectionFactory(address);
            broker.initializeConnection(connectionFactory.createConnection());
            broker.start();
            return broker;
        } catch (Exception e) {
            throw new InitializationException(e);
        }
    }

    @Override
    public void close() throws Exception {
        this.closeConnection();
        this.brokerService.stop();
    }
}