resource-assignment/provider/src/main/java/org/openecomp/sdnc/rm/dao/jdbc/ResourceDaoImpl.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.rm.dao.jdbc;
  23
  24 import java.util.ArrayList;
  25 import java.util.HashSet;
  26 import java.util.List;
  27
  28 import org.openecomp.sdnc.rm.dao.ResourceDao;
  29 import org.openecomp.sdnc.rm.data.LabelAllocationItem;
  30 import org.openecomp.sdnc.rm.data.LabelResource;
  31 import org.openecomp.sdnc.rm.data.LimitAllocationItem;
  32 import org.openecomp.sdnc.rm.data.LimitResource;
  33 import org.openecomp.sdnc.rm.data.RangeAllocationItem;
  34 import org.openecomp.sdnc.rm.data.RangeResource;
  35 import org.openecomp.sdnc.rm.data.ResourceKey;
  36 import org.openecomp.sdnc.rm.data.ResourceType;
  37 import org.openecomp.sdnc.util.str.StrUtil;
  38
  39 public class ResourceDaoImpl implements ResourceDao {
  40
  41         private ResourceJdbcDao resourceJdbcDao;
  42         private ResourceLoadJdbcDao resourceLoadJdbcDao;
  43         private AllocationItemJdbcDao allocationItemJdbcDao;
  44
  45         @Override
  46         public org.openecomp.sdnc.rm.data.Resource getResource(String assetId, String resourceName) {
  47                 Resource rEntity = resourceJdbcDao.getResource(assetId, resourceName);
  48                 org.openecomp.sdnc.rm.data.Resource r = createResource(rEntity);
  49
  50                 if (r != null) {
  51                         List<AllocationItem> aiEntityList = allocationItemJdbcDao.getAllocationItems(rEntity.id);
  52                         r.allocationItems = new ArrayList<org.openecomp.sdnc.rm.data.AllocationItem>();
  53                         for (AllocationItem aiEntity : aiEntityList) {
  54                                 org.openecomp.sdnc.rm.data.AllocationItem ai = createAllocationItem(r, aiEntity);
  55                                 r.allocationItems.add(ai);
  56                         }
  57
  58                         List<ResourceLoad> rlEntityList = resourceLoadJdbcDao.getResourceLoads(rEntity.id);
  59                         r.resourceLoadList = new ArrayList<org.openecomp.sdnc.rm.data.ResourceLoad>();
  60                         for (ResourceLoad rlEntity : rlEntityList) {
  61                                 org.openecomp.sdnc.rm.data.ResourceLoad rl = createResourceLoad(r, rlEntity);
  62                                 r.resourceLoadList.add(rl);
  63                         }
  64                 }
  65
  66                 return r;
  67         }
  68
  69         @Override
  70         public void saveResource(org.openecomp.sdnc.rm.data.Resource resource) {
  71                 if (resource == null)
  72                         return;
  73
  74                 org.openecomp.sdnc.rm.dao.jdbc.Resource resourceEntity =
  75                         resourceJdbcDao.getResource(resource.resourceKey.assetId, resource.resourceKey.resourceName);
  76                 if (resourceEntity == null) {
  77                         resourceEntity = createResourceEntity(resource);
  78                         resourceJdbcDao.add(resourceEntity);
  79                         if (resource.allocationItems != null)
  80                                 for (org.openecomp.sdnc.rm.data.AllocationItem ai : resource.allocationItems) {
  81                                         AllocationItem aiEntity = createAllocationItemEntity(resourceEntity.id, ai);
  82                                         allocationItemJdbcDao.add(aiEntity);
  83                                 }
  84                         if (resource.resourceLoadList != null)
  85                                 for (org.openecomp.sdnc.rm.data.ResourceLoad rl : resource.resourceLoadList) {
  86                                         ResourceLoad rlEntity = createResourceLoadEntity(resourceEntity.id, rl);
  87                                         resourceLoadJdbcDao.add(rlEntity);
  88                                 }
  89                 } else {
  90                         updateResourceEntity(resourceEntity, resource);
  91                         resourceJdbcDao.update(resourceEntity);
  92
  93                         List<AllocationItem> oldAiEntityList = allocationItemJdbcDao.getAllocationItems(resourceEntity.id);
  94                         if (resource.allocationItems != null)
  95                                 for (org.openecomp.sdnc.rm.data.AllocationItem newai : resource.allocationItems) {
  96                                         AllocationItem foundAiEntity = null;
  97                                         for (AllocationItem oldAiEntity : oldAiEntityList)
  98                                                 if (oldAiEntity.resourceSetId.equals(newai.resourceSetId)) {
  99                                                         foundAiEntity = oldAiEntity;
 100                                                         break;
 101                                                 }
 102                                         if (foundAiEntity != null) {
 103                                                 updateAllocationItemEntity(foundAiEntity, newai);
 104                                                 allocationItemJdbcDao.update(foundAiEntity);
 105                                         } else {
 106                                                 AllocationItem newAiEntity = createAllocationItemEntity(resourceEntity.id, newai);
 107                                                 allocationItemJdbcDao.add(newAiEntity);
 108                                         }
 109                                 }
 110                         for (AllocationItem oldAiEntity : oldAiEntityList) {
 111                                 boolean found = false;
 112                                 if (resource.allocationItems != null)
 113                                         for (org.openecomp.sdnc.rm.data.AllocationItem newai : resource.allocationItems)
 114                                                 if (oldAiEntity.resourceSetId.equals(newai.resourceSetId)) {
 115                                                         found = true;
 116                                                         break;
 117                                                 }
 118                                 if (!found)
 119                                         allocationItemJdbcDao.delete(oldAiEntity.id);
 120                         }
 121
 122                         List<ResourceLoad> oldRlEntityList = resourceLoadJdbcDao.getResourceLoads(resourceEntity.id);
 123                         if (resource.resourceLoadList != null)
 124                                 for (org.openecomp.sdnc.rm.data.ResourceLoad newrl : resource.resourceLoadList) {
 125                                         ResourceLoad foundRlEntity = null;
 126                                         for (ResourceLoad oldRlEntity : oldRlEntityList)
 127                                                 if (oldRlEntity.applicationId.equals(newrl.applicationId)) {
 128                                                         foundRlEntity = oldRlEntity;
 129                                                         break;
 130                                                 }
 131                                         if (foundRlEntity != null) {
 132                                                 updateResourceLoadEntity(foundRlEntity, newrl);
 133                                                 resourceLoadJdbcDao.update(foundRlEntity);
 134                                         } else {
 135                                                 ResourceLoad newRlEntity = createResourceLoadEntity(resourceEntity.id, newrl);
 136                                                 resourceLoadJdbcDao.add(newRlEntity);
 137                                         }
 138                                 }
 139                         for (ResourceLoad oldRlEntity : oldRlEntityList) {
 140                                 boolean found = false;
 141                                 if (resource.resourceLoadList != null)
 142                                         for (org.openecomp.sdnc.rm.data.ResourceLoad newrl : resource.resourceLoadList)
 143                                                 if (oldRlEntity.applicationId.equals(newrl.applicationId)) {
 144                                                         found = true;
 145                                                         break;
 146                                                 }
 147                                 if (!found)
 148                                         resourceLoadJdbcDao.delete(oldRlEntity.id);
 149                         }
 150                 }
 151         }
 152
 153         @Override
 154         public void deleteResource(String assetId, String resourceName) {
 155                 org.openecomp.sdnc.rm.dao.jdbc.Resource resourceEntity = resourceJdbcDao.getResource(assetId, resourceName);
 156                 if (resourceEntity != null)
 157                         resourceJdbcDao.delete(resourceEntity.id);
 158         }
 159
 160         @Override
 161         public List<org.openecomp.sdnc.rm.data.Resource> getResourceSet(String resourceSetId) {
 162                 List<Resource> rEntityList = resourceJdbcDao.getResourceSet(resourceSetId);
 163                 List<org.openecomp.sdnc.rm.data.Resource> rlist = new ArrayList<org.openecomp.sdnc.rm.data.Resource>();
 164                 for (Resource rEntity : rEntityList) {
 165                         org.openecomp.sdnc.rm.data.Resource r = createResource(rEntity);
 166                         rlist.add(r);
 167
 168                         List<AllocationItem> aiEntityList = allocationItemJdbcDao.getAllocationItems(rEntity.id);
 169                         r.allocationItems = new ArrayList<org.openecomp.sdnc.rm.data.AllocationItem>();
 170                         for (AllocationItem aiEntity : aiEntityList) {
 171                                 org.openecomp.sdnc.rm.data.AllocationItem ai = createAllocationItem(r, aiEntity);
 172                                 r.allocationItems.add(ai);
 173                         }
 174
 175                         List<ResourceLoad> rlEntityList = resourceLoadJdbcDao.getResourceLoads(rEntity.id);
 176                         r.resourceLoadList = new ArrayList<org.openecomp.sdnc.rm.data.ResourceLoad>();
 177                         for (ResourceLoad rlEntity : rlEntityList) {
 178                                 org.openecomp.sdnc.rm.data.ResourceLoad rl = createResourceLoad(r, rlEntity);
 179                                 r.resourceLoadList.add(rl);
 180                         }
 181                 }
 182                 return rlist;
 183         }
 184
 185         @Override
 186         public List<org.openecomp.sdnc.rm.data.Resource> getResourceUnion(String resourceUnionId) {
 187                 List<Resource> rEntityList = resourceJdbcDao.getResourceUnion(resourceUnionId);
 188                 List<org.openecomp.sdnc.rm.data.Resource> rlist = new ArrayList<org.openecomp.sdnc.rm.data.Resource>();
 189                 for (Resource rEntity : rEntityList) {
 190                         org.openecomp.sdnc.rm.data.Resource r = createResource(rEntity);
 191                         rlist.add(r);
 192
 193                         List<AllocationItem> aiEntityList = allocationItemJdbcDao.getAllocationItems(rEntity.id);
 194                         r.allocationItems = new ArrayList<org.openecomp.sdnc.rm.data.AllocationItem>();
 195                         for (AllocationItem aiEntity : aiEntityList) {
 196                                 org.openecomp.sdnc.rm.data.AllocationItem ai = createAllocationItem(r, aiEntity);
 197                                 r.allocationItems.add(ai);
 198                         }
 199
 200                         List<ResourceLoad> rlEntityList = resourceLoadJdbcDao.getResourceLoads(rEntity.id);
 201                         r.resourceLoadList = new ArrayList<org.openecomp.sdnc.rm.data.ResourceLoad>();
 202                         for (ResourceLoad rlEntity : rlEntityList) {
 203                                 org.openecomp.sdnc.rm.data.ResourceLoad rl = createResourceLoad(r, rlEntity);
 204                                 r.resourceLoadList.add(rl);
 205                         }
 206                 }
 207                 return rlist;
 208         }
 209
 210         private Resource createResourceEntity(org.openecomp.sdnc.rm.data.Resource resource) {
 211                 Resource resourceEntity = new Resource();
 212                 resourceEntity.assetId = resource.resourceKey.assetId;
 213                 resourceEntity.name = resource.resourceKey.resourceName;
 214                 resourceEntity.type = resource.resourceType.toString();
 215                 if (resource.resourceType == ResourceType.Limit)
 216                         resourceEntity.ltUsed = ((LimitResource) resource).used;
 217                 else if (resource.resourceType == ResourceType.Label) {
 218                         resourceEntity.llLabel = ((LabelResource) resource).label;
 219                         resourceEntity.llReferenceCount = ((LabelResource) resource).referenceCount;
 220                 } else if (resource.resourceType == ResourceType.Range)
 221                         resourceEntity.rrUsed = StrUtil.listInt(((RangeResource) resource).used);
 222
 223                 return resourceEntity;
 224         }
 225
 226         private ResourceLoad createResourceLoadEntity(long resourceId, org.openecomp.sdnc.rm.data.ResourceLoad rl) {
 227                 ResourceLoad rlEntity = new ResourceLoad();
 228                 rlEntity.resourceId = resourceId;
 229                 rlEntity.applicationId = rl.applicationId;
 230                 rlEntity.loadTime = rl.resourceLoadTime;
 231                 rlEntity.expirationTime = rl.resourceExpirationTime;
 232                 return rlEntity;
 233         }
 234
 235         private void updateResourceLoadEntity(ResourceLoad rlEntity, org.openecomp.sdnc.rm.data.ResourceLoad rl) {
 236                 rlEntity.loadTime = rl.resourceLoadTime;
 237                 rlEntity.expirationTime = rl.resourceExpirationTime;
 238         }
 239
 240         private AllocationItem createAllocationItemEntity(long resourceId, org.openecomp.sdnc.rm.data.AllocationItem ai) {
 241                 AllocationItem aiEntity = new AllocationItem();
 242                 aiEntity.resourceId = resourceId;
 243                 aiEntity.resourceSetId = ai.resourceSetId;
 244                 aiEntity.resourceUnionId = ai.resourceUnionId;
 245                 aiEntity.resourceShareGroupList = StrUtil.listStr(ai.resourceShareGroupList);
 246                 aiEntity.applicationId = ai.applicationId;
 247                 aiEntity.allocationTime = ai.allocationTime;
 248                 if (ai.resourceType == ResourceType.Limit)
 249                         aiEntity.ltUsed = ((LimitAllocationItem) ai).used;
 250                 else if (ai.resourceType == ResourceType.Label)
 251                         aiEntity.llLabel = ((LabelAllocationItem) ai).label;
 252                 else if (ai.resourceType == ResourceType.Range)
 253                         aiEntity.rrUsed = StrUtil.listInt(((RangeAllocationItem) ai).used);
 254                 return aiEntity;
 255         }
 256
 257         private void updateAllocationItemEntity(AllocationItem aiEntity, org.openecomp.sdnc.rm.data.AllocationItem ai) {
 258                 aiEntity.resourceShareGroupList = StrUtil.listStr(ai.resourceShareGroupList);
 259                 aiEntity.allocationTime = ai.allocationTime;
 260                 if (ai.resourceType == ResourceType.Limit)
 261                         aiEntity.ltUsed = ((LimitAllocationItem) ai).used;
 262                 else if (ai.resourceType == ResourceType.Label)
 263                         aiEntity.llLabel = ((LabelAllocationItem) ai).label;
 264                 else if (ai.resourceType == ResourceType.Range)
 265                         aiEntity.rrUsed = StrUtil.listInt(((RangeAllocationItem) ai).used);
 266         }
 267
 268         private void updateResourceEntity(Resource resourceEntity, org.openecomp.sdnc.rm.data.Resource resource) {
 269                 if (resource.resourceType == ResourceType.Limit)
 270                         resourceEntity.ltUsed = ((LimitResource) resource).used;
 271                 else if (resource.resourceType == ResourceType.Label) {
 272                         resourceEntity.llLabel = ((LabelResource) resource).label;
 273                         resourceEntity.llReferenceCount = ((LabelResource) resource).referenceCount;
 274                 } else if (resource.resourceType == ResourceType.Range)
 275                         resourceEntity.rrUsed = StrUtil.listInt(((RangeResource) resource).used);
 276         }
 277
 278         private org.openecomp.sdnc.rm.data.Resource createResource(Resource resourceEntity) {
 279                 if (resourceEntity == null)
 280                         return null;
 281
 282                 org.openecomp.sdnc.rm.data.Resource r = null;
 283                 ResourceType type = ResourceType.valueOf(resourceEntity.type);
 284                 if (type == ResourceType.Limit) {
 285                         LimitResource l = new LimitResource();
 286                         l.used = resourceEntity.ltUsed;
 287                         r = l;
 288                 } else if (type == ResourceType.Label) {
 289                         LabelResource l = new LabelResource();
 290                         l.label = resourceEntity.llLabel;
 291                         l.referenceCount = resourceEntity.llReferenceCount;
 292                         r = l;
 293                 } else if (type == ResourceType.Range) {
 294                         RangeResource rr = new RangeResource();
 295                         rr.used =
 296                                 StrUtil.listInt(resourceEntity.rrUsed, "Invalid data found in DB in for Resource Id: " +
 297                                         resourceEntity.id + ": RESOURCE.RR_USED: " + resourceEntity.rrUsed);
 298                         r = rr;
 299                 }
 300
 301                 r.resourceType = type;
 302                 r.resourceKey = new ResourceKey();
 303                 r.resourceKey.assetId = resourceEntity.assetId;
 304                 r.resourceKey.resourceName = resourceEntity.name;
 305
 306                 return r;
 307         }
 308
 309         private org.openecomp.sdnc.rm.data.AllocationItem createAllocationItem(
 310                 org.openecomp.sdnc.rm.data.Resource r,
 311                 AllocationItem aiEntity) {
 312                 if (r == null || aiEntity == null)
 313                         return null;
 314
 315                 org.openecomp.sdnc.rm.data.AllocationItem ai = null;
 316                 if (r.resourceType == ResourceType.Limit) {
 317                         LimitAllocationItem lai = new LimitAllocationItem();
 318                         lai.used = aiEntity.ltUsed;
 319                         ai = lai;
 320                 } else if (r.resourceType == ResourceType.Label) {
 321                         LabelAllocationItem lai = new LabelAllocationItem();
 322                         lai.label = aiEntity.llLabel;
 323                         ai = lai;
 324                 } else if (r.resourceType == ResourceType.Range) {
 325                         RangeAllocationItem rai = new RangeAllocationItem();
 326                         rai.used =
 327                                 StrUtil.listInt(aiEntity.rrUsed, "Invalid data found in DB in for Allocation Item Id: " +
 328                                         aiEntity.id + ": ALLOCATION_ITEM.RR_USED: " + aiEntity.rrUsed);
 329                         ai = rai;
 330                 }
 331
 332                 ai.resourceType = r.resourceType;
 333                 ai.resourceKey = r.resourceKey;
 334                 ai.resourceSetId = aiEntity.resourceSetId;
 335                 ai.resourceUnionId = aiEntity.resourceUnionId;
 336                 if (aiEntity.resourceShareGroupList != null)
 337                         ai.resourceShareGroupList = new HashSet<String>(StrUtil.listStr(aiEntity.resourceShareGroupList));
 338                 ai.applicationId = aiEntity.applicationId;
 339                 ai.allocationTime = aiEntity.allocationTime;
 340
 341                 return ai;
 342         }
 343
 344         private org.openecomp.sdnc.rm.data.ResourceLoad createResourceLoad(
 345                 org.openecomp.sdnc.rm.data.Resource r,
 346                 ResourceLoad rlEntity) {
 347                 if (rlEntity == null)
 348                         return null;
 349
 350                 org.openecomp.sdnc.rm.data.ResourceLoad rl = new org.openecomp.sdnc.rm.data.ResourceLoad();
 351                 rl.resourceKey = r.resourceKey;
 352                 rl.applicationId = rlEntity.applicationId;
 353                 rl.resourceLoadTime = rlEntity.loadTime;
 354                 rl.resourceExpirationTime = rlEntity.expirationTime;
 355
 356                 return rl;
 357         }
 358
 359         public void setResourceJdbcDao(ResourceJdbcDao resourceJdbcDao) {
 360                 this.resourceJdbcDao = resourceJdbcDao;
 361         }
 362
 363         public void setResourceLoadJdbcDao(ResourceLoadJdbcDao resourceLoadJdbcDao) {
 364                 this.resourceLoadJdbcDao = resourceLoadJdbcDao;
 365         }
 366
 367         public void setAllocationItemJdbcDao(AllocationItemJdbcDao allocationItemJdbcDao) {
 368                 this.allocationItemJdbcDao = allocationItemJdbcDao;
 369         }
 370 }