forked from opensearch-project/reporting
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Apply POC email notification * Roll back changes to IDEs config files * Roll back changes to IDEs config files * Refactor --------- Co-authored-by: Alex Ruiz <[email protected]>
- Loading branch information
1 parent
0410853
commit b165294
Showing
8 changed files
with
247 additions
and
10 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -159,6 +159,7 @@ $RECYCLE.BIN/ | |
| .idea/modules.xml | ||
| .idea/*.iml | ||
| .idea/modules | ||
| .idea/*.xml | ||
| *.iml | ||
| *.ipr | ||
|
|
||
|
|
||
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
121 changes: 121 additions & 0 deletions
121
src/main/kotlin/org/opensearch/reportsscheduler/util/NotificationApiUtils.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,121 @@ | ||
| /* | ||
| * Copyright OpenSearch Contributors | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
|
|
||
| package org.opensearch.reportsscheduler.util | ||
|
|
||
| import org.apache.logging.log4j.LogManager | ||
| import org.opensearch.OpenSearchSecurityException | ||
| import org.opensearch.OpenSearchStatusException | ||
| import org.opensearch.client.Client | ||
| import org.opensearch.client.node.NodeClient | ||
| import org.opensearch.commons.notifications.NotificationsPluginInterface | ||
| import org.opensearch.commons.notifications.action.GetNotificationConfigRequest | ||
| import org.opensearch.commons.notifications.action.GetNotificationConfigResponse | ||
| import org.opensearch.commons.notifications.action.SendNotificationResponse | ||
| import org.opensearch.commons.notifications.model.ChannelMessage | ||
| import org.opensearch.commons.notifications.model.EventSource | ||
| import org.opensearch.commons.notifications.model.NotificationConfigInfo | ||
| import org.opensearch.commons.notifications.model.SeverityType | ||
| import org.opensearch.core.action.ActionListener | ||
| import org.opensearch.core.rest.RestStatus | ||
| import kotlin.coroutines.resume | ||
| import kotlin.coroutines.resumeWithException | ||
| import kotlin.coroutines.suspendCoroutine | ||
|
|
||
| object NotificationApiUtils { | ||
|
|
||
| private val logger = LogManager.getLogger(NotificationApiUtils::class) | ||
|
|
||
| /** | ||
| * Gets a NotificationConfigInfo object by ID if it exists. | ||
| */ | ||
| suspend fun getNotificationConfigInfo(client: NodeClient, id: String): NotificationConfigInfo? { | ||
| return try { | ||
| val res: GetNotificationConfigResponse = | ||
| getNotificationConfig(client, GetNotificationConfigRequest(setOf(id))) | ||
| res.searchResult.objectList.firstOrNull() | ||
| } catch (e: OpenSearchSecurityException) { | ||
| throw e | ||
| } catch (e: OpenSearchStatusException) { | ||
| if (e.status() == RestStatus.NOT_FOUND) { | ||
| logger.debug("Notification config [$id] was not found") | ||
| } | ||
| null | ||
| } | ||
| } | ||
|
|
||
| private suspend fun getNotificationConfig( | ||
| client: NodeClient, | ||
| getNotificationConfigRequest: GetNotificationConfigRequest | ||
| ): GetNotificationConfigResponse { | ||
| val getNotificationConfigResponse: GetNotificationConfigResponse = | ||
| NotificationsPluginInterface.suspendUntil { | ||
| this.getNotificationConfig( | ||
| client, | ||
| getNotificationConfigRequest, | ||
| it | ||
| ) | ||
| } | ||
| return getNotificationConfigResponse | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Extension function for publishing a notification to a channel in the Notification plugin. | ||
| */ | ||
| suspend fun NotificationConfigInfo.sendNotificationWithHTML( | ||
| client: Client, | ||
| title: String, | ||
| compiledMessage: String, | ||
| compiledMessageHTML: String? | ||
| ): String { | ||
| val config = this | ||
| val res: SendNotificationResponse = NotificationsPluginInterface.suspendUntil { | ||
| this.sendNotification( | ||
| (client as NodeClient), | ||
| EventSource(title, config.configId, SeverityType.INFO), | ||
| ChannelMessage(compiledMessage, compiledMessageHTML, null), | ||
| listOf(config.configId), | ||
| it | ||
| ) | ||
| } | ||
| validateResponseStatus(res.getStatus(), res.notificationEvent.toString()) | ||
| return res.notificationEvent.toString() | ||
| } | ||
|
|
||
| /** | ||
| * Converts [NotificationsPluginInterface] methods that take a callback into a kotlin suspending function. | ||
| * | ||
| * @param block - a block of code that is passed an [ActionListener] that should be passed to the NotificationsPluginInterface API. | ||
| */ | ||
| suspend fun <T> NotificationsPluginInterface.suspendUntil(block: NotificationsPluginInterface.(ActionListener<T>) -> Unit): T = | ||
| suspendCoroutine { cont -> | ||
| block(object : ActionListener<T> { | ||
| override fun onResponse(response: T) = cont.resume(response) | ||
|
|
||
| override fun onFailure(e: Exception) = cont.resumeWithException(e) | ||
| }) | ||
| } | ||
|
|
||
| /** | ||
| * All valid response statuses. | ||
| */ | ||
| private val VALID_RESPONSE_STATUS = setOf( | ||
| RestStatus.OK.status, | ||
| RestStatus.CREATED.status, | ||
| RestStatus.ACCEPTED.status, | ||
| RestStatus.NON_AUTHORITATIVE_INFORMATION.status, | ||
| RestStatus.NO_CONTENT.status, | ||
| RestStatus.RESET_CONTENT.status, | ||
| RestStatus.PARTIAL_CONTENT.status, | ||
| RestStatus.MULTI_STATUS.status | ||
| ) | ||
|
|
||
| @Throws(OpenSearchStatusException::class) | ||
| fun validateResponseStatus(restStatus: RestStatus, responseContent: String) { | ||
| if (!VALID_RESPONSE_STATUS.contains(restStatus.status)) { | ||
| throw OpenSearchStatusException("Failed: $responseContent", restStatus) | ||
| } | ||
| } |