netbox-client/provider/src/main/java/org/onap/ccsdk/sli/adaptors/netbox/impl/NetboxClientImpl.java

   1 /*
   2  * Copyright (C) 2018 Bell Canada.
   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 package org.onap.ccsdk.sli.adaptors.netbox.impl;
  17
  18 import com.google.common.collect.Lists;
  19 import com.google.gson.JsonSyntaxException;
  20 import java.io.IOException;
  21 import java.sql.SQLException;
  22 import java.util.ArrayList;
  23 import java.util.Map;
  24 import org.apache.http.HttpResponse;
  25 import org.apache.http.util.EntityUtils;
  26 import org.onap.ccsdk.sli.adaptors.netbox.api.NetboxClient;
  27 import org.onap.ccsdk.sli.adaptors.netbox.model.IPAddress;
  28 import org.onap.ccsdk.sli.adaptors.netbox.model.IPStatus;
  29 import org.onap.ccsdk.sli.core.dblib.DbLibService;
  30 import org.onap.ccsdk.sli.core.sli.SvcLogicContext;
  31 import org.onap.ccsdk.sli.core.sli.SvcLogicException;
  32 import org.onap.ccsdk.sli.core.sli.SvcLogicResource.QueryStatus;
  33 import org.onap.ccsdk.sli.core.slipluginutils.SliPluginUtils;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 public class NetboxClientImpl implements NetboxClient {
  38
  39     private static final Logger LOG = LoggerFactory.getLogger(NetboxClientImpl.class);
  40
  41     private static final String NEXT_AVAILABLE_IP_IN_PREFIX_PATH = "/api/ipam/prefixes/%s/available-ips/";
  42     private static final String IP_ADDRESS_PATH = "/api/ipam/ip-addresses/%s/";
  43     private static final String EMPTY_STRING = "";
  44     private static final String SERVICE_INSTANCE_ID_PROP = "service_instance_id";
  45     private static final String VF_MODULE_ID_PROP = "vf_module_id";
  46
  47     private static final String ASSIGN_IP_SQL_STATEMENT =
  48         "INSERT INTO IPAM_IP_ASSIGNEMENT (service_instance_id, vf_module_id, prefix_id, ip_address_id, ip_address, ip_status, ip_response_json) \n"
  49             + "VALUES (?, ?, ?, ?, ?, ?, ?)";
  50     private static final String UNASSIGN_IP_SQL_STATEMENT =
  51         "UPDATE IPAM_IP_ASSIGNEMENT SET ip_status = ? WHERE service_instance_id = ? AND vf_module_id = ? AND ip_address_id = ?";
  52
  53     private final NetboxHttpClient client;
  54
  55     private final DbLibService dbLibService;
  56
  57     public NetboxClientImpl(final NetboxHttpClient client, final DbLibService dbLibService) {
  58         this.client = client;
  59         this.dbLibService = dbLibService;
  60     }
  61
  62     @Override
  63     public QueryStatus assignIpAddress(final Map<String, String> parameters, final SvcLogicContext ctx) {
  64
  65         try {
  66             SliPluginUtils
  67                 .checkParameters(parameters, new String[]{SERVICE_INSTANCE_ID_PROP, VF_MODULE_ID_PROP, "prefix_id"},
  68                     LOG);
  69         } catch (SvcLogicException e) {
  70             return QueryStatus.FAILURE;
  71         }
  72
  73         final String serviceInstanceId = parameters.get(SERVICE_INSTANCE_ID_PROP);
  74         LOG.trace("assignIpAddress: service_instance_id = {}", serviceInstanceId);
  75         final String vfModuleId = parameters.get(VF_MODULE_ID_PROP);
  76         LOG.trace("assignIpAddress: vf_module_id = {}", vfModuleId);
  77         final String prefixId = parameters.get("prefix_id");
  78         LOG.trace("assignIpAddress: prefix_id = {}", prefixId);
  79
  80         HttpResponse httpResp;
  81         try {
  82             httpResp = client
  83                 .post(String.format(NEXT_AVAILABLE_IP_IN_PREFIX_PATH, prefixId), EMPTY_STRING);
  84         } catch (IOException e) {
  85             LOG.error("Fail to assign IP for Prefix(id={}). {}", prefixId, e.getMessage(), e.getCause());
  86             return QueryStatus.FAILURE;
  87         }
  88
  89         String ipamRespJson;
  90         try {
  91             ipamRespJson = EntityUtils.toString(httpResp.getEntity(), "UTF-8");
  92         } catch (IOException e) {
  93             LOG.error("Fail to parse IPAM response for assign in Prefix(id={}). Response={}", prefixId,
  94                 httpResp.getEntity(), e);
  95             return QueryStatus.FAILURE;
  96         }
  97
  98         if (httpResp.getStatusLine().getStatusCode() != 201) {
  99             LOG.error("Fail to assign IP for Prefix(id={}). HTTP code 201!={}. Response={}", prefixId,
 100                 httpResp.getStatusLine().getStatusCode(), ipamRespJson);
 101             return QueryStatus.FAILURE;
 102         }
 103
 104         IPAddress ipAddress;
 105         try {
 106             ipAddress = IPAddress.fromJson(ipamRespJson);
 107         } catch (JsonSyntaxException e) {
 108             LOG.error("Fail to parse IPAM JSON reponse to IPAddress POJO. IPAM JSON Response={}", ipamRespJson, e);
 109             return QueryStatus.FAILURE;
 110         }
 111
 112         ArrayList<String> args = Lists.newArrayList(serviceInstanceId,
 113             vfModuleId,
 114             String.valueOf(prefixId),
 115             String.valueOf(ipAddress.getId()),
 116             ipAddress.getAddress(),
 117             IPStatus.ASSIGNED.name(),
 118             ipamRespJson);
 119
 120         try {
 121             dbLibService.writeData(ASSIGN_IP_SQL_STATEMENT, args, null);
 122         } catch (SQLException e) {
 123             LOG.error("Caught SQL exception", e);
 124             return QueryStatus.FAILURE;
 125         }
 126
 127         ctx.setAttribute("self_serve_netbox_ip_assignement.ip-address", ipAddress.getAddress());
 128
 129         return QueryStatus.SUCCESS;
 130     }
 131
 132     @Override
 133     public QueryStatus unassignIpAddress(final Map<String, String> parameters, final SvcLogicContext ctx) {
 134         try {
 135             SliPluginUtils
 136                 .checkParameters(parameters, new String[]{SERVICE_INSTANCE_ID_PROP, VF_MODULE_ID_PROP, "ip_address_id"},
 137                     LOG);
 138         } catch (SvcLogicException e) {
 139             return QueryStatus.FAILURE;
 140         }
 141
 142         final String serviceInstanceId = parameters.get(SERVICE_INSTANCE_ID_PROP);
 143         LOG.trace("assignIpAddress: service_instance_id = {}", serviceInstanceId);
 144         final String vfModuleId = parameters.get(VF_MODULE_ID_PROP);
 145         LOG.trace("assignIpAddress: vf_module_id = {}", vfModuleId);
 146         final String ipAddressId = parameters.get("ip_address_id");
 147         LOG.trace("assignIpAddress: ip_address_id = {}", ipAddressId);
 148         HttpResponse httpResp;
 149         try {
 150             httpResp = client.delete(String.format(IP_ADDRESS_PATH, ipAddressId));
 151         } catch (IOException e) {
 152             LOG.error("Fail to unassign IP for IPAddress(id= " + ipAddressId + "). " + e.getMessage(),
 153                 e.getCause());
 154             return QueryStatus.FAILURE;
 155         }
 156
 157         if (httpResp.getStatusLine().getStatusCode() - 200 >= 100) {
 158             LOG.error("Fail to unassign IP for IPAddress(id={}). HTTP code={}.", ipAddressId,
 159                 httpResp.getStatusLine().getStatusCode());
 160             return QueryStatus.FAILURE;
 161         }
 162
 163         ArrayList<String> args = Lists.newArrayList(
 164             IPStatus.UNASSIGNED.name(),
 165             serviceInstanceId,
 166             vfModuleId,
 167             String.valueOf(ipAddressId));
 168         try {
 169             dbLibService.writeData(UNASSIGN_IP_SQL_STATEMENT, args, null);
 170         } catch (SQLException e) {
 171             LOG.error("Caught SQL exception", e);
 172             return QueryStatus.FAILURE;
 173         }
 174
 175         return QueryStatus.SUCCESS;
 176     }
 177 }