Updates of the NBI
[nonrtric.git] / 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 package org.oransc.policyagent.clients;
21
22 import org.springframework.http.HttpStatus;
23 import org.springframework.http.MediaType;
24 import org.springframework.web.reactive.function.client.WebClient;
25 import reactor.core.publisher.Mono;
26
27 public class AsyncRestClient {
28     private final WebClient client;
29
30     private static class AsyncRestClientException extends Exception {
31
32         private static final long serialVersionUID = 1L;
33
34         public AsyncRestClientException(String message) {
35             super(message);
36         }
37     }
38
39     public AsyncRestClient(String baseUrl) {
40         this.client = WebClient.create(baseUrl);
41     }
42
43     public Mono<String> put(String uri, String body) {
44         return client.put() //
45             .uri(uri) //
46             .contentType(MediaType.APPLICATION_JSON) //
47             .syncBody(body) //
48             .retrieve() //
49             .onStatus(HttpStatus::isError,
50                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
51             .bodyToMono(String.class);
52     }
53
54     public Mono<String> get(String uri) {
55         return client.get() //
56             .uri(uri) //
57             .retrieve() //
58             .onStatus(HttpStatus::isError,
59                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
60             .bodyToMono(String.class);
61     }
62
63     public Mono<String> delete(String uri) {
64         return client.delete() //
65             .uri(uri) //
66             .retrieve() //
67             .onStatus(HttpStatus::isError,
68                 response -> Mono.error(new AsyncRestClientException(response.statusCode().toString()))) //
69             .bodyToMono(String.class);
70     }
71 }