artifactbroker/plugins/forwarding-plugins/src/main/java/org/onap/policy/distribution/forwarding/k8s/K8sArtifactForwarder.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.distribution.forwarding.k8s;
  22
  23 import com.google.gson.Gson;
  24 import com.google.gson.GsonBuilder;
  25 import com.google.gson.reflect.TypeToken;
  26
  27 import java.io.File;
  28 import java.io.IOException;
  29 import java.util.ArrayList;
  30 import java.util.Collection;
  31 import java.util.Collections;
  32 import java.util.LinkedHashMap;
  33 import java.util.List;
  34 import java.util.Map;
  35
  36
  37 import org.apache.http.HttpEntity;
  38 import org.apache.http.NameValuePair;
  39 import org.apache.http.client.ClientProtocolException;
  40 import org.apache.http.client.entity.UrlEncodedFormEntity;
  41 import org.apache.http.client.methods.CloseableHttpResponse;
  42 import org.apache.http.client.methods.HttpGet;
  43 import org.apache.http.client.methods.HttpPost;
  44 import org.apache.http.entity.FileEntity;
  45 import org.apache.http.entity.StringEntity;
  46 import org.apache.http.impl.client.CloseableHttpClient;
  47 import org.apache.http.impl.client.HttpClients;
  48 import org.apache.http.message.BasicNameValuePair;
  49 import org.apache.http.util.EntityUtils;
  50
  51 import org.onap.policy.common.logging.flexlogger.FlexLogger;
  52 import org.onap.policy.common.logging.flexlogger.Logger;
  53 import org.onap.policy.common.parameters.ParameterService;
  54 import org.onap.policy.distribution.forwarding.ArtifactForwarder;
  55 import org.onap.policy.distribution.model.CloudArtifact;
  56 import org.onap.policy.distribution.model.PolicyInput;
  57 import org.onap.policy.distribution.model.VfModuleModel;
  58 import org.onap.sdc.api.notification.IArtifactInfo;
  59
  60
  61 /**
  62  * Forwards policies to the XACML PDP.
  63  */
  64 public class K8sArtifactForwarder implements ArtifactForwarder {
  65
  66     private static final Logger LOGGER = FlexLogger.getLogger(K8sArtifactForwarder.class);
  67     private static final String BASE_PATH = "http://localhost:8081/v1/rb/definition";
  68     private static final String CLOUD_TECHNOLOGY_SPECIFIC_ARTIFACT = "CLOUD_TECHNOLOGY_SPECIFIC_ARTIFACT";
  69     private Map<String, IArtifactInfo> artifactMap;
  70
  71     private K8sArtifactForwarderParameterGroup configurationParameters = null;
  72
  73
  74
  75     @Override
  76     public void forward(PolicyInput  policyInput) {
  77         if (policyInput instanceof CloudArtifact) {
  78             System.out.println("get a CloudArtifact !");
  79             CloudArtifact cloudArtifact = (CloudArtifact) policyInput;
  80             artifactMap = cloudArtifact.getArtifactTypeMap();
  81             System.out.println("the artifactMap = " + artifactMap);
  82             ArrayList<VfModuleModel> vfModuleModels = cloudArtifact.getVfModulePayload();
  83             System.out.println("the size of vfModule = " + vfModuleModels.size());
  84
  85             for (VfModuleModel vfModule : vfModuleModels) {
  86                 forwardAndUpload(vfModule);
  87             }
  88         } else {
  89             System.out.println("NOT a CloudArtifact type !");
  90             return;
  91         }
  92     }
  93
  94     private void forwardAndUpload(VfModuleModel vfModule) {
  95
  96         System.out.println("before create type !");
  97         boolean definitionCreated = createDefinition(vfModule);
  98         System.out.println(" after create type !");
  99         if (definitionCreated) {
 100             uploadArtifact(vfModule);
 101         }
 102     }
 103
 104     private boolean createDefinition(VfModuleModel vfModule) {
 105         try {
 106             HttpPost httpPost = new HttpPost(BASE_PATH);
 107             httpPost.addHeader("Accept", "application/json");
 108             httpPost.addHeader("Content-Type", "application/json");
 109
 110             Gson gson = new GsonBuilder().enableComplexMapKeySerialization()
 111                 .create();
 112
 113             Map<String, Object> map = new LinkedHashMap<String, Object>();
 114             map.put("rb-name", vfModule.getVfModuleModelName());
 115             map.put("rb-version", vfModule.getVfModuleModelVersion());
 116             map.put("descritpion",vfModule.getVfModuleModelDescription());
 117             Map<String, String> labelMap = new LinkedHashMap<String, String>();
 118             labelMap.put("vnf_customization_uuid",vfModule.getVfModuleModelCustomizationUUID());
 119             map.put("labels", labelMap);
 120             String json = gson.toJson(map);
 121
 122             StringEntity entity = new StringEntity(json);
 123             httpPost.setEntity(entity);
 124             return invokeHttpPost("definition", httpPost);
 125         } catch (Exception e) {
 126             System.out.println("create definition error");
 127             return false;
 128         }
 129
 130     }
 131
 132     private boolean uploadArtifact(VfModuleModel vfModule) {
 133         String url = BASE_PATH + "/" + vfModule.getVfModuleModelName() + "/"
 134             + vfModule.getVfModuleModelVersion() + "/content";
 135         HttpPost httpPost = new HttpPost(url);
 136         httpPost.addHeader("content-type", "application/x-www-form-urlencoded;charset=utf-8");
 137
 138         List<String> artifacts = vfModule.getArtifacts();
 139         System.out.println("artifacts = " + artifacts);
 140
 141         String cloudUuid = null;
 142         IArtifactInfo cloudArtifact = null;
 143         boolean found = false;
 144
 145         for (String artifact: artifacts) {
 146             if ( artifactMap.get(artifact) != null
 147                 && artifactMap.get(artifact).getArtifactType().equals("CLOUD_TECHNOLOGY_SPECIFIC_ARTIFACT")) {
 148                 cloudArtifact = artifactMap.get(artifact);
 149                 cloudUuid = cloudArtifact.getArtifactUUID();
 150                 found = true;
 151                 break;
 152             }
 153         }
 154
 155         if ( found == false ) {
 156             System.out.println(" meets error , no CLOUD_TECHNOLOGY_SPECIFIC_ARTIFACT type found ");
 157             return false;
 158         }
 159         String cloudArtifactPath = "/data/" + vfModule.getVfModuleModelCustomizationUUID()
 160             + "/" + cloudArtifact.getArtifactName();
 161         File file = new File(cloudArtifactPath);
 162         FileEntity entity = new FileEntity(file);
 163         httpPost.setEntity(entity);
 164
 165         return invokeHttpPost("uploading", httpPost);
 166     }
 167
 168
 169     @Override
 170     public void configure(String parameterGroupName) {
 171         configurationParameters = ParameterService.get(parameterGroupName);
 172     }
 173
 174     protected static boolean invokeHttpPost(String action, HttpPost httpPost)  {
 175         System.out.println("httpPost begin!");
 176         boolean ret = false;
 177
 178         String errorMsg;
 179         label1: {
 180             try ( CloseableHttpClient httpClient = HttpClients.createDefault() ) {
 181                 System.out.println("result1") ;
 182                 CloseableHttpResponse closeableHttpResponse = httpClient.execute(httpPost);
 183                 System.out.println("result2") ;
 184                 String result = EntityUtils.toString(closeableHttpResponse.getEntity());
 185                 System.out.println("result = {}" + result);
 186                 System.out.println("status  = {}" + closeableHttpResponse.getStatusLine().getStatusCode());
 187                 if ( closeableHttpResponse.getStatusLine().getStatusCode() != 200 ) {
 188                     System.out.println("exception: ret= " + closeableHttpResponse.getStatusLine().getStatusCode());
 189                 } else {
 190                     ret = true;
 191                 }
 192
 193                 closeableHttpResponse.close();
 194                 break label1;
 195             } catch (IOException var) {
 196                 errorMsg = action + ":httpPostWithJSON connect faild";
 197                 System.out.println("exception: POST_CONNECT_FAILD : {}" + errorMsg);
 198             }
 199         }
 200
 201         System.out.println("httpPost end!");
 202         return ret;
 203     }
 204
 205 }