summaryrefslogtreecommitdiffstats
path: root/controlloop/common/actors/actor.vfc/src/main/java/org/onap/policy/controlloop/actor/vfc/VFCActorServiceProvider.java
blob: 81472625692abd98cfdcd74c0155545905db5463 (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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*-
 * ============LICENSE_START=======================================================
 * Copyright (C) 2017 Intel Corp. All rights reserved.
 * ================================================================================
 * 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.
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.controlloop.actor.vfc;

import java.util.Collections;
import java.util.List;
import java.util.UUID;
import org.onap.policy.controlloop.VirtualControlLoopEvent;
import org.onap.policy.vfc.VFCRequest;
import org.onap.policy.vfc.VFCHealRequest;
import org.onap.policy.vfc.VFCHealAdditionalParams;
import org.onap.policy.vfc.VFCHealActionVmInfo;
import org.onap.policy.controlloop.ControlLoopOperation;
import org.onap.policy.controlloop.policy.Policy;
import org.onap.policy.drools.system.PolicyEngine;
import org.onap.policy.controlloop.actorServiceProvider.spi.Actor;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import org.onap.policy.aai.AAIManager;
import org.onap.policy.aai.AAIGETVnfResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class VFCActorServiceProvider implements Actor {

    private static final Logger logger = LoggerFactory.getLogger(VFCActorServiceProvider.class);
    private static final ImmutableList<String> recipes = ImmutableList.of("Restart");
    private static final ImmutableMap<String, List<String>> targets = new ImmutableMap.Builder<String, List<String>>()
            .put("Restart", ImmutableList.of("VM")).build();

    @Override
    public String actor() {
        return "VFC";
    }

    @Override
    public List<String> recipes() {
        return ImmutableList.copyOf(recipes);
    }

    @Override
    public List<String> recipeTargets(String recipe) {
        return ImmutableList.copyOf(targets.getOrDefault(recipe, Collections.emptyList()));
    }

    @Override
    public List<String> recipePayloads(String recipe) {
        return Collections.emptyList();
    }

    public static VFCRequest constructRequest(VirtualControlLoopEvent onset, ControlLoopOperation operation,
                                              Policy policy, AAIGETVnfResponse vnfResponse) {
        // Construct an VFC request
        VFCRequest request = new VFCRequest();
        String serviceInstance = onset.AAI.get("service-instance.service-instance-id");
        if (serviceInstance == null || "".equals(serviceInstance))
        {
        	if(vnfResponse == null) //if the response is null, we haven't queried
        	{
        		vnfResponse = getAAIServiceInstance(onset); //This does the AAI query since we haven't already
        		if (vnfResponse == null)
        		    return null;
        	}
        	serviceInstance = vnfResponse.serviceId;
        }
        request.nsInstanceId = serviceInstance;
        request.requestId = onset.requestID;
        request.healRequest = new VFCHealRequest();
        request.healRequest.vnfInstanceId = onset.AAI.get("generic-vnf.vnf-id");
        request.healRequest.cause = operation.message;
        request.healRequest.additionalParams = new VFCHealAdditionalParams();
        
        switch (policy.getRecipe().toLowerCase()) {
            case "restart":
                request.healRequest.additionalParams.action = "restartvm";
                request.healRequest.additionalParams.actionInfo = new VFCHealActionVmInfo();
                request.healRequest.additionalParams.actionInfo.vmid = onset.AAI.get("vserver.vserver-id");
                request.healRequest.additionalParams.actionInfo.vmname = onset.AAI.get("vserver.vserver-name");
                break;
            default:
                return null;
        }
        return request;
    }


    private static AAIGETVnfResponse getAAIServiceInstance(VirtualControlLoopEvent event) {
        AAIGETVnfResponse response = null;
        UUID requestID = event.requestID;
        String vnfName = event.AAI.get("generic-vnf.vnf-name");
        String vnfID = event.AAI.get("generic-vnf.vnf-id");
        String aaiUrl = PolicyEngine.manager.getEnvironmentProperty("aai.url");
        String aaiUsername = PolicyEngine.manager.getEnvironmentProperty("aai.username");
        String aaiPassword = PolicyEngine.manager.getEnvironmentProperty("aai.password");
        try {
            if (vnfName != null) {
                String url = aaiUrl + "/aai/v11/network/generic-vnfs/generic-vnf?vnf-name=";
                response = AAIManager.getQueryByVnfName(url, aaiUsername, aaiPassword, requestID, vnfName);
            } else if (vnfID != null) {
                String url = aaiUrl + "/aai/v11/network/generic-vnfs/generic-vnf/";
                response = AAIManager.getQueryByVnfID(url, aaiUsername, aaiPassword, requestID, vnfID);
            } else {
                logger.error("getAAIServiceInstance failed");
            }
        } catch (Exception e) {
            logger.error("getAAIServiceInstance exception: ", e);
        }
        return response;
    }
}