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
26 open class MessageAggregateProcessor : AbstractMessagePrioritizeProcessor<String, String>() {
28 private val log = logger(MessageAggregateProcessor::class)
30 override suspend fun processNB(key: String, value: String) {
32 log.info("@@@@@ received in aggregation processor key($key), value($value)")
33 val ids = value.split(",").map { it.trim() }
34 if (!ids.isNullOrEmpty()) {
37 processorContext.forward(key, ids.first(), To.child(MessagePrioritizationConstants.PROCESSOR_OUTPUT))
39 /** Implement Aggregation logic in overridden class, If necessary,
40 Populate New Message and Update status with Prioritized, Forward the message to next processor */
41 handleAggregation(ids)
42 /** Update all messages to Aggregated state */
43 messagePrioritizationStateService.setMessagesState(ids, MessageState.AGGREGATED.name)
45 } catch (e: Exception) {
46 val error = "failed in Aggregate message($ids) : ${e.message}"
48 val storeMessages = messagePrioritizationStateService.getMessages(ids)
49 if (!storeMessages.isNullOrEmpty()) {
50 storeMessages.forEach { messagePrioritization ->
52 /** Update the data store */
53 messagePrioritizationStateService.setMessageStateANdError(messagePrioritization.id,
54 MessageState.ERROR.name, error)
55 /** Publish to Error topic */
56 this.processorContext.forward(messagePrioritization.id, messagePrioritization,
57 To.child(MessagePrioritizationConstants.SINK_OUTPUT))
58 } catch (sendException: Exception) {
59 log.error("failed to update/publish error message(${messagePrioritization.id}) : " +
60 "${sendException.message}", e)
69 /** Child will override this implementation , if necessary */
70 open suspend fun handleAggregation(messageIds: List<String>) {
71 log.info("messages($messageIds) aggregated")
72 messageIds.forEach { id ->
73 processorContext.forward(id, id, To.child(MessagePrioritizationConstants.PROCESSOR_OUTPUT))