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.MediaType;
  28 import org.springframework.http.ResponseEntity;
  29 import org.springframework.lang.Nullable;
  30 import org.springframework.web.reactive.function.client.WebClient;
  31 import org.springframework.web.reactive.function.client.WebClient.RequestHeadersSpec;
  32 import reactor.core.publisher.Mono;
  33
  34 public class AsyncRestClient {
  35     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
  36     private final WebClient client;
  37     private final String baseUrl;
  38
  39     public AsyncRestClient(String baseUrl) {
  40         this.client = WebClient.create(baseUrl);
  41         this.baseUrl = baseUrl;
  42     }
  43
  44     public Mono<ResponseEntity<String>> postForEntity(String uri, @Nullable String body) {
  45         logger.debug("POST uri = '{}{}''", baseUrl, uri);
  46         Mono<String> bodyProducer = body != null ? Mono.just(body) : Mono.empty();
  47         RequestHeadersSpec<?> request = client.post() //
  48             .uri(uri) //
  49             .contentType(MediaType.APPLICATION_JSON) //
  50             .body(bodyProducer, String.class);
  51         return retrieve(request);
  52     }
  53
  54     public Mono<String> post(String uri, @Nullable String body) {
  55         return postForEntity(uri, body) //
  56             .flatMap(this::toBody);
  57     }
  58
  59     public Mono<String> postWithAuthHeader(String uri, String body, String username, String password) {
  60         logger.debug("POST (auth) uri = '{}{}''", baseUrl, uri);
  61         RequestHeadersSpec<?> request = client.post() //
  62             .uri(uri) //
  63             .headers(headers -> headers.setBasicAuth(username, password)) //
  64             .contentType(MediaType.APPLICATION_JSON) //
  65             .bodyValue(body);
  66         return retrieve(request) //
  67             .flatMap(this::toBody);
  68     }
  69
  70     public Mono<ResponseEntity<String>> putForEntity(String uri, String body) {
  71         logger.debug("PUT uri = '{}{}''", baseUrl, uri);
  72         RequestHeadersSpec<?> request = client.put() //
  73             .uri(uri) //
  74             .contentType(MediaType.APPLICATION_JSON) //
  75             .bodyValue(body);
  76         return retrieve(request);
  77     }
  78
  79     public Mono<String> put(String uri, String body) {
  80         return putForEntity(uri, body) //
  81             .flatMap(this::toBody);
  82     }
  83
  84     public Mono<ResponseEntity<String>> getForEntity(String uri) {
  85         logger.debug("GET uri = '{}{}''", baseUrl, uri);
  86         RequestHeadersSpec<?> request = client.get().uri(uri);
  87         return retrieve(request);
  88     }
  89
  90     public Mono<String> get(String uri) {
  91         return getForEntity(uri) //
  92             .flatMap(this::toBody);
  93     }
  94
  95     public Mono<ResponseEntity<String>> deleteForEntity(String uri) {
  96         logger.debug("DELETE uri = '{}{}''", baseUrl, uri);
  97         RequestHeadersSpec<?> request = client.delete().uri(uri);
  98         return retrieve(request);
  99     }
 100
 101     public Mono<String> delete(String uri) {
 102         return deleteForEntity(uri) //
 103             .flatMap(this::toBody);
 104     }
 105
 106     private Mono<ResponseEntity<String>> retrieve(RequestHeadersSpec<?> request) {
 107         return request.retrieve() //
 108             .toEntity(String.class);
 109     }
 110
 111     Mono<String> toBody(ResponseEntity<String> entity) {
 112         if (entity.getBody() == null) {
 113             return Mono.just("");
 114         } else {
 115             return Mono.just(entity.getBody());
 116         }
 117     }
 118
 119 }