83da270711ed29b0971ed014f02858910863e93e
[ccsdk/sli.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * openECOMP : SDN-C
4  * ================================================================================
5  * Copyright (C) 2017 AT&T 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.onap.ccsdk.sli.adaptors.rm.comp;
23
24 import java.util.HashSet;
25 import java.util.List;
26 import java.util.Set;
27 import org.onap.ccsdk.sli.adaptors.lock.comp.LockHelper;
28 import org.onap.ccsdk.sli.adaptors.rm.dao.ResourceDao;
29 import org.onap.ccsdk.sli.adaptors.rm.data.AllocationOutcome;
30 import org.onap.ccsdk.sli.adaptors.rm.data.AllocationRequest;
31 import org.onap.ccsdk.sli.adaptors.rm.data.ReleaseRequest;
32 import org.onap.ccsdk.sli.adaptors.rm.data.Resource;
33 import org.onap.ccsdk.sli.adaptors.rm.util.ResourceUtil;
34 import org.onap.ccsdk.sli.adaptors.util.str.StrUtil;
35 import org.slf4j.Logger;
36 import org.slf4j.LoggerFactory;
37
38 public class ResourceManagerImpl implements ResourceManager {
39
40     private static final Logger log = LoggerFactory.getLogger(ResourceManagerImpl.class);
41
42     private LockHelper lockHelper;
43     private ResourceDao resourceDao;
44
45     private int lockTimeout = 10 * 60; // Default 10 min
46
47     public ResourceManagerImpl() {
48         log.info("ResourceManager created.");
49     }
50
51     @Override
52     public Resource getResource(String resourceName, String assetId) {
53         Resource r = resourceDao.getResource(assetId, resourceName);
54         ResourceUtil.recalculate(r);
55         return r;
56     }
57
58     @Override
59     public List<Resource> getResourceUnion(String resourceUnionId) {
60         List<Resource> rlist = resourceDao.getResourceUnion(resourceUnionId);
61         for (Resource r : rlist) {
62             ResourceUtil.recalculate(r);
63         }
64         return rlist;
65     }
66
67     @Override
68     public AllocationOutcome allocateResources(AllocationRequest allocationRequest) {
69         if (allocationRequest == null) {
70             throw new IllegalArgumentException("allocateResources called with null argument");
71         }
72
73         AllocationFunction allocationFunction =
74                 new AllocationFunction(lockHelper, resourceDao, allocationRequest, lockTimeout);
75         allocationFunction.exec();
76         AllocationOutcome allocationOutcome = allocationFunction.getAllocationOutcome();
77
78         StrUtil.info(log, allocationOutcome);
79
80         return allocationOutcome;
81     }
82
83     @Override
84     public void releaseResources(ReleaseRequest releaseRequest) {
85         if (releaseRequest.resourceSetId == null && releaseRequest.resourceUnionId == null) {
86             return;
87         }
88
89         Set<String> lockNames = new HashSet<>();
90         if (releaseRequest.assetId != null) {
91             lockNames.add(releaseRequest.assetId);
92         } else {
93             List<Resource> resourceList = null;
94             if (releaseRequest.resourceSetId != null) {
95                 resourceList = resourceDao.getResourceSet(releaseRequest.resourceSetId);
96             } else {
97                 resourceList = resourceDao.getResourceUnion(releaseRequest.resourceUnionId);
98             }
99
100             if (resourceList == null || resourceList.isEmpty()) {
101                 return;
102             }
103
104             lockNames = getLockNames(resourceList);
105         }
106
107         ReleaseFunction releaseFunction =
108                 new ReleaseFunction(lockHelper, resourceDao, releaseRequest, lockNames, lockTimeout);
109         releaseFunction.exec();
110     }
111
112     @Override
113     public Resource queryResource(String resourceName, String assetId, String resourceUnionFilter,
114             String resourceShareGroupFilter) {
115         Resource r = resourceDao.query(assetId, resourceName, resourceUnionFilter, resourceShareGroupFilter);
116         ResourceUtil.recalculate(r);
117         return r;
118     }
119
120     @Override
121     public List<Resource> queryResources(String resourceName, String assetIdFilter) {
122         List<Resource> rlist = resourceDao.query(assetIdFilter, resourceName);
123         for (Resource r : rlist) {
124             ResourceUtil.recalculate(r);
125         }
126         return rlist;
127     }
128
129     private Set<String> getLockNames(List<Resource> resourceList) {
130         Set<String> lockNames = new HashSet<>();
131         for (Resource r : resourceList) {
132             lockNames.add(r.resourceKey.assetId);
133         }
134         return lockNames;
135     }
136
137     public void setResourceDao(ResourceDao resourceDao) {
138         this.resourceDao = resourceDao;
139     }
140
141     public void setLockTimeout(int lockTimeout) {
142         this.lockTimeout = lockTimeout;
143     }
144
145     public void setLockHelper(LockHelper lockHelper) {
146         this.lockHelper = lockHelper;
147     }
148 }