[vfc/nfvo/resmanagement.git] /

/*
 * Copyright 2016-2017 Huawei Technologies Co., Ltd.
 *
 * 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.vfc.nfvo.resmanagement.service.group.impl;

import java.util.HashMap;
import java.util.Map;

import org.onap.vfc.nfvo.resmanagement.common.constant.ParamConstant;
import org.onap.vfc.nfvo.resmanagement.common.constant.UrlConstant;
import org.onap.vfc.nfvo.resmanagement.common.util.JsonUtil;
import org.onap.vfc.nfvo.resmanagement.common.util.RestfulUtil;
import org.onap.vfc.nfvo.resmanagement.service.base.openstack.inf.InterfaceResManagement;
import org.onap.vfc.nfvo.resmanagement.common.util.restclient.ServiceException;
import org.onap.vfc.nfvo.resmanagement.common.util.restclient.RestfulParametes;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.annotation.Transactional;

import net.sf.json.JSONArray;
import net.sf.json.JSONObject;

/**
 * iResource add service implementation.<br>
 * <p>
 * </p>
 *
 * @author
 * @version VFC 1.0 Sep 10, 2016
 */
public class IResourceAddServiceImpl {

    private static final Logger LOGGER = LoggerFactory.getLogger(IResourceAddServiceImpl.class);

    /**
     * Add iResource.<br>
     *
     * @param restParametes
     * @param iResMap
     * @throws ServiceException
     * @since VFC 1.0
     */
    @Transactional(rollbackFor = ServiceException.class)
    public void addIRes(RestfulParametes restParametes, Map<String, InterfaceResManagement> iResMap)
            throws ServiceException {
        addIResources(iResMap, createResUrlMap(), restParametes);
    }

    private void addIResources(Map<String, InterfaceResManagement> iResMap, HashMap<String, String> urlMap,
            RestfulParametes restParametes) throws ServiceException {
        for(String iResName : iResMap.keySet()) {
            if(ParamConstant.PARAM_HOST.equals(iResName)) {
                addHostResource(iResMap, restParametes, String.format(urlMap.get(iResName),
                        restParametes.get(ParamConstant.PARAM_VIMID), restParametes.get(ParamConstant.PARAM_TENANTID)),
                        iResName);
            } else {
                String url = String.format(urlMap.get(iResName), restParametes.get(ParamConstant.PARAM_VIMID),
                        restParametes.get(ParamConstant.PARAM_TENANTID));
                JSONArray iResArray = RestfulUtil.getResponseRes(new RestfulParametes(), url, iResName);
                LOGGER.warn("function=addIResources; iResArray={}", iResArray);
                for(Object object : iResArray) {
                    JSONObject iRes = JSONObject.fromObject(object);
                    int result = iResMap.get(iResName).add(iRes);
                    LOGGER.warn("function=addIResources; msg=iRes name is [{}],result is [{}]", iResName, result);
                }
            }
        }
    }

    /**
     * Add Host Resource.<br>
     *
     * @param iResMap
     * @param restParametes
     * @param url
     * @param iResName
     * @throws ServiceException
     * @since VFC 1.0
     */
    public static void addHostResource(Map<String, InterfaceResManagement> iResMap, RestfulParametes restParametes,
            String url, String iResName) throws ServiceException {
        JSONArray hostResArray = RestfulUtil.getResponseRes(new RestfulParametes(), url, iResName);
        LOGGER.warn("function=addHostResource; hostResArray={}", hostResArray);
        for(Object object : hostResArray) {
            JSONObject hostRes = JSONObject.fromObject(object);
            String hostZone = hostRes.getString("zone");
            if("internal".equals(hostZone)) {
                continue;
            }
            String hostName = hostRes.getString("name");
            String hostUrl = String.format(UrlConstant.GET_HOSTDETAIL_URL, restParametes.get(ParamConstant.PARAM_VIMID),
                    restParametes.get(ParamConstant.PARAM_TENANTID), hostName);

            String result = RestfulUtil.getResponseContent(hostUrl, new RestfulParametes(), ParamConstant.PARAM_GET);
            if(result == null) {
                continue;
            }
            JSONObject hostObj = JSONObject.fromObject(result);
            JSONObject host = hostDataParse(hostObj, hostName);
            int res = iResMap.get(iResName).add(host);
            LOGGER.warn("function=addHostResource; result={}, res={}", result, res);
            if(res < 0) {
                LOGGER.error("function=addHostResource; add into DB fail!");
            }

        }
    }

    /**
     * <br>
     *
     * @param hostObj
     * @param hostName
     * @return
     * @since VFC 1.0
     */
    public static JSONObject hostDataParse(JSONObject hostObj, String hostName) {
        LOGGER.warn("function=hostDataParse; hostObj={}, hostName={}", hostObj, hostName);
        JSONArray hostArray = hostObj.getJSONArray("host");
        for(Object object : hostArray) {
            JSONObject hostObject = JSONObject.fromObject(object);
            if(hostObject.getString("project").contains("total")) {
                String vimId = JsonUtil.getJsonFieldStr(hostObj, "vimId");
                String hostId = vimId + hostName;
                JSONObject host = new JSONObject();
                host.put("id", hostId);
                host.put("name", hostName);
                host.put("cpu", JsonUtil.getJsonFieldStr(hostObject, "cpu"));
                host.put("memory", JsonUtil.getJsonFieldStr(hostObject, "memory_mb"));
                host.put("disk", JsonUtil.getJsonFieldStr(hostObject, "disk_gb"));
                host.put("vimId", JsonUtil.getJsonFieldStr(hostObj, "vimId"));
                host.put("vimName", JsonUtil.getJsonFieldStr(hostObj, "vimName"));
                return host;
            }
        }
        return new JSONObject();
    }

    private HashMap<String, String> createResUrlMap() {
        HashMap<String, String> urlMap = new HashMap<String, String>(10);
        urlMap.put(ParamConstant.PARAM_NETWORK, UrlConstant.GET_NETWORK_URL);
        urlMap.put(ParamConstant.PARAM_HOST, UrlConstant.GET_HOST_URL);
        urlMap.put(ParamConstant.PARAM_PORT, UrlConstant.GET_PORT_URL);
        return urlMap;
    }
}