0cb9135355fa1f61a852243a10bf81eedfe199c9
[ccsdk/oran.git] /
1 /*-
2  * ========================LICENSE_START=================================
3  * ONAP : ccsdk oran
4  * ======================================================================
5  * Copyright (C) 2024 OpenInfra Foundation Europe. All rights reserved.
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.onap.ccsdk.oran.a1policymanagementservice.util.v3;
22
23 import com.google.gson.Gson;
24 import com.google.gson.GsonBuilder;
25 import lombok.RequiredArgsConstructor;
26 import org.onap.ccsdk.oran.a1policymanagementservice.exceptions.ServiceException;
27 import org.onap.ccsdk.oran.a1policymanagementservice.models.v3.PolicyInformation;
28 import org.onap.ccsdk.oran.a1policymanagementservice.models.v3.PolicyObjectInformation;
29 import org.onap.ccsdk.oran.a1policymanagementservice.models.v3.PolicyTypeInformation;
30 import org.onap.ccsdk.oran.a1policymanagementservice.repository.*;
31 import org.slf4j.Logger;
32 import org.slf4j.LoggerFactory;
33 import org.springframework.http.HttpHeaders;
34 import org.springframework.http.HttpStatus;
35 import org.springframework.stereotype.Component;
36 import reactor.core.publisher.Flux;
37 import org.springframework.web.server.ServerWebExchange;
38 import org.springframework.web.util.UriComponentsBuilder;
39 import reactor.core.publisher.Mono;
40
41 import java.lang.invoke.MethodHandles;
42 import java.time.Instant;
43 import java.util.Collection;
44 import java.util.UUID;
45 import java.util.stream.Collectors;
46
47 @Component
48 @RequiredArgsConstructor
49 public class Helper {
50
51     private final Services services;
52
53     private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
54
55     private static final Gson gson = new GsonBuilder().create();
56
57     public void keepServiceAlive(String name) {
58         Service s = this.services.get(name);
59         if (s != null) {
60             s.keepAlive();
61         }
62     }
63
64     public Mono<Ric> checkRicStateIdle(Ric ric) {
65         if (ric.getState() == Ric.RicState.AVAILABLE) {
66             return Mono.just(ric);
67         } else {
68             logger.debug("Request rejected Near-RT RIC not IDLE, ric: {}", ric);
69             ServiceException e = new ServiceException(
70                     "Near-RT RIC: is not operational, id: " + ric.id() + ", state: " + ric.getState(),
71                     HttpStatus.LOCKED);
72             return Mono.error(e);
73         }
74     }
75
76     public Mono<Ric> checkSupportedType(Ric ric, PolicyType type) {
77         if (!ric.isSupportingType(type.getId())) {
78             logger.debug("Request rejected, type not supported, RIC: {}", ric);
79             ServiceException e = new ServiceException(
80                     "Type: " + type.getId() + " not supported by RIC: " + ric.id(), HttpStatus.BAD_REQUEST);
81             return Mono.error(e);
82         }
83         return Mono.just(ric);
84     }
85
86     public Policy buildPolicy(PolicyObjectInformation policyObjectInformation, PolicyType policyType, Ric ric, String policyId) {
87         return Policy.builder()
88                 .id(policyId)
89                 .json(toJson(policyObjectInformation.getPolicyObject()))
90                 .type(policyType)
91                 .ric(ric)
92                 .ownerServiceId(policyObjectInformation.getServiceId() == null ? ""
93                         : policyObjectInformation.getServiceId())
94                 .lastModified(Instant.now())
95                 .isTransient(policyObjectInformation.getTransient())
96                 .statusNotificationUri(policyObjectInformation.getStatusNotificationUri() == null ? ""
97                         : policyObjectInformation.getStatusNotificationUri())
98                 .build();
99     }
100
101     public Boolean jsonSchemaValidation(Object jsonObject) {
102         String jsonString = toJson(jsonObject);
103         return true;
104     }
105
106     public String policyIdGeneration(PolicyObjectInformation policyObjectInfo) {
107         if (policyObjectInfo.getPolicyId() == null || policyObjectInfo.getPolicyId().isEmpty() ||
108                 policyObjectInfo.getPolicyId().isBlank())
109             return UUID.randomUUID().toString();
110         else
111             return policyObjectInfo.getPolicyId().trim();
112     }
113
114     public String toJson(Object jsonObject) {
115         return gson.toJson(jsonObject);
116     }
117
118     public HttpHeaders createHttpHeaders(String headerName, String headerValue) {
119         HttpHeaders httpHeaders = new HttpHeaders();
120         httpHeaders.add(headerName, headerValue);
121         return httpHeaders;
122     }
123
124     public String buildURI(String policyId, ServerWebExchange serverWebExchange) {
125         return UriComponentsBuilder.fromHttpRequest(serverWebExchange.getRequest())
126                 .path("/{id}")
127                 .buildAndExpand(policyId)
128                 .toString();
129     }
130
131     public Mono<Policy> isPolicyAlreadyCreated(Policy policy, Policies policies) {
132         if (policies.get(policy.getId()) != null) {
133             return Mono.error(new ServiceException
134                     ("Policy already created with ID: " + policy.getId(), HttpStatus.CONFLICT));
135         }
136             return Mono.just(policy);
137     }
138
139     public Flux<PolicyInformation> toFluxPolicyInformation(Collection<Policy> policyCollection) {
140         return Flux.fromIterable(policyCollection.stream()
141                 .map(singlePolicy -> new PolicyInformation(singlePolicy.getId(), singlePolicy.getRic().getConfig().getRicId()))
142                 .collect(Collectors.toList()));
143     }
144
145     public Collection<PolicyTypeInformation> toPolicyTypeInfoCollection(Collection<PolicyType> policyTypeCollection, Ric ric) {
146         return policyTypeCollection.stream()
147                 .map(type -> new PolicyTypeInformation(type.getId(), ric.getConfig().getRicId()))
148                 .collect(Collectors.toList());
149     }
150 }