Examples with NotificationFailedException org.apache.nifi.bootstrap.notification.NotificationFailedException used on opensource projects

Example 1 with NotificationFailedException

use of org.apache.nifi.bootstrap.notification.NotificationFailedException in project nifi by apache.

the class EmailNotificationService method notify.

@Override
public void notify(final NotificationContext context, final NotificationType notificationType, final String subject, final String messageText) throws NotificationFailedException {
    final Properties properties = getMailProperties(context);
    final Session mailSession = createMailSession(properties);
    final Message message = new MimeMessage(mailSession);
    try {
        message.setFrom(InternetAddress.parse(context.getProperty(FROM).evaluateAttributeExpressions().getValue())[0]);
        final InternetAddress[] toAddresses = toInetAddresses(context.getProperty(TO).evaluateAttributeExpressions().getValue());
        message.setRecipients(RecipientType.TO, toAddresses);
        final InternetAddress[] ccAddresses = toInetAddresses(context.getProperty(CC).evaluateAttributeExpressions().getValue());
        message.setRecipients(RecipientType.CC, ccAddresses);
        final InternetAddress[] bccAddresses = toInetAddresses(context.getProperty(BCC).evaluateAttributeExpressions().getValue());
        message.setRecipients(RecipientType.BCC, bccAddresses);
        message.setHeader("X-Mailer", context.getProperty(HEADER_XMAILER).evaluateAttributeExpressions().getValue());
        message.setSubject(subject);
        final String contentType = context.getProperty(CONTENT_TYPE).evaluateAttributeExpressions().getValue();
        message.setContent(messageText, contentType);
        message.setSentDate(new Date());
        Transport.send(message);
    } catch (final ProcessException | MessagingException e) {
        throw new NotificationFailedException("Failed to send E-mail Notification", e);
    }
}

Example 2 with NotificationFailedException

use of org.apache.nifi.bootstrap.notification.NotificationFailedException in project nifi by apache.

the class HttpNotificationService method notify.

@Override
public void notify(NotificationContext context, NotificationType notificationType, String subject, String message) throws NotificationFailedException {
    try {
        final RequestBody requestBody = RequestBody.create(MediaType.parse("text/plain"), message);
        Request.Builder requestBuilder = new Request.Builder().post(requestBody).url(urlReference.get());
        Map<PropertyDescriptor, String> configuredProperties = context.getProperties();
        for (PropertyDescriptor propertyDescriptor : configuredProperties.keySet()) {
            if (propertyDescriptor.isDynamic()) {
                String propertyValue = context.getProperty(propertyDescriptor).evaluateAttributeExpressions().getValue();
                requestBuilder = requestBuilder.addHeader(propertyDescriptor.getDisplayName(), propertyValue);
            }
        }
        final Request request = requestBuilder.addHeader(NOTIFICATION_SUBJECT_KEY, subject).addHeader(NOTIFICATION_TYPE_KEY, notificationType.name()).build();
        final OkHttpClient httpClient = httpClientReference.get();
        final Call call = httpClient.newCall(request);
        try (final Response response = call.execute()) {
            if (!response.isSuccessful()) {
                throw new NotificationFailedException("Failed to send Http Notification. Received an unsuccessful status code response '" + response.code() + "'. The message was '" + response.message() + "'");
            }
        }
    } catch (NotificationFailedException e) {
        throw e;
    } catch (Exception e) {
        throw new NotificationFailedException("Failed to send Http Notification", e);
    }
}