c4d53ce708dbf9aeae4d7f63395513810f54799c
[ccsdk/sli/adaptors.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.Resource;
32 import org.onap.ccsdk.sli.adaptors.rm.util.ResourceUtil;
33 import org.onap.ccsdk.sli.adaptors.util.str.StrUtil;
34 import org.slf4j.Logger;
35 import org.slf4j.LoggerFactory;
36
37 public class ResourceManagerImpl implements ResourceManager {
38
39     private static final Logger log = LoggerFactory.getLogger(ResourceManagerImpl.class);
40
41     private LockHelper lockHelper;
42     private ResourceDao resourceDao;
43
44     private int lockTimeout = 10 * 60; // Default 10 min
45
46     public ResourceManagerImpl() {
47         log.info("ResourceManager created.");
48     }
49
50     @Override
51     public Resource getResource(String resourceName, String assetId) {
52         Resource r = resourceDao.getResource(assetId, resourceName);
53         ResourceUtil.recalculate(r);
54         return r;
55     }
56
57     @Override
58     public List<Resource> getResourceUnion(String resourceUnionId) {
59         List<Resource> rlist = resourceDao.getResourceUnion(resourceUnionId);
60         for (Resource r : rlist) {
61             ResourceUtil.recalculate(r);
62         }
63         return rlist;
64     }
65
66     @Override
67     public AllocationOutcome allocateResources(AllocationRequest allocationRequest) {
68         if (allocationRequest == null) {
69             throw new IllegalArgumentException("allocateResources called with null argument");
70         }
71
72         AllocationFunction allocationFunction =
73                 new AllocationFunction(lockHelper, resourceDao, allocationRequest, lockTimeout);
74         allocationFunction.exec();
75         AllocationOutcome allocationOutcome = allocationFunction.getAllocationOutcome();
76
77         StrUtil.info(log, allocationOutcome);
78
79         return allocationOutcome;
80     }
81
82     @Override
83     public void releaseResourceSet(String resourceSetId) {
84         List<Resource> resourceList = resourceDao.getResourceSet(resourceSetId);
85         if (resourceList == null || resourceList.isEmpty()) {
86             return;
87         }
88
89         Set<String> lockNames = getLockNames(resourceList);
90         ReleaseFunction releaseFunction =
91                 new ReleaseFunction(lockHelper, resourceDao, resourceSetId, null, null, lockNames, lockTimeout);
92         releaseFunction.exec();
93     }
94
95     @Override
96     public void releaseResourceUnion(String resourceUnionId) {
97         List<Resource> resourceList = resourceDao.getResourceUnion(resourceUnionId);
98         if (resourceList == null || resourceList.isEmpty()) {
99             return;
100         }
101
102         Set<String> lockNames = getLockNames(resourceList);
103         ReleaseFunction releaseFunction =
104                 new ReleaseFunction(lockHelper, resourceDao, null, resourceUnionId, null, lockNames, lockTimeout);
105         releaseFunction.exec();
106     }
107
108     @Override
109     public void releaseResourceSet(String resourceSetId, String assetId) {
110         List<Resource> resourceList = resourceDao.getResourceSetForAsset(resourceSetId, assetId);
111         if (resourceList == null || resourceList.isEmpty()) {
112             return;
113         }
114
115         Set<String> lockNames = getLockNames(resourceList);
116         ReleaseFunction releaseFunction =
117                 new ReleaseFunction(lockHelper, resourceDao, resourceSetId, null, assetId, lockNames, lockTimeout);
118         releaseFunction.exec();
119     }
120
121     @Override
122     public void releaseResourceUnion(String resourceUnionId, String assetId) {
123         List<Resource> resourceList = resourceDao.getResourceUnionForAsset(resourceUnionId, assetId);
124         if (resourceList == null || resourceList.isEmpty()) {
125             return;
126         }
127
128         Set<String> lockNames = getLockNames(resourceList);
129         ReleaseFunction releaseFunction =
130                 new ReleaseFunction(lockHelper, resourceDao, null, resourceUnionId, assetId, lockNames, lockTimeout);
131         releaseFunction.exec();
132     }
133
134     @Override
135     public Resource queryResource(String resourceName, String assetId, String resourceUnionFilter,
136             String resourceShareGroupFilter) {
137         Resource r = resourceDao.query(assetId, resourceName, resourceUnionFilter, resourceShareGroupFilter);
138         ResourceUtil.recalculate(r);
139         return r;
140     }
141
142     @Override
143     public List<Resource> queryResources(String resourceName, String assetIdFilter) {
144         List<Resource> rlist = resourceDao.query(assetIdFilter, resourceName);
145         for (Resource r : rlist) {
146             ResourceUtil.recalculate(r);
147         }
148         return rlist;
149     }
150
151     private Set<String> getLockNames(List<Resource> resourceList) {
152         Set<String> lockNames = new HashSet<>();
153         for (Resource r : resourceList) {
154             lockNames.add(r.resourceKey.assetId);
155         }
156         return lockNames;
157     }
158
159     public void setResourceDao(ResourceDao resourceDao) {
160         this.resourceDao = resourceDao;
161     }
162
163     public void setLockTimeout(int lockTimeout) {
164         this.lockTimeout = lockTimeout;
165     }
166
167     public void setLockHelper(LockHelper lockHelper) {
168         this.lockHelper = lockHelper;
169     }
170 }