2 * Copyright © 2018-2019 AT&T Intellectual Property.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.topology
19 import org.apache.kafka.streams.processor.To
20 import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.AbstractMessagePrioritizeProcessor
21 import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.MessagePrioritizationConstants
22 import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.MessageState
23 import org.onap.ccsdk.cds.controllerblueprints.core.logger
25 open class MessageAggregateProcessor : AbstractMessagePrioritizeProcessor<String, String>() {
27 private val log = logger(MessageAggregateProcessor::class)
29 override suspend fun processNB(key: String, value: String) {
31 log.info("@@@@@ received in aggregation processor key($key), value($value)")
32 val ids = value.split(",").map { it.trim() }
33 if (!ids.isNullOrEmpty()) {
36 processorContext.forward(key, ids.first(), To.child(MessagePrioritizationConstants.PROCESSOR_OUTPUT))
38 /** Implement Aggregation logic in overridden class, If necessary,
39 Populate New Message and Update status with Prioritized, Forward the message to next processor */
40 handleAggregation(ids)
41 /** Update all messages to Aggregated state */
42 messagePrioritizationStateService.setMessagesState(ids, MessageState.AGGREGATED.name)
44 } catch (e: Exception) {
45 val error = "failed in Aggregate message($ids) : ${e.message}"
47 val storeMessages = messagePrioritizationStateService.getMessages(ids)
48 if (!storeMessages.isNullOrEmpty()) {
49 storeMessages.forEach { messagePrioritization ->
51 /** Update the data store */
52 messagePrioritizationStateService.setMessageStateANdError(
53 messagePrioritization.id,
54 MessageState.ERROR.name, error
56 /** Publish to Error topic */
57 this.processorContext.forward(
58 messagePrioritization.id, messagePrioritization,
59 To.child(MessagePrioritizationConstants.SINK_OUTPUT)
61 } catch (sendException: Exception) {
63 "failed to update/publish error message(${messagePrioritization.id}) : " +
64 "${sendException.message}", e
73 /** Child will override this implementation , if necessary */
74 open suspend fun handleAggregation(messageIds: List<String>) {
75 log.info("messages($messageIds) aggregated")
76 messageIds.forEach { id ->
77 processorContext.forward(id, id, To.child(MessagePrioritizationConstants.PROCESSOR_OUTPUT))