aboutsummaryrefslogtreecommitdiffstats
path: root/ms/blueprintsprocessor/functions/message-prioritization/src/main/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/functions/message/prioritization/utils/MessageCorrelationUtils.kt
blob: 7ab0be09849de0f0a3770de72066ad3768949ddd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
 * Copyright © 2018-2019 AT&T Intellectual Property.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.utils

import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.CorrelationCheckResponse
import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.db.MessagePrioritization
import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.toFormatedCorrelation
import org.onap.ccsdk.cds.blueprintsprocessor.functions.message.prioritization.toTypeNCorrelation
import org.onap.ccsdk.cds.controllerblueprints.core.BluePrintProcessorException

object MessageCorrelationUtils {

    /** Assumption is message is of same group **/
    fun correlatedMessages(collectedMessages: List<MessagePrioritization>): CorrelationCheckResponse {
        val correlationCheckResponse = CorrelationCheckResponse(message = "not correlated")
        if (collectedMessages.size > 1) {
            val filteredMessage = collectedMessages.filter { !it.correlationId.isNullOrBlank() }
            if (filteredMessage.isNotEmpty()) {
                val groupedMessage = filteredMessage.groupBy { it.toFormatedCorrelation() }
                if (groupedMessage.size == 1) {
                    correlationCheckResponse.correlated = true
                    correlationCheckResponse.message = null
                }
            }
        } else {
            correlationCheckResponse.message = "received only one message for that group"
        }
        return correlationCheckResponse
    }

    /** Assumption is message is of same group and checking for required types **/
    fun correlatedMessagesWithTypes(collectedMessages: List<MessagePrioritization>, types: List<String>?):
        CorrelationCheckResponse {

            return if (!types.isNullOrEmpty() && collectedMessages.size > 1) {

                val unknownMessageTypes = collectedMessages.filter { !types.contains(it.type) }.map { it.id }
                if (!unknownMessageTypes.isNullOrEmpty()) {
                    throw BluePrintProcessorException("Messages($unknownMessageTypes) is not in type of($types)")
                }

                val copyTypes = types.toTypedArray().copyOf().toMutableList()

                val filteredMessage = collectedMessages.filter {
                    !it.correlationId.isNullOrBlank() &&
                        types.contains(it.type)
                }
                var correlatedKeys: MutableSet<String> = mutableSetOf()
                if (filteredMessage.isNotEmpty()) {
                    val correlatedMap = filteredMessage.groupBy { it.toTypeNCorrelation() }
                    val foundType = correlatedMap.keys.map { it.type }
                    copyTypes.removeAll(foundType)
                    correlatedKeys = correlatedMap.keys.map {
                        it.correlationId
                    }.toMutableSet()
                }
                /** Check if any Types missing and same correlation id for all types */
                return if (copyTypes.isEmpty()) {
                    if (correlatedKeys.size == 1) CorrelationCheckResponse(correlated = true)
                    else CorrelationCheckResponse(message = "not matching correlation keys($correlatedKeys)")
                } else {
                    CorrelationCheckResponse(message = "couldn't find types($copyTypes)")
                }
            } else {
                return correlatedMessages(collectedMessages)
            }
        }
}