Examples with EventSubscriber org.apache.cloudstack.framework.events.EventSubscriber used on opensource projects

Example 1 with EventSubscriber

use of org.apache.cloudstack.framework.events.EventSubscriber in project cloudstack by apache.

the class InMemoryEventBusTest method testPublishEmpty.

@Test
public void testPublishEmpty() throws Exception {
    EventSubscriber subscriber = mock(EventSubscriber.class);
    Event event = mock(Event.class);
    InMemoryEventBus bus = new InMemoryEventBus();
    bus.publish(event);
    verify(subscriber, times(0)).onEvent(event);
}

Example 2 with EventSubscriber

use of org.apache.cloudstack.framework.events.EventSubscriber in project cloudstack by apache.

the class InMemoryEventBusTest method testSubscribe.

@Test
public void testSubscribe() throws Exception {
    EventTopic topic = mock(EventTopic.class);
    EventSubscriber subscriber = mock(EventSubscriber.class);
    InMemoryEventBus bus = new InMemoryEventBus();
    UUID uuid = bus.subscribe(topic, subscriber);
    assertNotNull(uuid);
    String uuidStr = uuid.toString();
    assertTrue(UuidUtils.validateUUID(uuidStr));
    assertTrue(bus.totalSubscribers() == 1);
    bus.unsubscribe(uuid, subscriber);
    assertTrue(bus.totalSubscribers() == 0);
}

Example 3 with EventSubscriber

use of org.apache.cloudstack.framework.events.EventSubscriber in project cloudstack by apache.

the class InMemoryEventBusTest method testUnsubscribeFailWrongUUID.

@Test(expected = EventBusException.class)
public void testUnsubscribeFailWrongUUID() throws Exception {
    EventSubscriber subscriber = mock(EventSubscriber.class);
    InMemoryEventBus bus = new InMemoryEventBus();
    UUID uuid = UUID.randomUUID();
    bus.unsubscribe(uuid, subscriber);
}

Example 4 with EventSubscriber

use of org.apache.cloudstack.framework.events.EventSubscriber in project cloudstack by apache.

the class RabbitMQEventBus method stop.

@Override
public synchronized boolean stop() {
    if (s_connection.isOpen()) {
        for (String subscriberId : s_subscribers.keySet()) {
            Ternary<String, Channel, EventSubscriber> subscriberDetails = s_subscribers.get(subscriberId);
            Channel channel = subscriberDetails.second();
            String queueName = subscriberId;
            try {
                channel.queueDelete(queueName);
                channel.abort();
            } catch (IOException ioe) {
                s_logger.warn("Failed to delete queue: " + queueName + " on AMQP server due to " + ioe.getMessage());
            }
        }
    }
    closeConnection();
    return true;
}

Example 5 with EventSubscriber

use of org.apache.cloudstack.framework.events.EventSubscriber in project cloudstack by apache.

the class RabbitMQEventBus method subscribe.

/** Call to subscribe to interested set of events
     *
     * @param topic defines category and type of the events being subscribed to
     * @param subscriber subscriber that intends to receive event notification
     * @return UUID that represents the subscription with event bus
     * @throws EventBusException
     */
@Override
public UUID subscribe(EventTopic topic, EventSubscriber subscriber) throws EventBusException {
    if (subscriber == null || topic == null) {
        throw new EventBusException("Invalid EventSubscriber/EventTopic object passed.");
    }
    // create a UUID, that will be used for managing subscriptions and also used as queue name
    // for on the queue used for the subscriber on the AMQP broker
    UUID queueId = UUID.randomUUID();
    String queueName = queueId.toString();
    try {
        String bindingKey = createBindingKey(topic);
        // store the subscriber details before creating channel
        s_subscribers.put(queueName, new Ternary(bindingKey, null, subscriber));
        // create a channel dedicated for this subscription
        Connection connection = getConnection();
        Channel channel = createChannel(connection);
        // create a queue and bind it to the exchange with binding key formed from event topic
        createExchange(channel, amqpExchangeName);
        channel.queueDeclare(queueName, false, false, false, null);
        channel.queueBind(queueName, amqpExchangeName, bindingKey);
        // register a callback handler to receive the events that a subscriber subscribed to
        channel.basicConsume(queueName, s_autoAck, queueName, new DefaultConsumer(channel) {

            @Override
            public void handleDelivery(String queueName, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
                Ternary<String, Channel, EventSubscriber> queueDetails = s_subscribers.get(queueName);
                if (queueDetails != null) {
                    EventSubscriber subscriber = queueDetails.third();
                    String routingKey = envelope.getRoutingKey();
                    String eventSource = getEventSourceFromRoutingKey(routingKey);
                    String eventCategory = getEventCategoryFromRoutingKey(routingKey);
                    String eventType = getEventTypeFromRoutingKey(routingKey);
                    String resourceType = getResourceTypeFromRoutingKey(routingKey);
                    String resourceUUID = getResourceUUIDFromRoutingKey(routingKey);
                    Event event = new Event(eventSource, eventCategory, eventType, resourceType, resourceUUID);
                    event.setDescription(new String(body));
                    // deliver the event to call back object provided by subscriber
                    subscriber.onEvent(event);
                }
            }
        });
        // update the channel details for the subscription
        Ternary<String, Channel, EventSubscriber> queueDetails = s_subscribers.get(queueName);
        queueDetails.second(channel);
        s_subscribers.put(queueName, queueDetails);
    } catch (AlreadyClosedException closedException) {
        s_logger.warn("Connection to AMQP service is lost. Subscription:" + queueName + " will be active after reconnection");
    } catch (ConnectException connectException) {
        s_logger.warn("Connection to AMQP service is lost. Subscription:" + queueName + " will be active after reconnection");
    } catch (Exception e) {
        throw new EventBusException("Failed to subscribe to event due to " + e.getMessage());
    }
    return queueId;
}