Add message prioritization module
[ccsdk/cds.git] / ms / blueprintsprocessor / functions / message-prioritizaion / src / main / kotlin / org / onap / ccsdk / cds / blueprintsprocessor / functions / message / prioritization / MessagePrioritizeExtensions.kt
1 /*
2  * Copyright © 2018-2019 AT&T Intellectual Property.
3  *
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
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
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.
15  */
16
17 package org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization
18
19 import org.apache.kafka.streams.processor.ProcessorSupplier
20 import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.db.MessagePrioritization
21 import org.onap.ccsdk.cds.controllerblueprints.core.service.BluePrintDependencyService
22
23
24 fun <K, V> bluePrintProcessorSupplier(name: String, prioritizationConfiguration: PrioritizationConfiguration)
25         : ProcessorSupplier<K, V> {
26     return ProcessorSupplier<K, V> {
27         // Dynamically resolve the Prioritization Processor
28         val processorInstance = BluePrintDependencyService.instance<AbstractMessagePrioritizeProcessor<K, V>>(name)
29         processorInstance.prioritizationConfiguration = prioritizationConfiguration
30         processorInstance
31     }
32 }
33
34 fun MessagePrioritization.toFormatedCorrelation(): String {
35     val ascendingKey = this.correlationId!!.split(",")
36             .map { it.trim() }.sorted().joinToString(",")
37     return ascendingKey
38 }
39
40 fun MessagePrioritization.toTypeNCorrelation(): TypeCorrelationKey {
41     val ascendingKey = this.correlationId!!.split(",")
42             .map { it.trim() }.sorted().joinToString(",")
43     return TypeCorrelationKey(this.type, ascendingKey)
44 }