huawei/vnfmadapter/VnfmadapterService/service/src/main/java/org/openo/nfvo/vnfmadapter/service/process/VnfResourceMgr.java

   1 /*
   2  * Copyright 2016-2017 Huawei Technologies Co., Ltd.
   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.openo.nfvo.vnfmadapter.service.process;
  18
  19 import java.util.HashMap;
  20 import java.util.Map;
  21
  22 import org.apache.commons.lang.StringUtils;
  23 import org.openo.baseservice.roa.util.restclient.RestfulResponse;
  24 import org.openo.nfvo.vnfmadapter.common.servicetoken.VnfmRestfulUtil;
  25 import org.openo.nfvo.vnfmadapter.service.constant.Constant;
  26 import org.openo.nfvo.vnfmadapter.service.constant.ParamConstants;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29
  30 import net.sf.json.JSONArray;
  31 import net.sf.json.JSONException;
  32 import net.sf.json.JSONObject;
  33
  34 /**
  35  * Provide function of resource for VNFM.
  36  * <br/>
  37  * <p>
  38  * </p>
  39  *
  40  * @author
  41  * @version NFVO 0.5 Aug 25, 2016
  42  */
  43 public class VnfResourceMgr {
  44
  45     private static final Logger LOG = LoggerFactory.getLogger(VnfResourceMgr.class);
  46
  47     /**
  48      * Provide function of grant resource for VNFM.
  49      * <br/>
  50      *
  51      * @param vnfObj
  52      * @param vnfId
  53      * @param vnfmId
  54      * @return
  55      * @since NFVO 0.5
  56      */
  57     public JSONObject grantVnfResource(JSONObject vnfObj, String vnfId, String vnfmId) {
  58         LOG.warn("function=grantVnfResource, msg=enter to grant vnf resource, params: {}", vnfObj);
  59         JSONObject resultJson = new JSONObject();
  60         resultJson.put("retCode", Constant.REST_FAIL);
  61         try {
  62             String type = vnfObj.getString("type");
  63             String requestType = vnfObj.getString("operation_right");
  64             String vnfName = vnfObj.getString("vnf_name");
  65
  66             if(StringUtils.isEmpty(type) || StringUtils.isEmpty(requestType) || StringUtils.isEmpty(vnfName)
  67                     || StringUtils.isEmpty(vnfId)) {
  68                 LOG.error("function=grantVnfResource, msg=grant basic params error");
  69                 resultJson.put("errorMsg", "basic params error");
  70                 return resultJson;
  71             }
  72
  73             JSONArray vmList = vnfObj.getJSONArray("vm_list");
  74
  75             Map<String, Integer> resMap = calculateGrantRes(vmList);
  76
  77             if(null == resMap) {
  78                 LOG.error("function=grantVnfResource, msg=grant resource params error");
  79                 resultJson.put("errorMsg", "resource params error");
  80                 return resultJson;
  81             }
  82
  83             JSONObject grantObj = new JSONObject();
  84             grantObj.put("vimId", vnfObj.getString("vim_id"));
  85             grantObj.put("vnfId", vnfId);
  86             grantObj.put("vnfName", vnfName);
  87             grantObj.put("vnfmId", vnfmId);
  88             String action = getGrantAction(type, requestType);
  89             grantObj.put("action", action);
  90
  91             JSONObject grantParam = parseGrantParam(resMap, grantObj);
  92             resultJson = sendGrantToResmgr(grantParam);
  93             LOG.error("function=grantVnfResource, resultJson={}.", resultJson);
  94         } catch(JSONException e) {
  95             LOG.error("function=grantVnfResource, msg=parse params occoured JSONException e={}.", e);
  96             resultJson.put("errorMsg", "params parse exception");
  97         }
  98
  99         return resultJson;
 100     }
 101
 102     /**
 103      * <br>
 104      *
 105      * @param grantParam
 106      * @return
 107      * @since NFVO 0.5
 108      */
 109     private JSONObject sendGrantToResmgr(JSONObject grantParam) {
 110         RestfulResponse rsp = VnfmRestfulUtil.getRemoteResponse(ParamConstants.GRANT_RES_URL, VnfmRestfulUtil.TYPE_PUT,
 111                 grantParam.toString());
 112         if(rsp == null || rsp.getStatus() != Constant.HTTP_OK) {
 113             return null;
 114         }
 115         LOG.error("funtion=sendGrantToResmgr, status={}", rsp.getStatus());
 116         return JSONObject.fromObject(rsp.getResponseContent());
 117     }
 118
 119     /**
 120      * <br>
 121      *
 122      * @param resMap
 123      * @param grantParam
 124      * @return
 125      * @since NFVO 0.5
 126      */
 127     private JSONObject parseGrantParam(Map<String, Integer> resMap, JSONObject grantParam) {
 128         JSONObject result = new JSONObject();
 129         result.put("vnfInstanceId", grantParam.getString("vnfId"));
 130         result.put("vimId", grantParam.getString("vimId"));
 131
 132         JSONArray resource = new JSONArray();
 133         JSONObject resourceObj = new JSONObject();
 134         resourceObj.put("type", "vdu");
 135         JSONObject resourceTemplate = new JSONObject();
 136         JSONObject storage = new JSONObject();
 137         storage.put("sizeOfStorage", resMap.get("diskNum"));
 138         storage.put("typeOfStorage", "");
 139         storage.put("swImageDescriptor", "");
 140         JSONObject compute = new JSONObject();
 141         JSONObject virtualMemory = new JSONObject();
 142         virtualMemory.put("virtualMemSize", resMap.get("memNum"));
 143         JSONObject virtualCpu = new JSONObject();
 144         virtualCpu.put("numVirtualCpu", resMap.get("cpuNum"));
 145         compute.put("virtualMemory", virtualMemory);
 146         compute.put("virtualCpu", virtualCpu);
 147         resourceTemplate.put("virtualStorageDescriptor", storage);
 148         resourceTemplate.put("virtualComputeDescriptor", compute);
 149         resourceObj.put("resourceTemplate", resourceTemplate);
 150         resourceObj.put("resourceDefinitionId", "");
 151         resourceObj.put("vdu", grantParam.getString("vnfName"));
 152         resource.add(resourceObj);
 153
 154         if("online".equals(grantParam.getString("action")) || "scaleOut".equals(grantParam.getString("action"))) {
 155             result.put("addResource", resource);
 156         } else {
 157             result.put("removeResource", resource);
 158         }
 159
 160         JSONObject additionalParam = new JSONObject();
 161         additionalParam.put("vnfmId", grantParam.getString("vnfmId"));
 162         additionalParam.put("vimId", grantParam.getString("vimId"));
 163         additionalParam.put("tenant", "");
 164         result.put("additionalParam", additionalParam);
 165         LOG.info("funtion=parseGrantParam, result={}", result);
 166         return result;
 167     }
 168
 169     private Map<String, Integer> calculateGrantRes(JSONArray vmList) {
 170         Map<String, Integer> resMap = new HashMap<>(Constant.DEFAULT_COLLECTION_SIZE);
 171         int vmSize = vmList.size();
 172         int cpuNum = 0;
 173         int memNum = 0;
 174         int diskNum = 0;
 175         int diskSize = 0;
 176         int cpuTmp = 0;
 177         int memTmp = 0;
 178         int diskTmp = 0;
 179         int initNum = 0;
 180
 181         try {
 182             for(int i = 0; i < vmSize; i++) {
 183                 JSONObject resInfo = vmList.getJSONObject(i);
 184                 JSONObject vmFlavor = resInfo.getJSONObject("vm_flavor");
 185                 initNum = Integer.parseInt(resInfo.getString("init_number"));
 186
 187                 if(initNum == 0) {
 188                     continue;
 189                 }
 190
 191                 JSONArray volumList = vmFlavor.getJSONArray("storage");
 192                 diskSize = volumList.size();
 193
 194                 for(int j = 0; j < diskSize; j++) {
 195                     JSONObject volumeInfo = volumList.getJSONObject(j);
 196                     diskTmp += getDiskQuantity(volumeInfo);
 197                 }
 198
 199                 cpuTmp = Integer.parseInt(vmFlavor.getString("num_cpus"));
 200                 memTmp = Integer.parseInt(vmFlavor.getString("mem_size"));
 201
 202                 cpuNum += cpuTmp * initNum;
 203                 memNum += memTmp * initNum;
 204                 diskNum += diskTmp * initNum;
 205
 206                 diskTmp = 0;
 207
 208             }
 209         } catch(JSONException e) {
 210             LOG.error("function=calculateGrantRes, msg=parse params occoured JSONException e={}.", e);
 211             return null;
 212         }
 213
 214         resMap.put("cpuNum", cpuNum);
 215         resMap.put("memNum", memNum);
 216         resMap.put("diskNum", diskNum);
 217         return resMap;
 218     }
 219
 220     private String getGrantAction(String type, String requestType) {
 221         String action = "unknown";
 222
 223         if(("increase").equals(requestType)) {
 224             if(("instantiation").equals(type)) {
 225                 action = "online";
 226             } else if(("scale").equals(type)) {
 227                 action = "scaleOut";
 228             }
 229
 230         } else if(("decrease").equals(requestType)) {
 231             if(("instantiation").equals(type)) {
 232                 action = "offline";
 233             } else if(("scale").equals(type)) {
 234                 action = "scaleIn";
 235             }
 236         }
 237
 238         return action;
 239     }
 240
 241     private int getDiskQuantity(JSONObject volumeObj) {
 242         int disk = 0;
 243         if(volumeObj.containsKey("vol_type")) {
 244             if("local_volume".equals(volumeObj.getString("vol_type"))) {
 245                 disk = Integer.parseInt(volumeObj.getString("vol_size"));
 246             }
 247         } else if(volumeObj.containsKey("storage_type") && "local_image".equals(volumeObj.getString("storage_type"))) {
 248
 249             disk = Integer.parseInt(volumeObj.getString("disk_size"));
 250
 251         }
 252         return disk;
 253     }
 254 }