enrichment-coordinator-service/src/main/java/org/oransc/enrichment/controllers/a1e/A1eController.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.enrichment.controllers.a1e;
  22
  23 import com.fasterxml.jackson.databind.ObjectMapper;
  24 import com.google.gson.Gson;
  25 import com.google.gson.GsonBuilder;
  26
  27 import io.swagger.v3.oas.annotations.Operation;
  28 import io.swagger.v3.oas.annotations.Parameter;
  29 import io.swagger.v3.oas.annotations.media.ArraySchema;
  30 import io.swagger.v3.oas.annotations.media.Content;
  31 import io.swagger.v3.oas.annotations.media.Schema;
  32 import io.swagger.v3.oas.annotations.responses.ApiResponse;
  33 import io.swagger.v3.oas.annotations.responses.ApiResponses;
  34 import io.swagger.v3.oas.annotations.tags.Tag;
  35
  36 import java.lang.invoke.MethodHandles;
  37 import java.net.URI;
  38 import java.net.URISyntaxException;
  39 import java.util.ArrayList;
  40 import java.util.List;
  41
  42 import org.json.JSONObject;
  43 import org.oransc.enrichment.configuration.ApplicationConfig;
  44 import org.oransc.enrichment.controllers.ErrorResponse;
  45 import org.oransc.enrichment.controllers.VoidResponse;
  46 import org.oransc.enrichment.controllers.r1producer.ProducerCallbacks;
  47 import org.oransc.enrichment.exceptions.ServiceException;
  48 import org.oransc.enrichment.repository.InfoJob;
  49 import org.oransc.enrichment.repository.InfoJobs;
  50 import org.oransc.enrichment.repository.InfoProducers;
  51 import org.oransc.enrichment.repository.InfoType;
  52 import org.oransc.enrichment.repository.InfoTypes;
  53 import org.slf4j.Logger;
  54 import org.slf4j.LoggerFactory;
  55 import org.springframework.beans.factory.annotation.Autowired;
  56 import org.springframework.http.HttpStatus;
  57 import org.springframework.http.MediaType;
  58 import org.springframework.http.ResponseEntity;
  59 import org.springframework.web.bind.annotation.DeleteMapping;
  60 import org.springframework.web.bind.annotation.GetMapping;
  61 import org.springframework.web.bind.annotation.PathVariable;
  62 import org.springframework.web.bind.annotation.PutMapping;
  63 import org.springframework.web.bind.annotation.RequestBody;
  64 import org.springframework.web.bind.annotation.RequestMapping;
  65 import org.springframework.web.bind.annotation.RequestParam;
  66 import org.springframework.web.bind.annotation.RestController;
  67 import reactor.core.publisher.Mono;
  68
  69 @SuppressWarnings("java:S3457") // No need to call "toString()" method as formatting and string ..
  70 @RestController("A1-EI")
  71 @Tag(name = A1eConsts.CONSUMER_API_NAME)
  72 @RequestMapping(path = A1eConsts.API_ROOT, produces = MediaType.APPLICATION_JSON_VALUE)
  73 public class A1eController {
  74
  75     private final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
  76
  77     @Autowired
  78     ApplicationConfig applicationConfig;
  79
  80     @Autowired
  81     private InfoJobs eiJobs;
  82
  83     @Autowired
  84     private InfoTypes eiTypes;
  85
  86     @Autowired
  87     private InfoProducers infoProducers;
  88
  89     @Autowired
  90     ProducerCallbacks producerCallbacks;
  91
  92     private static Gson gson = new GsonBuilder().create();
  93
  94     @GetMapping(path = "/eitypes", produces = MediaType.APPLICATION_JSON_VALUE)
  95     @Operation(summary = "EI type identifiers", description = "")
  96     @ApiResponses(
  97         value = { //
  98             @ApiResponse(
  99                 responseCode = "200",
 100                 description = "EI type identifiers", //
 101                 content = @Content(array = @ArraySchema(schema = @Schema(implementation = String.class)))), //
 102         })
 103     public ResponseEntity<Object> getEiTypeIdentifiers( //
 104     ) {
 105         List<String> result = new ArrayList<>();
 106         for (InfoType eiType : this.eiTypes.getAllInfoTypes()) {
 107             result.add(eiType.getId());
 108         }
 109
 110         return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
 111     }
 112
 113     @GetMapping(path = "/eitypes/{eiTypeId}", produces = MediaType.APPLICATION_JSON_VALUE)
 114     @Operation(summary = "Individual EI type", description = "")
 115     @ApiResponses(
 116         value = { //
 117             @ApiResponse(
 118                 responseCode = "200",
 119                 description = "EI type", //
 120                 content = @Content(schema = @Schema(implementation = A1eEiTypeInfo.class))), //
 121             @ApiResponse(
 122                 responseCode = "404",
 123                 description = "Enrichment Information type is not found", //
 124                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 125         })
 126     public ResponseEntity<Object> getEiType( //
 127         @PathVariable("eiTypeId") String eiTypeId) {
 128         try {
 129             this.eiTypes.getType(eiTypeId); // Make sure that the type exists
 130             A1eEiTypeInfo info = toEiTypeInfo();
 131             return new ResponseEntity<>(gson.toJson(info), HttpStatus.OK);
 132         } catch (Exception e) {
 133             return ErrorResponse.create(e, HttpStatus.NOT_FOUND);
 134         }
 135     }
 136
 137     @GetMapping(path = "/eijobs", produces = MediaType.APPLICATION_JSON_VALUE)
 138     @Operation(summary = "EI job identifiers", description = "query for EI job identifiers")
 139     @ApiResponses(
 140         value = { //
 141             @ApiResponse(
 142                 responseCode = "200",
 143                 description = "EI job identifiers", //
 144                 content = @Content(array = @ArraySchema(schema = @Schema(implementation = String.class)))),
 145             @ApiResponse(
 146                 responseCode = "404",
 147                 description = "Enrichment Information type is not found", //
 148                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 149         })
 150     public ResponseEntity<Object> getEiJobIds( //
 151         @Parameter(
 152             name = A1eConsts.EI_TYPE_ID_PARAM,
 153             required = false, //
 154             description = A1eConsts.EI_TYPE_ID_PARAM_DESCRIPTION) //
 155         @RequestParam(name = A1eConsts.EI_TYPE_ID_PARAM, required = false) String eiTypeId,
 156         @Parameter(
 157             name = A1eConsts.OWNER_PARAM,
 158             required = false, //
 159             description = A1eConsts.OWNER_PARAM_DESCRIPTION) //
 160         @RequestParam(name = A1eConsts.OWNER_PARAM, required = false) String owner) {
 161         try {
 162             List<String> result = new ArrayList<>();
 163             if (owner != null) {
 164                 for (InfoJob job : this.eiJobs.getJobsForOwner(owner)) {
 165                     if (eiTypeId == null || job.getTypeId().equals(eiTypeId)) {
 166                         result.add(job.getId());
 167                     }
 168                 }
 169             } else if (eiTypeId != null) {
 170                 this.eiJobs.getJobsForType(eiTypeId).forEach(job -> result.add(job.getId()));
 171             } else {
 172                 this.eiJobs.getJobs().forEach(job -> result.add(job.getId()));
 173             }
 174             return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
 175         } catch (
 176
 177         Exception e) {
 178             return ErrorResponse.create(e, HttpStatus.NOT_FOUND);
 179         }
 180     }
 181
 182     @GetMapping(path = "/eijobs/{eiJobId}", produces = MediaType.APPLICATION_JSON_VALUE) //
 183     @Operation(summary = "Individual EI job", description = "") //
 184     @ApiResponses(
 185         value = { //
 186             @ApiResponse(
 187                 responseCode = "200",
 188                 description = "EI job", //
 189                 content = @Content(schema = @Schema(implementation = A1eEiJobInfo.class))), //
 190             @ApiResponse(
 191                 responseCode = "404",
 192                 description = "Enrichment Information job is not found", //
 193                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 194         })
 195     public ResponseEntity<Object> getIndividualEiJob( //
 196         @PathVariable("eiJobId") String eiJobId) {
 197         try {
 198             InfoJob job = this.eiJobs.getJob(eiJobId);
 199             return new ResponseEntity<>(gson.toJson(toEiJobInfo(job)), HttpStatus.OK);
 200         } catch (Exception e) {
 201             return ErrorResponse.create(e, HttpStatus.NOT_FOUND);
 202         }
 203     }
 204
 205     @GetMapping(path = "/eijobs/{eiJobId}/status", produces = MediaType.APPLICATION_JSON_VALUE)
 206     @Operation(summary = "EI job status", description = "")
 207     @ApiResponses(
 208         value = { //
 209             @ApiResponse(
 210                 responseCode = "200",
 211                 description = "EI job status", //
 212                 content = @Content(schema = @Schema(implementation = A1eEiJobStatus.class))), //
 213             @ApiResponse(
 214                 responseCode = "404",
 215                 description = "Enrichment Information job is not found", //
 216                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 217         })
 218     public ResponseEntity<Object> getEiJobStatus( //
 219         @PathVariable("eiJobId") String eiJobId) {
 220         try {
 221             InfoJob job = this.eiJobs.getJob(eiJobId);
 222             return new ResponseEntity<>(gson.toJson(toEiJobStatus(job)), HttpStatus.OK);
 223         } catch (Exception e) {
 224             return ErrorResponse.create(e, HttpStatus.NOT_FOUND);
 225         }
 226     }
 227
 228     private A1eEiJobStatus toEiJobStatus(InfoJob job) {
 229         return this.infoProducers.isJobEnabled(job) ? new A1eEiJobStatus(A1eEiJobStatus.EiJobStatusValues.ENABLED)
 230             : new A1eEiJobStatus(A1eEiJobStatus.EiJobStatusValues.DISABLED);
 231
 232     }
 233
 234     @DeleteMapping(path = "/eijobs/{eiJobId}", produces = MediaType.APPLICATION_JSON_VALUE)
 235     @Operation(summary = "Individual EI job", description = "")
 236     @ApiResponses(
 237         value = { //
 238             @ApiResponse(
 239                 responseCode = "200",
 240                 description = "Not used", //
 241                 content = @Content(schema = @Schema(implementation = VoidResponse.class))),
 242             @ApiResponse(
 243                 responseCode = "204",
 244                 description = "Job deleted", //
 245                 content = @Content(schema = @Schema(implementation = VoidResponse.class))), //
 246             @ApiResponse(
 247                 responseCode = "404",
 248                 description = "Enrichment Information job is not found", //
 249                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 250         })
 251     public ResponseEntity<Object> deleteIndividualEiJob( //
 252         @PathVariable("eiJobId") String eiJobId) {
 253         try {
 254             InfoJob job = this.eiJobs.getJob(eiJobId);
 255             this.eiJobs.remove(job, this.infoProducers);
 256             return new ResponseEntity<>(HttpStatus.NO_CONTENT);
 257         } catch (Exception e) {
 258             return ErrorResponse.create(e, HttpStatus.NOT_FOUND);
 259         }
 260     }
 261
 262     @PutMapping(
 263         path = "/eijobs/{eiJobId}", //
 264         produces = MediaType.APPLICATION_JSON_VALUE, //
 265         consumes = MediaType.APPLICATION_JSON_VALUE)
 266     @Operation(summary = "Individual EI job", description = "")
 267     @ApiResponses(
 268         value = { //
 269             @ApiResponse(
 270                 responseCode = "201",
 271                 description = "Job created", //
 272                 content = @Content(schema = @Schema(implementation = VoidResponse.class))), //
 273             @ApiResponse(
 274                 responseCode = "200",
 275                 description = "Job updated", //
 276                 content = @Content(schema = @Schema(implementation = VoidResponse.class))), //
 277             @ApiResponse(
 278                 responseCode = "404",
 279                 description = "Enrichment Information type is not found", //
 280                 content = @Content(schema = @Schema(implementation = ErrorResponse.ErrorInfo.class))) //
 281         })
 282     public Mono<ResponseEntity<Object>> putIndividualEiJob( //
 283         @PathVariable("eiJobId") String eiJobId, //
 284         @RequestBody A1eEiJobInfo eiJobObject) {
 285
 286         final boolean isNewJob = this.eiJobs.get(eiJobId) == null;
 287
 288         return validatePutEiJob(eiJobId, eiJobObject) //
 289             .flatMap(this::startEiJob) //
 290             .doOnNext(newEiJob -> this.eiJobs.put(newEiJob)) //
 291             .flatMap(newEiJob -> Mono.just(new ResponseEntity<>(isNewJob ? HttpStatus.CREATED : HttpStatus.OK)))
 292             .onErrorResume(throwable -> Mono.just(ErrorResponse.create(throwable, HttpStatus.NOT_FOUND)));
 293     }
 294
 295     private Mono<InfoJob> startEiJob(InfoJob newEiJob) {
 296         return this.producerCallbacks.startInfoSubscriptionJob(newEiJob, infoProducers) //
 297             .doOnNext(noOfAcceptingProducers -> this.logger.debug(
 298                 "Started EI job {}, number of activated producers: {}", newEiJob.getId(), noOfAcceptingProducers)) //
 299             .flatMap(noOfAcceptingProducers -> Mono.just(newEiJob));
 300     }
 301
 302     private Mono<InfoJob> validatePutEiJob(String eiJobId, A1eEiJobInfo eiJobInfo) {
 303         try {
 304             InfoType eiType = this.eiTypes.getType(eiJobInfo.eiTypeId);
 305             validateJsonObjectAgainstSchema(eiType.getJobDataSchema(), eiJobInfo.jobDefinition);
 306             InfoJob existingEiJob = this.eiJobs.get(eiJobId);
 307             validateUri(eiJobInfo.jobResultUri);
 308             validateUri(eiJobInfo.statusNotificationUri);
 309
 310             if (existingEiJob != null && !existingEiJob.getTypeId().equals(eiJobInfo.eiTypeId)) {
 311                 throw new ServiceException("Not allowed to change type for existing EI job", HttpStatus.CONFLICT);
 312             }
 313             return Mono.just(toEiJob(eiJobInfo, eiJobId, eiType));
 314         } catch (Exception e) {
 315             return Mono.error(e);
 316         }
 317     }
 318
 319     private void validateUri(String url) throws URISyntaxException, ServiceException {
 320         if (url != null && !url.isEmpty()) {
 321             URI uri = new URI(url);
 322             if (!uri.isAbsolute()) {
 323                 throw new ServiceException("URI: " + url + " is not absolute", HttpStatus.CONFLICT);
 324             }
 325         }
 326     }
 327
 328     private void validateJsonObjectAgainstSchema(Object schemaObj, Object object) throws ServiceException {
 329         if (schemaObj != null) { // schema is optional for now
 330             try {
 331                 ObjectMapper mapper = new ObjectMapper();
 332
 333                 String schemaAsString = mapper.writeValueAsString(schemaObj);
 334                 JSONObject schemaJSON = new JSONObject(schemaAsString);
 335                 var schema = org.everit.json.schema.loader.SchemaLoader.load(schemaJSON);
 336
 337                 String objectAsString = mapper.writeValueAsString(object);
 338                 JSONObject json = new JSONObject(objectAsString);
 339                 schema.validate(json);
 340             } catch (Exception e) {
 341                 throw new ServiceException("Json validation failure " + e.toString(), HttpStatus.CONFLICT);
 342             }
 343         }
 344     }
 345
 346     private InfoJob toEiJob(A1eEiJobInfo info, String id, InfoType type) {
 347         return InfoJob.builder() //
 348             .id(id) //
 349             .typeId(type.getId()) //
 350             .owner(info.owner) //
 351             .jobData(info.jobDefinition) //
 352             .targetUrl(info.jobResultUri) //
 353             .jobStatusUrl(info.statusNotificationUri == null ? "" : info.statusNotificationUri) //
 354             .build();
 355     }
 356
 357     private A1eEiTypeInfo toEiTypeInfo() {
 358         return new A1eEiTypeInfo();
 359     }
 360
 361     private A1eEiJobInfo toEiJobInfo(InfoJob s) {
 362         return new A1eEiJobInfo(s.getTypeId(), s.getJobData(), s.getOwner(), s.getTargetUrl(), s.getJobStatusUrl());
 363     }
 364 }