Examples with NotificationException org.apache.atlas.notification.NotificationException used on opensource projects

Example 1 with NotificationException

use of org.apache.atlas.notification.NotificationException in project incubator-atlas by apache.

the class KafkaNotification method sendInternalToProducer.

@VisibleForTesting
void sendInternalToProducer(Producer p, NotificationType type, String[] messages) throws NotificationException {
    String topic = TOPIC_MAP.get(type);
    List<MessageContext> messageContexts = new ArrayList<>();
    for (String message : messages) {
        ProducerRecord record = new ProducerRecord(topic, message);
        LOG.debug("Sending message for topic {}: {}", topic, message);
        Future future = p.send(record);
        messageContexts.add(new MessageContext(future, message));
    }
    List<String> failedMessages = new ArrayList<>();
    Exception lastFailureException = null;
    for (MessageContext context : messageContexts) {
        try {
            RecordMetadata response = context.getFuture().get();
            LOG.debug("Sent message for topic - {}, partition - {}, offset - {}", response.topic(), response.partition(), response.offset());
        } catch (Exception e) {
            lastFailureException = e;
            failedMessages.add(context.getMessage());
        }
    }
    if (lastFailureException != null) {
        throw new NotificationException(lastFailureException, failedMessages);
    }
}

Example 2 with NotificationException

use of org.apache.atlas.notification.NotificationException in project incubator-atlas by apache.

the class AtlasHookTest method testFailedMessageIsLoggedIfRequired.

@Test
public void testFailedMessageIsLoggedIfRequired() throws NotificationException {
    List<HookNotification.HookNotificationMessage> hookNotificationMessages = new ArrayList<HookNotification.HookNotificationMessage>() {

        {
            add(new HookNotification.EntityCreateRequest("user"));
        }
    };
    doThrow(new NotificationException(new Exception(), Arrays.asList("test message"))).when(notificationInterface).send(NotificationInterface.NotificationType.HOOK, hookNotificationMessages);
    AtlasHook.notifyEntitiesInternal(hookNotificationMessages, 2, notificationInterface, true, failedMessagesLogger);
    verify(failedMessagesLogger, times(1)).log("test message");
}

Example 3 with NotificationException

use of org.apache.atlas.notification.NotificationException in project incubator-atlas by apache.

the class AtlasHookTest method testFailedMessageIsNotLoggedIfNotRequired.

@Test
public void testFailedMessageIsNotLoggedIfNotRequired() throws NotificationException {
    List<HookNotification.HookNotificationMessage> hookNotificationMessages = new ArrayList<>();
    doThrow(new NotificationException(new Exception(), Arrays.asList("test message"))).when(notificationInterface).send(NotificationInterface.NotificationType.HOOK, hookNotificationMessages);
    AtlasHook.notifyEntitiesInternal(hookNotificationMessages, 2, notificationInterface, false, failedMessagesLogger);
    verifyZeroInteractions(failedMessagesLogger);
}

Example 4 with NotificationException

use of org.apache.atlas.notification.NotificationException in project incubator-atlas by apache.

the class AtlasHookTest method testNotifyEntitiesRetriesOnException.

@Test
public void testNotifyEntitiesRetriesOnException() throws NotificationException {
    List<HookNotification.HookNotificationMessage> hookNotificationMessages = new ArrayList<HookNotification.HookNotificationMessage>() {

        {
            add(new HookNotification.EntityCreateRequest("user"));
        }
    };
    doThrow(new NotificationException(new Exception())).when(notificationInterface).send(NotificationInterface.NotificationType.HOOK, hookNotificationMessages);
    AtlasHook.notifyEntitiesInternal(hookNotificationMessages, 2, notificationInterface, false, failedMessagesLogger);
    verify(notificationInterface, times(2)).send(NotificationInterface.NotificationType.HOOK, hookNotificationMessages);
}

Example 5 with NotificationException

use of org.apache.atlas.notification.NotificationException in project incubator-atlas by apache.

the class KafkaNotificationMockTest method shouldCollectAllFailedMessagesIfProducerFails.

@Test
@SuppressWarnings("unchecked")
public void shouldCollectAllFailedMessagesIfProducerFails() throws NotificationException, ExecutionException, InterruptedException {
    Properties configProperties = mock(Properties.class);
    KafkaNotification kafkaNotification = new KafkaNotification(configProperties);
    Producer producer = mock(Producer.class);
    String topicName = kafkaNotification.getTopicName(NotificationInterface.NotificationType.HOOK);
    String message1 = "This is a test message1";
    String message2 = "This is a test message2";
    Future returnValue1 = mock(Future.class);
    when(returnValue1.get()).thenThrow(new RuntimeException("Simulating exception"));
    Future returnValue2 = mock(Future.class);
    when(returnValue2.get()).thenThrow(new RuntimeException("Simulating exception"));
    ProducerRecord expectedRecord1 = new ProducerRecord(topicName, message1);
    when(producer.send(expectedRecord1)).thenReturn(returnValue1);
    ProducerRecord expectedRecord2 = new ProducerRecord(topicName, message2);
    when(producer.send(expectedRecord2)).thenReturn(returnValue1);
    try {
        kafkaNotification.sendInternalToProducer(producer, NotificationInterface.NotificationType.HOOK, new String[] { message1, message2 });
        fail("Should have thrown NotificationException");
    } catch (NotificationException e) {
        assertEquals(e.getFailedMessages().size(), 2);
        assertEquals(e.getFailedMessages().get(0), "This is a test message1");
        assertEquals(e.getFailedMessages().get(1), "This is a test message2");
    }
}