policy-agent/src/main/java/org/oransc/policyagent/clients/AsyncRestClient.java

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2019 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.policyagent.clients;
  22
  23 import java.lang.invoke.MethodHandles;
  24
  25 import org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27 import org.springframework.http.HttpStatus;
  28 import org.springframework.http.MediaType;
  29 import org.springframework.web.reactive.function.client.WebClient;
  30 import reactor.core.publisher.Mono;
  31
  32 public class AsyncRestClient {
  33     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
  34     private final WebClient client;
  35
  36     public class AsyncRestClientException extends Exception {
  37
  38         private static final long serialVersionUID = 1L;
  39
  40         public AsyncRestClientException(String message) {
  41             super(message);
  42         }
  43     }
  44
  45     public AsyncRestClient(String baseUrl) {
  46         this.client = WebClient.create(baseUrl);
  47     }
  48
  49     public Mono<String> post(String uri, String body) {
  50         return client.post() //
  51             .uri(uri) //
  52             .contentType(MediaType.APPLICATION_JSON) //
  53             .bodyValue(body) //
  54             .retrieve() //
  55             .onStatus(HttpStatus::isError,
  56                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
  57             .bodyToMono(String.class);
  58     }
  59
  60     public Mono<String> postWithAuthHeader(String uri, String body, String username, String password) {
  61         return client.post() //
  62             .uri(uri) //
  63             .headers(headers -> headers.setBasicAuth(username, password)) //
  64             .contentType(MediaType.APPLICATION_JSON) //
  65             .bodyValue(body) //
  66             .retrieve() //
  67             .onStatus(HttpStatus::isError,
  68                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
  69             .bodyToMono(String.class);
  70     }
  71
  72     public Mono<String> put(String uri, String body) {
  73         logger.debug("PUT uri = '{}''", uri);
  74         return client.put() //
  75             .uri(uri) //
  76             .contentType(MediaType.APPLICATION_JSON) //
  77             .bodyValue(body) //
  78             .retrieve() //
  79             .onStatus(HttpStatus::isError,
  80                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
  81             .bodyToMono(String.class);
  82     }
  83
  84     public Mono<String> get(String uri) {
  85         logger.debug("GET uri = '{}''", uri);
  86         return client.get() //
  87             .uri(uri) //
  88             .retrieve() //
  89             .onStatus(HttpStatus::isError,
  90                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
  91             .bodyToMono(String.class);
  92     }
  93
  94     public Mono<String> delete(String uri) {
  95         logger.debug("DELETE uri = '{}''", uri);
  96         return client.delete() //
  97             .uri(uri) //
  98             .retrieve() //
  99             .onStatus(HttpStatus::isError,
 100                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
 101             .bodyToMono(String.class);
 102     }
 103 }