org.wso2.carbon.sample.consumer.JMSClient.java Source code

Introduction

Here is the source code for org.wso2.carbon.sample.consumer.JMSClient.java

Source

/*
 * Copyright (c) 2016, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * WSO2 Inc. 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.wso2.carbon.sample.consumer;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.log4j.Logger;

import javax.jms.QueueConnectionFactory;
import javax.jms.TopicConnectionFactory;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import java.io.IOException;
import java.util.Properties;

public class JMSClient {

    private static Log log = LogFactory.getLog(JMSClient.class);

    public static void main(String[] args) throws InterruptedException {

        String broker = args[0];
        String topic = args[1];
        String queue = args[2];
        Properties properties = new Properties();

        try {
            boolean validBroker = true;
            if (broker.equalsIgnoreCase("qpid")) {
                properties.load(ClassLoader.getSystemClassLoader().getResourceAsStream("qpid.properties"));
            } else if (broker.equalsIgnoreCase("activemq")) {
                properties.load(ClassLoader.getSystemClassLoader().getResourceAsStream("activemq.properties"));
            } else if (broker.equalsIgnoreCase("mb")) {
                properties.load(ClassLoader.getSystemClassLoader().getResourceAsStream("mb.properties"));
            } else {
                validBroker = false;
                log.error("Entered broker is invalid! ");
            }
            if (validBroker) {
                if (topic != null && topic.isEmpty() || topic.equals("\"\"")) {
                    topic = null;
                }

                if (queue != null && queue.isEmpty() || queue.equals("\"\"")) {
                    queue = null;
                }

                if (topic == null && queue == null) {
                    log.error("Enter topic value or queue value! ");
                } else if (topic != null) {
                    Context context = new InitialContext(properties);
                    TopicConnectionFactory topicConnectionFactory = (TopicConnectionFactory) context
                            .lookup("ConnectionFactory");
                    TopicConsumer topicConsumer = new TopicConsumer(topicConnectionFactory, topic);
                    Thread consumerThread = new Thread(topicConsumer);
                    log.info("Starting" + broker + "consumerTopic thread...");
                    consumerThread.start();
                    Thread.sleep(5 * 60000);
                    log.info("Shutting down " + broker + " consumerTopic...");
                    topicConsumer.shutdown();
                } else {
                    Context context = new InitialContext(properties);
                    QueueConnectionFactory queueConnectionFactory = (QueueConnectionFactory) context
                            .lookup("ConnectionFactory");
                    QueueConsumer queueConsumer = new QueueConsumer(queueConnectionFactory, queue);
                    Thread consumerThread = new Thread(queueConsumer);
                    log.info("Starting" + broker + "consumerQueue thread...");
                    consumerThread.start();
                    Thread.sleep(5 * 60000);
                    log.info("Shutting down " + broker + " consumerQueue...");
                    queueConsumer.shutdown();
                }
            }
        } catch (IOException e) {
            log.error("Cannot read properties file from resources. " + e.getMessage(), e);
        } catch (NamingException e) {
            log.error("Invalid properties in the properties " + e.getMessage(), e);
        }

    }
}