Merge "Junit tests for ServiceException"
[nonrtric.git] / policy-agent / src / main / java / org / oransc / policyagent / clients / SdncOscA1Client.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.policyagent.clients;
22
23 import java.lang.invoke.MethodHandles;
24 import java.util.List;
25 import org.oransc.policyagent.configuration.RicConfig;
26 import org.oransc.policyagent.repository.Policy;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
29 import reactor.core.publisher.Flux;
30 import reactor.core.publisher.Mono;
31
32 public class SdncOscA1Client implements A1Client {
33     private static final String URL_PREFIX = "/A1-ADAPTER-API:";
34
35     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
36
37     private final String a1ControllerUsername;
38     private final String a1ControllerPassword;
39     private final RicConfig ricConfig;
40     private final AsyncRestClient restClient;
41
42     public SdncOscA1Client(RicConfig ricConfig, String baseUrl, String username, String password) {
43         this(ricConfig, username, password, new AsyncRestClient(baseUrl + "/restconf/operations"));
44         if (logger.isDebugEnabled()) {
45             logger.debug("SdncOscA1Client for ric: {}, a1ControllerBaseUrl: {}", ricConfig.name(), baseUrl);
46         }
47     }
48
49     public SdncOscA1Client(RicConfig ricConfig, String username, String password, AsyncRestClient restClient) {
50         this.ricConfig = ricConfig;
51         this.a1ControllerUsername = username;
52         this.a1ControllerPassword = password;
53         this.restClient = restClient;
54     }
55
56     @Override
57     public Mono<List<String>> getPolicyTypeIdentities() {
58         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
59             .nearRtRicUrl(ricConfig.baseUrl()) //
60             .build();
61         String inputJsonString = JsonHelper.createInputJsonString(inputParams);
62         logger.debug("POST getPolicyTypeIdentities inputJsonString = {}", inputJsonString);
63
64         return restClient
65             .postWithAuthHeader(URL_PREFIX + "getPolicyTypeIdentities", inputJsonString, a1ControllerUsername,
66                 a1ControllerPassword) //
67             .flatMap(response -> JsonHelper.getValueFromResponse(response, "policy-type-id-list")) //
68             .flatMapMany(JsonHelper::parseJsonArrayOfString) //
69             .collectList();
70     }
71
72     @Override
73     public Mono<List<String>> getPolicyIdentities() {
74         return getPolicyIds() //
75             .collectList();
76     }
77
78     @Override
79     public Mono<String> getPolicyTypeSchema(String policyTypeId) {
80         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
81             .nearRtRicUrl(ricConfig.baseUrl()) //
82             .policyTypeId(policyTypeId) //
83             .build();
84         String inputJsonString = JsonHelper.createInputJsonString(inputParams);
85         logger.debug("POST getPolicyType inputJsonString = {}", inputJsonString);
86
87         return restClient
88             .postWithAuthHeader(URL_PREFIX + "getPolicyType", inputJsonString, a1ControllerUsername,
89                 a1ControllerPassword) //
90             .flatMap(response -> JsonHelper.getValueFromResponse(response, "policy-type")) //
91             .flatMap(JsonHelper::extractPolicySchema);
92     }
93
94     @Override
95     public Mono<String> putPolicy(Policy policy) {
96         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
97             .nearRtRicUrl(ricConfig.baseUrl()) //
98             .policyTypeId(policy.type().name()) //
99             .policyId(policy.id()) //
100             .policy(policy.json()) //
101             .build();
102         String inputJsonString = JsonHelper.createInputJsonString(inputParams);
103         logger.debug("POST putPolicy inputJsonString = {}", inputJsonString);
104
105         return restClient
106             .postWithAuthHeader(URL_PREFIX + "putPolicy", inputJsonString, a1ControllerUsername, a1ControllerPassword)
107             .flatMap(response -> JsonHelper.getValueFromResponse(response, "returned-policy")) //
108             .flatMap(JsonHelper::validateJson);
109     }
110
111     @Override
112     public Mono<String> deletePolicy(Policy policy) {
113         return deletePolicyById(policy.id());
114     }
115
116     @Override
117     public Flux<String> deleteAllPolicies() {
118         return getPolicyIds() //
119             .flatMap(this::deletePolicyById); //
120     }
121
122     @Override
123     public Mono<A1ProtocolType> getProtocolVersion() {
124         return getPolicyTypeIdentities() //
125             .flatMap(x -> Mono.just(A1ProtocolType.SDNC_OSC));
126     }
127
128     @Override
129     public Mono<String> getPolicyStatus(Policy policy) {
130         return Mono.error(new Exception("Status not implemented in the SDNC controller"));
131     }
132
133     private Flux<String> getPolicyIds() {
134         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
135             .nearRtRicUrl(ricConfig.baseUrl()) //
136             .build();
137         String inputJsonString = JsonHelper.createInputJsonString(inputParams);
138         logger.debug("POST getPolicyIdentities inputJsonString = {}", inputJsonString);
139
140         return restClient
141             .postWithAuthHeader("/A1-ADAPTER-API:getPolicyIdentities", inputJsonString, a1ControllerUsername,
142                 a1ControllerPassword) //
143             .flatMap(response -> JsonHelper.getValueFromResponse(response, "policy-id-list")) //
144             .flatMapMany(JsonHelper::parseJsonArrayOfString);
145     }
146
147     private Mono<String> deletePolicyById(String policyId) {
148         SdncOscAdapterInput inputParams = ImmutableSdncOscAdapterInput.builder() //
149             .nearRtRicUrl(ricConfig.baseUrl()) //
150             .policyId(policyId) //
151             .build();
152
153         String inputJsonString = JsonHelper.createInputJsonString(inputParams);
154         logger.debug("POST deletePolicy inputJsonString = {}", inputJsonString);
155
156         return restClient.postWithAuthHeader(URL_PREFIX + "deletePolicy", inputJsonString, a1ControllerUsername,
157             a1ControllerPassword);
158     }
159 }