modules/rocketmq-ext/src/test/java/org/apache/ignite/stream/rocketmq/TestRocketMQServer.java

/*
 *  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.ignite.stream.rocketmq;

import java.util.UUID;
import org.apache.ignite.IgniteLogger;
import org.apache.rocketmq.broker.BrokerController;
import org.apache.rocketmq.common.BrokerConfig;
import org.apache.rocketmq.common.namesrv.NamesrvConfig;
import org.apache.rocketmq.namesrv.NamesrvController;
import org.apache.rocketmq.remoting.netty.NettyClientConfig;
import org.apache.rocketmq.remoting.netty.NettyServerConfig;
import org.apache.rocketmq.store.config.MessageStoreConfig;

import static java.io.File.separator;

/**
 * Test RocketMQ server handling a broker and a nameserver.
 */
class TestRocketMQServer {
    /** Nameserver port. */
    protected static final int NAME_SERVER_PORT = 9000;

    /** Broker port. */
    private static final int BROKER_PORT = 8000;

    /** Broker HA port. */
    private static final int HA_PORT = 8001;

    /** Test ip address. */
    protected static final String TEST_IP = "127.0.0.1";

    /** Test broker name. */
    private static final String TEST_BROKER = "testBroker";

    /** Test cluster name. */
    private static final String TEST_CLUSTER = "testCluster";

    /** Nameserver. */
    private static NamesrvController nameSrv;

    /** Broker. */
    private static BrokerController broker;

    /** Logger. */
    private final IgniteLogger log;

    /**
     * Test server constructor.
     *
     * @param log Logger.
     */
    TestRocketMQServer(IgniteLogger log) {
        this.log = log;

        try {
            startNameServer();
            startBroker();
        }
        catch (Exception e) {
            throw new RuntimeException("Failed to start RocketMQ: " + e);
        }
    }

    /**
     * Starts a test nameserver.
     *
     * @throws Exception If fails.
     */
    private void startNameServer() throws Exception {
        NamesrvConfig namesrvConfig = new NamesrvConfig();
        NettyServerConfig nameServerNettyServerConfig = new NettyServerConfig();

        namesrvConfig.setKvConfigPath(System.getProperty("java.io.tmpdir") + separator + "namesrv" + separator + "kvConfig.json");
        nameServerNettyServerConfig.setListenPort(NAME_SERVER_PORT);

        nameSrv = new NamesrvController(namesrvConfig, nameServerNettyServerConfig);

        nameSrv.initialize();
        nameSrv.start();

        log.info("Started nameserver at " + NAME_SERVER_PORT);
    }

    /**
     * Starts a test broker.
     *
     * @throws Exception If fails.
     */
    private void startBroker() throws Exception {
        BrokerConfig brokerCfg = new BrokerConfig();
        NettyServerConfig nettySrvCfg = new NettyServerConfig();
        MessageStoreConfig storeCfg = new MessageStoreConfig();

        brokerCfg.setBrokerName(TEST_BROKER);
        brokerCfg.setBrokerClusterName(TEST_CLUSTER);
        brokerCfg.setBrokerIP1(TEST_IP);
        brokerCfg.setNamesrvAddr(TEST_IP + ":" + NAME_SERVER_PORT);

        storeCfg.setStorePathRootDir(System.getProperty("java.io.tmpdir") + separator + "store-" + UUID.randomUUID());
        storeCfg.setStorePathCommitLog(System.getProperty("java.io.tmpdir") + separator + "commitlog");
        storeCfg.setHaListenPort(HA_PORT);

        nettySrvCfg.setListenPort(BROKER_PORT);

        broker = new BrokerController(brokerCfg, nettySrvCfg, new NettyClientConfig(), storeCfg);

        broker.initialize();
        broker.start();

        log.info("Started broker [" + TEST_BROKER + "] at " + BROKER_PORT);
    }

    /**
     * Obtains the broker address.
     *
     * @return Broker address.
     */
    String getBrokerAddr() {
        return broker.getBrokerAddr();
    }

    /**
     * Shuts test server down.
     */
    void shutdown() {
        if (broker != null)
            broker.shutdown();

        if (nameSrv != null)
            nameSrv.shutdown();
    }
}