summaryrefslogtreecommitdiffstats
path: root/engine-d/src/main/java/org/onap/holmes/engine/utils/AlarmUtil.java
blob: c7eb02bf3a0fc198aea6b1f7b21d57ebb8d77d27 (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
83
84
85
86
87
88
89
90
91
92
93
/**
 * Copyright 2017 ZTE Corporation.
 *
 * 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.holmes.engine.utils;

import java.util.HashMap;
import java.util.Map;
import org.jvnet.hk2.annotations.Service;
import org.onap.holmes.common.api.stat.Alarm;

@Service
public class AlarmUtil {

    private final static AlarmUtil alarmUtil = new AlarmUtil();
    /**
     * Map<ruleId, <ProbableCause-EquipType, priority>>
     */
    private final Map<String, Map<String, Integer>> rootPriorityMap =
        new HashMap<String, Map<String, Integer>>();
    /**
     * Map<rule, ProbableCause+EquipType+priority>
     */
    private final Map<String, String> saveRuleMsg = new HashMap<String, String>();

    private AlarmUtil() {
    }

    public static AlarmUtil getInstance() {
        return alarmUtil;
    }

    public boolean equipTypeFilter(String probableCauseStr, String equipType, Alarm alarm) {
        if ("null".equals(probableCauseStr)) {
            return true;
        }
        String[] equipTypes = equipType.replace(" ", "").split(",");
        String[] probableCauseStrs = probableCauseStr.replace(" ", "").split(",");
        for (int i = 0; i < probableCauseStrs.length; i++) {
            if (alarm.getProbableCause() == Long.parseLong(probableCauseStrs[i])
                && alarm.getEquipType().equals(equipTypes[i])) {
                return true;
            }
        }
        return false;
    }

    public Integer getPriority(String ruleId, String probableCauseStr, String rootAlarmFeatureStr,
        String equipTypeStr, Alarm alarm) {
        if (rootPriorityMap.containsKey(ruleId)) {
            if (!saveRuleMsg.get(ruleId)
                .equals(probableCauseStr + equipTypeStr + rootAlarmFeatureStr)) {
                setPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr);
            }
        } else {
            setPriority(ruleId, probableCauseStr, rootAlarmFeatureStr, equipTypeStr);
        }

        Integer priority =
            rootPriorityMap.get(ruleId).get(alarm.getProbableCause() + "-" + alarm.getEquipType());
        if (priority == null) {
            priority = 0;
        }
        return priority;
    }

    private void setPriority(String ruleId, String probableCauseStr, String rootAlarmFeatureStr,
        String equipTypeStr) {
        saveRuleMsg.put(ruleId, probableCauseStr + equipTypeStr + rootAlarmFeatureStr);

        Map<String, Integer> map = new HashMap<String, Integer>();
        String[] probableCauseStrs = probableCauseStr.replace(" ", "").split(",");
        String[] rootAlarmFeatureStrs = rootAlarmFeatureStr.replace(" ", "").split(",");
        String[] equipTypes = equipTypeStr.replace(" ", "").split(",");
        for (int i = 0; i < rootAlarmFeatureStrs.length; i++) {
            map.put(probableCauseStrs[i] + "-" + equipTypes[i],
                Integer.parseInt(rootAlarmFeatureStrs[i]));
        }

        rootPriorityMap.put(ruleId, map);
    }
}