Adding customer and serive subscription handling
[integration/csit.git] / plans / so / integration-etsi-testing / so-simulators / aai-simulator / src / main / java / org / onap / so / aai / simulator / service / providers / CustomerServiceProviderImpl.java
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2019 Nordix Foundation.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20 package org.onap.so.aai.simulator.service.providers;
21
22 import java.util.Optional;
23 import java.util.concurrent.ConcurrentHashMap;
24 import org.onap.aai.domain.yang.Customer;
25 import org.onap.aai.domain.yang.ServiceSubscription;
26 import org.onap.so.aai.simulator.utils.Constant;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
29 import org.springframework.beans.factory.annotation.Autowired;
30 import org.springframework.cache.Cache;
31 import org.springframework.cache.CacheManager;
32 import org.springframework.stereotype.Service;
33
34 /**
35  * @author waqas.ikram@ericsson.com
36  *
37  */
38 @Service
39 public class CustomerServiceProviderImpl implements CustomerServiceProvider {
40     private static final Logger LOGGER = LoggerFactory.getLogger(CustomerServiceProviderImpl.class);
41
42     public final CacheManager cacheManager;
43
44     @Autowired
45     public CustomerServiceProviderImpl(final CacheManager cacheManager) {
46         this.cacheManager = cacheManager;
47     }
48
49     @Override
50     public Optional<Customer> getCustomer(final String globalCustomerId) {
51         LOGGER.info("getting customer from cache using key: {}", globalCustomerId);
52         final Cache cache = cacheManager.getCache(Constant.CUSTOMER_CACHE);
53         final Customer value = cache.get(globalCustomerId, Customer.class);
54         if (value != null) {
55             return Optional.of(value);
56         }
57         return Optional.empty();
58     }
59
60     @Override
61     public void putCustomer(final String globalCustomerId, final Customer customer) {
62         LOGGER.info("Adding customer: {} with key: {} in cache ...", customer, globalCustomerId);
63         final Cache cache = cacheManager.getCache(Constant.CUSTOMER_CACHE);
64
65         cache.put(globalCustomerId, customer);
66     }
67
68     @Override
69     public Optional<ServiceSubscription> getServiceSubscription(final String globalCustomerId,
70             final String serviceType) {
71         LOGGER.info("getting service subscription from cache for globalCustomerId: {} and serviceType: {}",
72                 globalCustomerId, serviceType);
73
74         final Cache cache = cacheManager.getCache(Constant.CUSTOMER_CACHE);
75
76         final Customer value = cache.get(globalCustomerId, Customer.class);
77
78         if (value != null) {
79             return Optional.ofNullable(value.getServiceSubscriptions().getServiceSubscription().stream()
80                     .filter(s -> serviceType.equals(s.getServiceType())).findFirst().orElse(null));
81         }
82         return Optional.empty();
83
84     }
85
86     @Override
87     public void clearAll() {
88         final Cache cache = cacheManager.getCache(Constant.CUSTOMER_CACHE);
89         final ConcurrentHashMap<?, ?> nativeCache = (ConcurrentHashMap<?, ?>) cache.getNativeCache();
90         LOGGER.info("Clear all entries from cahce: {}", cache.getName());
91         nativeCache.clear();
92     }
93
94 }