2 * ========================LICENSE_START=================================
4 * ======================================================================
5 * Copyright (C) 2019-2020 Nordix Foundation. All rights reserved.
6 * ======================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
18 * ========================LICENSE_END===================================
21 package org.onap.ccsdk.oran.a1policymanagementservice.tasks;
23 import static org.onap.ccsdk.oran.a1policymanagementservice.repository.Ric.RicState;
25 import org.onap.ccsdk.oran.a1policymanagementservice.clients.A1Client;
26 import org.onap.ccsdk.oran.a1policymanagementservice.clients.A1ClientFactory;
27 import org.onap.ccsdk.oran.a1policymanagementservice.clients.AsyncRestClientFactory;
28 import org.onap.ccsdk.oran.a1policymanagementservice.controllers.ServiceCallbacks;
29 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Lock.LockType;
30 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Policies;
31 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Policy;
32 import org.onap.ccsdk.oran.a1policymanagementservice.repository.PolicyType;
33 import org.onap.ccsdk.oran.a1policymanagementservice.repository.PolicyTypes;
34 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Ric;
35 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Rics;
36 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Services;
37 import org.slf4j.Logger;
38 import org.slf4j.LoggerFactory;
40 import reactor.core.publisher.Flux;
41 import reactor.core.publisher.Mono;
44 * Synchronizes the content of a Near-RT RIC with the content in the repository.
47 * load all policy types
49 * send all policy instances to the Near-RT RIC
51 * if that fails remove all policy instances
53 * Notify subscribing services
55 @SuppressWarnings("squid:S2629") // Invoke method(s) only conditionally
56 public class RicSynchronizationTask {
58 private static final Logger logger = LoggerFactory.getLogger(RicSynchronizationTask.class);
59 static final int CONCURRENCY_RIC = 1; // How many paralell requests that is sent to one NearRT RIC
61 private final A1ClientFactory a1ClientFactory;
62 private final PolicyTypes policyTypes;
63 private final Policies policies;
64 private final Services services;
65 private final Rics rics;
66 private final AsyncRestClientFactory restClientFactory;
68 public RicSynchronizationTask(A1ClientFactory a1ClientFactory, PolicyTypes policyTypes, Policies policies,
69 Services services, AsyncRestClientFactory restClientFactory, Rics rics) {
70 this.a1ClientFactory = a1ClientFactory;
71 this.policyTypes = policyTypes;
72 this.policies = policies;
73 this.services = services;
74 this.restClientFactory = restClientFactory;
78 public void run(Ric ric) {
79 logger.debug("Ric synchronization task created: {}", ric.getConfig().getRicId());
81 if (ric.getState() == RicState.SYNCHRONIZING) {
82 logger.debug("Ric: {} is already being synchronized", ric.getConfig().getRicId());
86 ric.getLock().lock(LockType.EXCLUSIVE, "RicSynchronizationTask") //
87 .flatMap(notUsed -> synchronizeRic(ric)) //
88 .doFinally(sig -> ric.getLock().unlockBlocking()) //
92 public Mono<Ric> synchronizeRic(Ric ric) {
93 return this.a1ClientFactory.createA1Client(ric) //
94 .doOnNext(client -> ric.setState(RicState.SYNCHRONIZING)) //
95 .flatMapMany(client -> runSynchronization(ric, client)) //
96 .onErrorResume(throwable -> deleteAllPolicyInstances(ric, throwable)) //
98 .map(notUsed -> ric) //
100 logger.warn("Synchronization failure for ric: {}, reason: {}", ric.id(), t.getMessage()); //
101 ric.setState(RicState.UNAVAILABLE); //
103 .flatMap(notUsed -> onSynchronizationComplete(ric)) //
104 .onErrorResume(t -> Mono.just(ric));
107 public Flux<PolicyType> synchronizePolicyTypes(Ric ric, A1Client a1Client) {
108 return a1Client.getPolicyTypeIdentities() //
109 .doOnNext(x -> ric.clearSupportedPolicyTypes()) //
110 .flatMapMany(Flux::fromIterable) //
111 .doOnNext(typeId -> logger.debug("For ric: {}, handling type: {}", ric.getConfig().getRicId(), typeId)) //
112 .flatMap(policyTypeId -> getPolicyType(policyTypeId, a1Client), CONCURRENCY_RIC) //
113 .doOnNext(ric::addSupportedPolicyType); //
116 private Flux<Object> runSynchronization(Ric ric, A1Client a1Client) {
117 Flux<PolicyType> synchronizedTypes = synchronizePolicyTypes(ric, a1Client);
118 Flux<?> policiesDeletedInRic = a1Client.deleteAllPolicies();
119 Flux<Policy> policiesRecreatedInRic = recreateAllPoliciesInRic(ric, a1Client);
121 return Flux.concat(synchronizedTypes, policiesDeletedInRic, policiesRecreatedInRic);
124 private Mono<Ric> onSynchronizationComplete(Ric ric) {
125 if (this.rics.get(ric.id()) == null) {
126 logger.debug("Policies removed in removed ric: {}", ric.id());
129 logger.debug("Synchronization completed for: {}", ric.id());
130 ric.setState(RicState.AVAILABLE);
131 ServiceCallbacks callbacks = new ServiceCallbacks(this.restClientFactory);
132 return callbacks.notifyServicesRicAvailable(ric, services) //
137 private Flux<Object> deleteAllPolicyInstances(Ric ric, Throwable t) {
138 logger.warn("Recreation of policies failed for ric: {}, reason: {}", ric.id(), t.getMessage());
139 deleteAllPoliciesInRepository(ric);
141 Flux<PolicyType> synchronizedTypes = this.a1ClientFactory.createA1Client(ric) //
142 .flatMapMany(a1Client -> synchronizePolicyTypes(ric, a1Client));
143 Flux<?> deletePoliciesInRic = this.a1ClientFactory.createA1Client(ric) //
144 .flatMapMany(A1Client::deleteAllPolicies) //
145 .doOnComplete(() -> deleteAllPoliciesInRepository(ric));
147 return Flux.concat(synchronizedTypes, deletePoliciesInRic);
150 private Mono<PolicyType> getPolicyType(String policyTypeId, A1Client a1Client) {
151 if (policyTypes.contains(policyTypeId)) {
152 return Mono.just(policyTypes.get(policyTypeId));
154 return a1Client.getPolicyTypeSchema(policyTypeId) //
155 .map(schema -> createPolicyType(policyTypeId, schema));
158 private PolicyType createPolicyType(String policyTypeId, String schema) {
159 PolicyType pt = PolicyType.builder().id(policyTypeId).schema(schema).build();
164 private void deleteAllPoliciesInRepository(Ric ric) {
165 for (Policy policy : policies.getForRic(ric.id())) {
166 this.policies.remove(policy);
170 private Flux<Policy> putPolicy(Policy policy, Ric ric, A1Client a1Client) {
171 logger.trace("Recreating policy: {}, for ric: {}", policy.getId(), ric.getConfig().getRicId());
172 return a1Client.putPolicy(policy) //
173 .flatMapMany(notUsed -> Flux.just(policy));
176 private boolean checkTransient(Policy policy) {
177 if (policy.isTransient()) {
178 this.policies.remove(policy);
180 return policy.isTransient();
183 private Flux<Policy> recreateAllPoliciesInRic(Ric ric, A1Client a1Client) {
184 return Flux.fromIterable(policies.getForRic(ric.id())) //
185 .doOnNext(policy -> logger.debug("Recreating policy: {}, ric: {}", policy.getId(), ric.id())) //
186 .filter(policy -> !checkTransient(policy)) //
187 .flatMap(policy -> putPolicy(policy, ric, a1Client), CONCURRENCY_RIC)
188 .doOnError(t -> logger.warn("Recreating policy failed, ric: {}, reason: {}", ric.id(), t.getMessage()));