87d1dba1ad56f817cd8d815f062cd3d2f3f16e46
[nonrtric.git] / enrichment-coordinator-service / src / main / java / org / oransc / enrichment / clients / ProducerCallbacks.java
1 /*-
2  * ========================LICENSE_START=================================
3  * O-RAN-SC
4  * %%
5  * Copyright (C) 2020 Nordix Foundation
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
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
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===================================
19  */
20
21 package org.oransc.enrichment.clients;
22
23 import com.google.gson.Gson;
24 import com.google.gson.GsonBuilder;
25
26 import java.lang.invoke.MethodHandles;
27
28 import org.oransc.enrichment.configuration.ApplicationConfig;
29 import org.oransc.enrichment.repository.EiJob;
30 import org.oransc.enrichment.repository.EiProducer;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33
34 import reactor.core.publisher.Flux;
35 import reactor.core.publisher.Mono;
36
37 /**
38  * Callbacks to the EiProducer
39  */
40 @SuppressWarnings("java:S3457") // No need to call "toString()" method as formatting and string ..
41 public class ProducerCallbacks {
42
43     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
44     private static Gson gson = new GsonBuilder().create();
45
46     private final AsyncRestClient restClient;
47
48     public ProducerCallbacks(ApplicationConfig config) {
49         AsyncRestClientFactory restClientFactory = new AsyncRestClientFactory(config.getWebClientConfig());
50         this.restClient = restClientFactory.createRestClient("");
51     }
52
53     public void notifyProducersJobDeleted(EiJob eiJob) {
54         ProducerJobInfo request = new ProducerJobInfo(eiJob);
55         String body = gson.toJson(request);
56         for (EiProducer producer : eiJob.type().getProducers()) {
57             restClient.post(producer.getJobDeletionCallbackUrl(), body) //
58                 .subscribe(notUsed -> logger.debug("Job deleted OK {}", producer.getId()), //
59                     throwable -> logger.warn("Job delete failed {}", producer.getId(), throwable.toString()), null);
60         }
61     }
62
63     /**
64      * Calls all producers for an EiJob activation.
65      * 
66      * @param eiJob an EI job
67      * @return the number of producers that returned OK
68      */
69     public Mono<Integer> notifyProducersJobStarted(EiJob eiJob) {
70         return Flux.fromIterable(eiJob.type().getProducers()) //
71             .flatMap(eiProducer -> notifyProducerJobStarted(eiProducer, eiJob)) //
72             .collectList() //
73             .flatMap(okResponses -> Mono.just(Integer.valueOf(okResponses.size()))); //
74     }
75
76     /**
77      * Calls one producer for an EiJob activation.
78      * 
79      * @param producer a producer
80      * @param eiJob an EI job
81      * @return the body of the response from the REST call
82      */
83     public Mono<String> notifyProducerJobStarted(EiProducer producer, EiJob eiJob) {
84         ProducerJobInfo request = new ProducerJobInfo(eiJob);
85         String body = gson.toJson(request);
86
87         return restClient.post(producer.getJobCreationCallbackUrl(), body)
88             .doOnNext(resp -> logger.debug("Job subscription started OK {}", producer.getId()))
89             .onErrorResume(throwable -> {
90                 logger.warn("Job subscription failed {}", producer.getId(), throwable.toString());
91                 return Mono.empty();
92             });
93     }
94
95 }