influxlogger/src/main/java/org/oran/pmlog/ConsumerRegstrationTask.java

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2023 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.oran.pmlog;
  22
  23 import java.time.Duration;
  24
  25 import lombok.Getter;
  26
  27 import org.oran.pmlog.clients.AsyncRestClient;
  28 import org.oran.pmlog.clients.AsyncRestClientFactory;
  29 import org.oran.pmlog.configuration.ApplicationConfig;
  30 import org.oran.pmlog.oauth2.SecurityContext;
  31 import org.slf4j.Logger;
  32 import org.slf4j.LoggerFactory;
  33 import org.springframework.beans.factory.annotation.Autowired;
  34 import org.springframework.http.ResponseEntity;
  35 import org.springframework.stereotype.Component;
  36
  37 import reactor.core.publisher.Mono;
  38 import reactor.util.retry.Retry;
  39
  40 /**
  41  * Registers the types and this producer in Innformation Coordinator Service.
  42  * This is done when needed.
  43  */
  44 @Component
  45 @SuppressWarnings("squid:S2629") // Invoke method(s) only conditionally
  46 public class ConsumerRegstrationTask {
  47
  48     private static final Logger logger = LoggerFactory.getLogger(ConsumerRegstrationTask.class);
  49     private final AsyncRestClient restClient;
  50     private final ApplicationConfig applicationConfig;
  51
  52     @Getter
  53     private boolean isRegisteredInIcs = false;
  54
  55     public ConsumerRegstrationTask(@Autowired ApplicationConfig applicationConfig,
  56             @Autowired SecurityContext securityContext) {
  57         AsyncRestClientFactory restClientFactory =
  58                 new AsyncRestClientFactory(applicationConfig.getWebClientConfig(), securityContext);
  59         this.restClient = restClientFactory.createRestClientNoHttpProxy("");
  60         this.applicationConfig = applicationConfig;
  61
  62         if (this.applicationConfig.getIcsBaseUrl().isEmpty()) {
  63             logger.info("Skipping createtion of info job. app.ics-base-url is empty.");
  64         } else {
  65             createSubscription();
  66         }
  67     }
  68
  69     private void createSubscription() {
  70         putInfoJob() //
  71                 .doOnError(this::handleRegistrationFailure)
  72                 .retryWhen(Retry.fixedDelay(100, Duration.ofMillis(5L * 1000))) //
  73                 .subscribe( //
  74                         null, //
  75                         this::handleRegistrationFailure, //
  76                         this::handleRegistrationCompleted);
  77     }
  78
  79     private void handleRegistrationCompleted() {
  80         logger.info("Registration of subscription/info job succeeded");
  81         isRegisteredInIcs = true;
  82     }
  83
  84     private void handleRegistrationFailure(Throwable t) {
  85         logger.warn("Creation of subscription/info job failed {}", t.getMessage());
  86     }
  87
  88     private Mono<ResponseEntity<String>> putInfoJob() {
  89         try {
  90             String jobId = this.applicationConfig.getConsumerJobId();
  91
  92             if (jobId.isBlank()) {
  93                 logger.info("Skipping creation of infojob. job_id is empty.");
  94                 return Mono.empty();
  95             }
  96             String url = applicationConfig.getIcsBaseUrl() + "/data-consumer/v1/info-jobs/" + jobId;
  97             String body = this.applicationConfig.getConsumerJobInfo();
  98             return restClient.putForEntity(url, body);
  99         } catch (Exception e) {
 100             logger.error("Registration of subscription failed {}", e.getMessage());
 101             return Mono.error(e);
 102         }
 103     }
 104
 105 }