ms/blueprintsprocessor/functions/message-prioritization/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.services.execution.AbstractComponentFunction
  21 import org.onap.ccsdk.cds.controllerblueprints.core.service.BluePrintDependencyService
  22
  23 /**
  24  * Register the MessagePrioritizationStateService and exposed dependency
  25  */
  26 fun BluePrintDependencyService.messagePrioritizationStateService(): MessagePrioritizationStateService =
  27     instance(MessagePrioritizationStateService::class)
  28
  29 /**
  30  * Expose messagePrioritizationStateService to AbstractComponentFunction
  31  */
  32 fun AbstractComponentFunction.messagePrioritizationStateService() =
  33     BluePrintDependencyService.messagePrioritizationStateService()
  34
  35 /**
  36  * MessagePrioritization correlation extensions
  37  */
  38
  39 /**
  40  * Arrange comma separated correlation keys in ascending order.
  41  */
  42 fun MessagePrioritization.toFormatedCorrelation(): String {
  43     return this.correlationId!!.split(",")
  44         .map { it.trim() }.sorted().joinToString(",")
  45 }
  46
  47 /**
  48  * Used to group the correlation with respect to types.
  49  */
  50 fun MessagePrioritization.toTypeNCorrelation(): TypeCorrelationKey {
  51     return TypeCorrelationKey(this.type, this.toFormatedCorrelation())
  52 }
  53
  54 /** get list of message ids **/
  55 fun List<MessagePrioritization>.ids(): List<String> {
  56     return this.map { it.id }
  57 }
  58
  59 /** Ordered by highest priority and updated date **/
  60 fun List<MessagePrioritization>.orderByHighestPriority(): List<MessagePrioritization> {
  61     return this.sortedWith(compareBy(MessagePrioritization::priority, MessagePrioritization::updatedDate))
  62 }
  63
  64 /** Ordered by Updated date **/
  65 fun List<MessagePrioritization>.orderByUpdatedDate(): List<MessagePrioritization> {
  66     return this.sortedWith(compareBy(MessagePrioritization::updatedDate))
  67 }