resource-assignment/provider/src/main/java/org/openecomp/sdnc/ra/comp/EndPointAllocatorImpl.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * openECOMP : SDN-C
   4  * ================================================================================
   5  * Copyright (C) 2017 ONAP Intellectual Property. All rights
   6  * reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *      http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.openecomp.sdnc.ra.comp;
  23
  24 import java.util.ArrayList;
  25 import java.util.Collections;
  26 import java.util.HashMap;
  27 import java.util.List;
  28 import java.util.Map;
  29
  30 import org.apache.commons.lang.NotImplementedException;
  31 import org.openecomp.sdnc.ra.equip.data.EquipmentData;
  32 import org.openecomp.sdnc.rm.comp.ResourceManager;
  33 import org.openecomp.sdnc.rm.data.AllocationItem;
  34 import org.openecomp.sdnc.rm.data.AllocationOutcome;
  35 import org.openecomp.sdnc.rm.data.AllocationRequest;
  36 import org.openecomp.sdnc.rm.data.AllocationStatus;
  37 import org.openecomp.sdnc.rm.data.LimitAllocationItem;
  38 import org.openecomp.sdnc.rm.data.LimitResource;
  39 import org.openecomp.sdnc.rm.data.RangeAllocationItem;
  40 import org.openecomp.sdnc.rm.data.RangeResource;
  41 import org.openecomp.sdnc.rm.data.Resource;
  42 import org.slf4j.Logger;
  43 import org.slf4j.LoggerFactory;
  44
  45 public class EndPointAllocatorImpl implements EndPointAllocator {
  46
  47     private static final Logger log = LoggerFactory.getLogger(EndPointAllocatorImpl.class);
  48
  49     private Map<String, List<EndPointAllocationDefinition>> endPointAllocationDefinitionMap;
  50
  51     private ResourceManager resourceManager;
  52
  53     @Override
  54     public List<EndPointData> allocateEndPoints(
  55             ServiceData serviceData,
  56             Map<String, Object> equipmentConstraints,
  57             boolean checkOnly,
  58             boolean change,
  59             int changeNumber) {
  60         List<EndPointAllocationDefinition> defList = endPointAllocationDefinitionMap.get(serviceData.serviceModel);
  61         if (defList == null)
  62             throw new NotImplementedException("Service model: " + serviceData.serviceModel + " not supported");
  63
  64         List<EndPointData> epList = new ArrayList<>();
  65         for (EndPointAllocationDefinition def : defList) {
  66             if (serviceData.endPointPosition != null && !serviceData.endPointPosition.equals(def.endPointPosition))
  67                 continue;
  68
  69             log.info(
  70                     "Starting allocation of end point: " + def.endPointPosition + ": " + serviceData.serviceInstanceId);
  71
  72             String resourceUnionId = serviceData.serviceInstanceId + '/' + def.endPointPosition;
  73             String resourceSetId = resourceUnionId + '/' + changeNumber;
  74
  75             String equipmentId = (String) equipmentConstraints.get("equipment-id");
  76             if (equipmentId == null) {
  77                 EndPointData epExisting = readEndPoint(resourceUnionId, resourceSetId);
  78                 if (epExisting != null && epExisting.equipmentId != null) {
  79                     equipmentConstraints.put("equipment-id", epExisting.equipmentId);
  80
  81                     log.info("Trying assignment on the current equipment: " + epExisting.equipmentId);
  82                 }
  83             }
  84
  85             List<EquipmentData> equipList = def.equipmentReader.readEquipment(equipmentConstraints);
  86             if (equipList == null || equipList.isEmpty()) {
  87                 log.info("Equipment not found for " + def.endPointPosition);
  88                 break;
  89             }
  90
  91             if (def.equipmentCheckList != null) {
  92                 for (EquipmentCheck filter : def.equipmentCheckList) {
  93                     List<EquipmentData> newEquipList = new ArrayList<>();
  94                     for (EquipmentData equipData : equipList)
  95                         if (filter.checkEquipment(def.endPointPosition, serviceData, equipData, equipmentConstraints))
  96                             newEquipList.add(equipData);
  97                     equipList = newEquipList;
  98                 }
  99                 if (equipList.isEmpty()) {
 100                     log.info("No equipment meets the requiremets for the service for: " + def.endPointPosition);
 101                     break;
 102                 }
 103             }
 104
 105             if (equipList.size() > 1 && def.preferenceRuleList != null && !def.preferenceRuleList.isEmpty()) {
 106
 107                 List<PrefEquipment> prefEquipList = new ArrayList<>();
 108                 for (EquipmentData equipData : equipList) {
 109                     PrefEquipment prefEquip = new PrefEquipment();
 110                     prefEquip.equipData = equipData;
 111                     prefEquip.prefNumbers = new long[def.preferenceRuleList.size()];
 112                     prefEquipList.add(prefEquip);
 113
 114                     int i = 0;
 115                     for (PreferenceRule prefRule : def.preferenceRuleList)
 116                         prefEquip.prefNumbers[i++] =
 117                                 prefRule.assignOrderNumber(def.endPointPosition, serviceData, equipData);
 118                 }
 119
 120                 Collections.sort(prefEquipList);
 121
 122                 equipList = new ArrayList<>();
 123                 for (PrefEquipment prefEquip : prefEquipList)
 124                     equipList.add(prefEquip.equipData);
 125             }
 126
 127             for (EquipmentData equipData : equipList) {
 128                 boolean allgood = true;
 129                 if (def.allocationRuleList != null)
 130                     for (AllocationRule allocationRule : def.allocationRuleList) {
 131                         AllocationRequest ar = allocationRule.buildAllocationRequest(resourceUnionId, resourceSetId,
 132                                 def.endPointPosition, serviceData, equipData, checkOnly, change);
 133                         if (ar != null) {
 134                             AllocationOutcome ao = resourceManager.allocateResources(ar);
 135                             if (ao.status != AllocationStatus.Success) {
 136                                 allgood = false;
 137                                 break;
 138                             }
 139                         }
 140                     }
 141                 if (allgood) {
 142                     EndPointData ep = readEndPoint(resourceUnionId, resourceSetId);
 143                     epList.add(ep);
 144                     break;
 145                 }
 146             }
 147         }
 148
 149         return epList;
 150     }
 151
 152     private EndPointData readEndPoint(String resourceUnionId, String resourceSetId) {
 153         EndPointData ep = new EndPointData();
 154         ep.resourceUnionId = resourceUnionId;
 155         ep.resourceSetId = resourceSetId;
 156
 157         int i1 = resourceUnionId.indexOf('/');
 158         if (i1 > 0)
 159             ep.endPointPosition = resourceUnionId.substring(i1 + 1);
 160
 161         ep.data = new HashMap<>();
 162
 163         List<Resource> rlist = resourceManager.getResourceUnion(resourceUnionId);
 164         for (Resource r : rlist) {
 165             if (r instanceof RangeResource) {
 166                 RangeResource rr = (RangeResource) r;
 167                 for (AllocationItem ai : r.allocationItems)
 168                     if (ai.resourceUnionId.equals(resourceUnionId)) {
 169                         RangeAllocationItem rai = (RangeAllocationItem) ai;
 170                         ep.data.put(ep.endPointPosition + '.' + rr.resourceKey.resourceName, rai.used.first());
 171                     }
 172             }
 173             if (r instanceof LimitResource) {
 174                 LimitResource rr = (LimitResource) r;
 175                 for (AllocationItem ai : r.allocationItems)
 176                     if (ai.resourceUnionId.equals(resourceUnionId)) {
 177                         LimitAllocationItem rai = (LimitAllocationItem) ai;
 178                         ep.data.put(ep.endPointPosition + '.' + rr.resourceKey.resourceName + ".allocated", rai.used);
 179                         ep.data.put(ep.endPointPosition + '.' + rr.resourceKey.resourceName + ".used", rr.used);
 180                         ep.data.put(ep.endPointPosition + '.' + rr.resourceKey.resourceName + ".assetId",
 181                                 r.resourceKey.assetId);
 182                     }
 183             }
 184         }
 185
 186         return ep;
 187     }
 188
 189     private static class PrefEquipment implements Comparable<PrefEquipment> {
 190
 191         public long[] prefNumbers;
 192         public EquipmentData equipData;
 193
 194         @Override
 195         public int compareTo(PrefEquipment o) {
 196             for (int i = 0; i < prefNumbers.length; i++) {
 197                 if (prefNumbers[i] < o.prefNumbers[i])
 198                     return -1;
 199                 if (prefNumbers[i] > o.prefNumbers[i])
 200                     return 1;
 201             }
 202             return 0;
 203         }
 204     }
 205
 206     public void setEndPointAllocationDefinitionMap(
 207             Map<String, List<EndPointAllocationDefinition>> endPointAllocationDefinitionMap) {
 208         this.endPointAllocationDefinitionMap = endPointAllocationDefinitionMap;
 209     }
 210
 211     public void setResourceManager(ResourceManager resourceManager) {
 212         this.resourceManager = resourceManager;
 213     }
 214 }