testsuites/integration/integration-uservice-test/src/test/java/org/onap/policy/apex/testsuites/integration/uservice/adapt/jms/JmsEventSubscriber.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2019-2020 Nordix Foundation.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.apex.testsuites.integration.uservice.adapt.jms;
  23
  24 import javax.jms.Connection;
  25 import javax.jms.ConnectionFactory;
  26 import javax.jms.JMSException;
  27 import javax.jms.Message;
  28 import javax.jms.MessageConsumer;
  29 import javax.jms.ObjectMessage;
  30 import javax.jms.Session;
  31 import javax.jms.TextMessage;
  32 import javax.jms.Topic;
  33
  34 import org.apache.activemq.command.ActiveMQTopic;
  35 import org.onap.policy.apex.core.infrastructure.threading.ThreadUtilities;
  36 import org.onap.policy.apex.service.engine.event.ApexEventException;
  37 import org.onap.policy.apex.service.engine.event.ApexEventRuntimeException;
  38 import org.onap.policy.apex.testsuites.integration.common.testclasses.PingTestClass;
  39 import org.slf4j.Logger;
  40 import org.slf4j.LoggerFactory;
  41
  42 /**
  43  * The Class JmsEventSubscriber.
  44  *
  45  * @author Liam Fallon (liam.fallon@ericsson.com)
  46  */
  47 public class JmsEventSubscriber implements Runnable {
  48     private static final Logger LOGGER = LoggerFactory.getLogger(JmsEventSubscriber.class);
  49
  50     private final String topic;
  51     private long eventsReceivedCount = 0;
  52
  53     private final Thread subscriberThread;
  54     private final Connection connection;
  55
  56     /**
  57      * Instantiates a new jms event subscriber.
  58      *
  59      * @param topic the topic
  60      * @param connectionFactory the connection factory
  61      * @param username the username
  62      * @param password the password
  63      * @throws JMSException the JMS exception
  64      */
  65     public JmsEventSubscriber(final String topic, final ConnectionFactory connectionFactory, final String username,
  66             final String password) throws JMSException {
  67         this.topic = topic;
  68         connection = connectionFactory.createConnection(username, password);
  69         connection.start();
  70
  71         subscriberThread = new Thread(this);
  72         subscriberThread.start();
  73     }
  74
  75     /**
  76      * {@inheritDoc}.
  77      */
  78     @Override
  79     public void run() {
  80         final Topic jmsTopic = new ActiveMQTopic(topic);
  81         try (final Session jmsSession = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
  82                 final MessageConsumer jmsConsumer = jmsSession.createConsumer(jmsTopic)) {
  83
  84             while (subscriberThread.isAlive() && !subscriberThread.isInterrupted()) {
  85                 try {
  86                     final Message message = jmsConsumer.receive(100);
  87                     if (message == null) {
  88                         continue;
  89                     }
  90
  91                     if (message instanceof ObjectMessage) {
  92                         final PingTestClass testPing = (PingTestClass) ((ObjectMessage) message).getObject();
  93                         testPing.verify();
  94                     } else if (message instanceof TextMessage) {
  95                         ((TextMessage) message).getText();
  96                     } else {
  97                         throw new ApexEventException("unknowm message \"" + message + "\" of type \""
  98                                 + message.getClass().getName() + "\" received");
  99                     }
 100                     eventsReceivedCount++;
 101                 } catch (final Exception e) {
 102                     if (!(e.getCause() instanceof InterruptedException)) {
 103                         throw new ApexEventRuntimeException("JMS message reception failed", e);
 104                     }
 105                 }
 106             }
 107
 108         } catch (final Exception e) {
 109             throw new ApexEventRuntimeException("JMS event consumption failed", e);
 110         }
 111
 112         LOGGER.info("{} : event reception completed, {} events received", this.getClass().getName(),
 113                 eventsReceivedCount);
 114     }
 115
 116     /**
 117      * Gets the events received count.
 118      *
 119      * @return the events received count
 120      */
 121     public long getEventsReceivedCount() {
 122         return eventsReceivedCount;
 123     }
 124
 125     /**
 126      * Shutdown.
 127      *
 128      * @throws JMSException the JMS exception
 129      */
 130     public void shutdown() throws JMSException {
 131         LOGGER.info("{} : stopping...", this.getClass().getName());
 132
 133         subscriberThread.interrupt();
 134
 135         while (subscriberThread.isAlive()) {
 136             ThreadUtilities.sleep(10);
 137         }
 138
 139         connection.close();
 140         LOGGER.info("{} : stopped", this.getClass().getName());
 141     }
 142
 143 }