9ba88912db3ae4c633d11e1b7c1c86570e0dff5c
[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, 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, lockNames, lockTimeout);
105         releaseFunction.exec();
106     }
107
108     @Override
109     public Resource queryResource(String resourceName, String assetId, String resourceUnionFilter,
110             String resourceShareGroupFilter) {
111         Resource r = resourceDao.query(assetId, resourceName, resourceUnionFilter, resourceShareGroupFilter);
112         ResourceUtil.recalculate(r);
113         return r;
114     }
115
116     @Override
117     public List<Resource> queryResources(String resourceName, String assetIdFilter) {
118         List<Resource> rlist = resourceDao.query(assetIdFilter, resourceName);
119         for (Resource r : rlist) {
120             ResourceUtil.recalculate(r);
121         }
122         return rlist;
123     }
124
125     private Set<String> getLockNames(List<Resource> resourceList) {
126         Set<String> lockNames = new HashSet<>();
127         for (Resource r : resourceList) {
128             lockNames.add(r.resourceKey.assetId);
129         }
130         return lockNames;
131     }
132
133     public void setResourceDao(ResourceDao resourceDao) {
134         this.resourceDao = resourceDao;
135     }
136
137     public void setLockTimeout(int lockTimeout) {
138         this.lockTimeout = lockTimeout;
139     }
140
141     public void setLockHelper(LockHelper lockHelper) {
142         this.lockHelper = lockHelper;
143     }
144 }