dashboard/webapp-backend/src/main/java/org/oransc/ric/portal/dashboard/policyagentapi/PolicyAgentApiImpl.java

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2019 AT&T Intellectual Property
   6  * %%
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ========================LICENSE_END===================================
  19  */
  20 package org.oransc.ric.portal.dashboard.policyagentapi;
  21
  22 import org.oransc.ric.portal.dashboard.DashboardConstants;
  23 import org.oransc.ric.portal.dashboard.model.ImmutablePolicyInfo;
  24 import org.oransc.ric.portal.dashboard.model.PolicyInfo;
  25 import org.oransc.ric.portal.dashboard.model.PolicyInstances;
  26 import org.oransc.ric.portal.dashboard.model.PolicyType;
  27 import org.oransc.ric.portal.dashboard.model.PolicyTypes;
  28 import org.springframework.beans.factory.annotation.Autowired;
  29 import org.springframework.http.HttpStatus;
  30 import org.springframework.http.ResponseEntity;
  31 import org.springframework.stereotype.Component;
  32 import org.springframework.web.client.RestClientException;
  33 import org.springframework.web.client.RestTemplate;
  34
  35 import java.util.Collection;
  36 import java.util.List;
  37 import java.util.Map;
  38 import java.util.Vector;
  39 import java.lang.invoke.MethodHandles;
  40 import java.lang.reflect.Type;
  41
  42 import com.google.gson.GsonBuilder;
  43 import com.google.gson.JsonArray;
  44 import com.google.gson.JsonElement;
  45 import com.google.gson.JsonObject;
  46 import com.google.gson.JsonParser;
  47 import com.google.gson.annotations.SerializedName;
  48 import com.google.gson.reflect.TypeToken;
  49
  50 import org.slf4j.Logger;
  51 import org.slf4j.LoggerFactory;
  52 import org.immutables.gson.Gson;
  53 import org.immutables.value.Value;
  54
  55 @Component("PolicyAgentApi")
  56 public class PolicyAgentApiImpl implements PolicyAgentApi {
  57     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
  58
  59     RestTemplate restTemplate = new RestTemplate();
  60
  61     private static com.google.gson.Gson gson = new GsonBuilder() //
  62             .serializeNulls() //
  63             .create(); //
  64
  65     private final String urlPrefix;
  66
  67     @Autowired
  68     public PolicyAgentApiImpl(
  69             @org.springframework.beans.factory.annotation.Value("${policycontroller.url.prefix}") final String urlPrefix) {
  70         logger.debug("ctor prefix '{}'", urlPrefix);
  71         this.urlPrefix = urlPrefix;
  72     }
  73
  74     private String baseUrl() {
  75         return urlPrefix;
  76     }
  77
  78     @Value.Immutable
  79     @Gson.TypeAdapters
  80     interface PolicyTypeInfo {
  81
  82         public String name();
  83
  84         public String schema();
  85     }
  86
  87     @Override
  88     public ResponseEntity<PolicyTypes> getAllPolicyTypes() throws RestClientException {
  89         String url = baseUrl() + "/policy_schemas";
  90         ResponseEntity<String> rsp = this.restTemplate.getForEntity(url, String.class);
  91         if (!rsp.getStatusCode().is2xxSuccessful()) {
  92             return new ResponseEntity<>(rsp.getStatusCode());
  93         }
  94
  95         PolicyTypes result = new PolicyTypes();
  96         JsonParser jsonParser = new JsonParser();
  97         JsonArray schemas = jsonParser.parse(rsp.getBody()).getAsJsonArray();
  98         for (JsonElement schema : schemas) {
  99             JsonObject schemaObj = schema.getAsJsonObject();
 100             String title = schemaObj.get("title").getAsString();
 101             String schemaAsStr = schemaObj.toString();
 102             PolicyType pt = new PolicyType(title, schemaAsStr);
 103             result.add(pt);
 104         }
 105         return new ResponseEntity<>(result, rsp.getStatusCode());
 106     }
 107
 108     @Override
 109     public ResponseEntity<PolicyInstances> getPolicyInstancesForType(String type) {
 110         String url = baseUrl() + "/policies?type={type}";
 111         Map<String, ?> uriVariables = Map.of("type", type);
 112         ResponseEntity<String> rsp = this.restTemplate.getForEntity(url, String.class, uriVariables);
 113         if (!rsp.getStatusCode().is2xxSuccessful()) {
 114             return new ResponseEntity<>(rsp.getStatusCode());
 115         }
 116
 117         Type listType = new TypeToken<List<ImmutablePolicyInfo>>() {
 118         }.getType();
 119         List<PolicyInfo> rspParsed = gson.fromJson(rsp.getBody(), listType);
 120
 121         PolicyInstances result = new PolicyInstances();
 122         for (PolicyInfo p : rspParsed) {
 123             result.add(p);
 124         }
 125         return new ResponseEntity<>(result, rsp.getStatusCode());
 126     }
 127
 128     @Override
 129     public ResponseEntity<String> getPolicyInstance(String id) throws RestClientException {
 130         String url = baseUrl() + "/policy?instance={id}";
 131         Map<String, ?> uriVariables = Map.of("id", id);
 132
 133         return this.restTemplate.getForEntity(url, String.class, uriVariables);
 134     }
 135
 136     @Override
 137     public ResponseEntity<String> putPolicy(String policyTypeIdString, String policyInstanceId, String json, String ric)
 138             throws RestClientException {
 139         String url = baseUrl() + "/policy?type={type}&instance={instance}&ric={ric}&service={service}";
 140         Map<String, ?> uriVariables = Map.of( //
 141                 "type", policyTypeIdString, //
 142                 "instance", policyInstanceId, //
 143                 "ric", ric, //
 144                 "service", "dashboard");
 145
 146         try {
 147             this.restTemplate.put(url, json, uriVariables);
 148             return new ResponseEntity<>(HttpStatus.OK);
 149         } catch (Exception e) {
 150             return new ResponseEntity<>(e.getMessage(), HttpStatus.NOT_FOUND);
 151         }
 152     }
 153
 154     @Override
 155     public void deletePolicy(String policyInstanceId) throws RestClientException {
 156         String url = baseUrl() + "/policy?instance={instance}";
 157         Map<String, ?> uriVariables = Map.of("instance", policyInstanceId);
 158         this.restTemplate.delete(url, uriVariables);
 159     }
 160
 161     @Value.Immutable
 162     @Gson.TypeAdapters
 163     interface RicInfo {
 164         public String name();
 165
 166         public Collection<String> nodeNames();
 167
 168         public Collection<String> policyTypes();
 169     }
 170
 171     @Override
 172     public ResponseEntity<Collection<String>> getRicsSupportingType(String typeName) {
 173         String url = baseUrl() + "/rics?policyType={typeName}";
 174         Map<String, ?> uriVariables = Map.of("typeName", typeName);
 175         String rsp = this.restTemplate.getForObject(url, String.class, uriVariables);
 176
 177         Type listType = new TypeToken<List<ImmutableRicInfo>>() {
 178         }.getType();
 179         List<RicInfo> rspParsed = gson.fromJson(rsp, listType);
 180
 181         Collection<String> result = new Vector<>(rspParsed.size());
 182         for (RicInfo ric : rspParsed) {
 183             result.add(ric.name());
 184         }
 185         return new ResponseEntity<>(result, HttpStatus.OK);
 186     }
 187
 188 }