39d081455f31c7e2846f0cafc6c55561ba10c3ea
[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.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.db.MessagePrioritization
20 import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.service.MessagePrioritizationStateService
21 import org.onap.ccsdk.cds.blueprintsprocessor.services.execution.AbstractComponentFunction
22 import org.onap.ccsdk.cds.controllerblueprints.core.service.BluePrintDependencyService
23
24 /**
25  * Register the MessagePrioritizationStateService and exposed dependency
26  */
27 fun BluePrintDependencyService.messagePrioritizationStateService(): MessagePrioritizationStateService =
28     instance(MessagePrioritizationStateService::class)
29
30 /**
31  * Expose messagePrioritizationStateService to AbstractComponentFunction
32  */
33 fun AbstractComponentFunction.messagePrioritizationStateService() =
34     BluePrintDependencyService.messagePrioritizationStateService()
35
36 /**
37  * MessagePrioritization correlation extensions
38  */
39 fun MessagePrioritization.toFormatedCorrelation(): String {
40     val ascendingKey = this.correlationId!!.split(",")
41         .map { it.trim() }.sorted().joinToString(",")
42     return ascendingKey
43 }
44
45 fun MessagePrioritization.toTypeNCorrelation(): TypeCorrelationKey {
46     val ascendingKey = this.correlationId!!.split(",")
47         .map { it.trim() }.sorted().joinToString(",")
48     return TypeCorrelationKey(this.type, ascendingKey)
49 }