Add tests for RefreshConfigTask
[nonrtric.git] / policy-agent / src / main / java / org / oransc / policyagent / clients / OscA1Client.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
26 import org.json.JSONObject;
27 import org.oransc.policyagent.configuration.RicConfig;
28 import org.oransc.policyagent.repository.Policy;
29 import org.slf4j.Logger;
30 import org.slf4j.LoggerFactory;
31 import org.springframework.web.util.UriComponentsBuilder;
32
33 import reactor.core.publisher.Flux;
34 import reactor.core.publisher.Mono;
35
36 public class OscA1Client implements A1Client {
37     private static final String URL_PREFIX = "/a1-p";
38
39     private static final String POLICY_TYPES = "/policytypes";
40     private static final String CREATE_SCHEMA = "create_schema";
41     private static final String TITLE = "title";
42
43     private static final String HEALTHCHECK = "/healthcheck";
44
45     private static final UriComponentsBuilder POLICY_TYPE_SCHEMA_URI =
46         UriComponentsBuilder.fromPath("/policytypes/{policy-type-name}");
47
48     private static final UriComponentsBuilder POLICY_URI =
49         UriComponentsBuilder.fromPath("/policytypes/{policy-type-name}/policies/{policy-id}");
50
51     private static final UriComponentsBuilder POLICY_IDS_URI =
52         UriComponentsBuilder.fromPath("/policytypes/{policy-type-name}/policies");
53
54     private static final UriComponentsBuilder POLICY_STATUS_URI =
55         UriComponentsBuilder.fromPath("/policytypes/{policy-type-name}/policies/{policy-id}/status");
56
57     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
58
59     private final AsyncRestClient restClient;
60
61     public OscA1Client(RicConfig ricConfig) {
62         String baseUrl = ricConfig.baseUrl() + URL_PREFIX;
63         this.restClient = new AsyncRestClient(baseUrl);
64         if (logger.isDebugEnabled()) {
65             logger.debug("OscA1Client for ric: {}", ricConfig.name());
66         }
67     }
68
69     public OscA1Client(AsyncRestClient restClient) {
70         this.restClient = restClient;
71     }
72
73     @Override
74     public Mono<List<String>> getPolicyTypeIdentities() {
75         return getPolicyTypeIds() //
76             .collectList();
77     }
78
79     @Override
80     public Mono<List<String>> getPolicyIdentities() {
81         return getPolicyTypeIds() //
82             .flatMap(this::getPolicyIdentitiesByType) //
83             .collectList();
84     }
85
86     @Override
87     public Mono<String> getPolicyTypeSchema(String policyTypeId) {
88         String uri = POLICY_TYPE_SCHEMA_URI.buildAndExpand(policyTypeId).toUriString();
89         return restClient.get(uri) //
90             .flatMap(response -> getCreateSchema(response, policyTypeId));
91     }
92
93     @Override
94     public Mono<String> putPolicy(Policy policy) {
95         String uri = POLICY_URI.buildAndExpand(policy.type().name(), policy.id()).toUriString();
96         return restClient.put(uri, policy.json());
97     }
98
99     @Override
100     public Mono<String> deletePolicy(Policy policy) {
101         return deletePolicyById(policy.type().name(), policy.id());
102     }
103
104     @Override
105     public Mono<A1ProtocolType> getProtocolVersion() {
106         return restClient.get(HEALTHCHECK) //
107             .flatMap(notUsed -> Mono.just(A1ProtocolType.OSC_V1));
108     }
109
110     @Override
111     public Flux<String> deleteAllPolicies() {
112         return getPolicyTypeIds() //
113             .flatMap(this::deletePoliciesForType);
114     }
115
116     @Override
117     public Mono<String> getPolicyStatus(Policy policy) {
118         String uri = POLICY_STATUS_URI.buildAndExpand(policy.type().name(), policy.id()).toUriString();
119         return restClient.get(uri);
120
121     }
122
123     private Flux<String> getPolicyTypeIds() {
124         return restClient.get(POLICY_TYPES) //
125             .flatMapMany(JsonHelper::parseJsonArrayOfString);
126     }
127
128     private Flux<String> getPolicyIdentitiesByType(String typeId) {
129         return restClient.get(POLICY_IDS_URI.buildAndExpand(typeId).toUriString()) //
130             .flatMapMany(JsonHelper::parseJsonArrayOfString);
131     }
132
133     private Mono<String> getCreateSchema(String policyTypeResponse, String policyTypeId) {
134         try {
135             JSONObject obj = new JSONObject(policyTypeResponse);
136             JSONObject schemaObj = obj.getJSONObject(CREATE_SCHEMA);
137             schemaObj.put(TITLE, policyTypeId);
138             return Mono.just(schemaObj.toString());
139         } catch (Exception e) {
140             String exceptionString = e.toString();
141             logger.error("Unexpected response for policy type: {}, exception: {}", policyTypeResponse, exceptionString);
142             return Mono.error(e);
143         }
144     }
145
146     private Mono<String> deletePolicyById(String typeId, String policyId) {
147         String uri = POLICY_URI.buildAndExpand(typeId, policyId).toUriString();
148         return restClient.delete(uri);
149     }
150
151     private Flux<String> deletePoliciesForType(String typeId) {
152         return getPolicyIdentitiesByType(typeId) //
153             .flatMap(policyId -> deletePolicyById(typeId, policyId));
154     }
155 }