gerrit.onap Code Review - dcaegen2/collectors/datafile.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * Datafile Collector Service
   4  * ================================================================================
   5  * Copyright (C) 2018 NOKIA Intellectual Property. 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.dcaegen2.collectors.datafile.service;
  22
  23 import com.google.gson.JsonElement;
  24 import com.google.gson.JsonObject;
  25 import com.google.gson.JsonParser;
  26 import java.util.Optional;
  27 import java.util.stream.StreamSupport;
  28
  29 import org.onap.dcaegen2.collectors.datafile.exceptions.DmaapEmptyResponseException;
  30 import org.onap.dcaegen2.collectors.datafile.exceptions.DmaapNotFoundException;
  31 import org.onap.dcaegen2.collectors.datafile.model.ConsumerDmaapModel;
  32 import org.onap.dcaegen2.collectors.datafile.model.ImmutableConsumerDmaapModel;
  33 import org.springframework.util.StringUtils;
  34 import reactor.core.publisher.Mono;
  35
  36 /**
  37  * @author <a href="mailto:przemyslaw.wasala@nokia.com">Przemysław Wąsala</a> on 5/8/18
  38  */
  39 public class DmaapConsumerJsonParser {
  40
  41     private static final String EVENT = "event";
  42     private static final String OTHER_FIELDS = "otherFields";
  43     private static final String PNF_OAM_IPV_4_ADDRESS = "pnfOamIpv4Address";
  44     private static final String PNF_OAM_IPV_6_ADDRESS = "pnfOamIpv6Address";
  45     private static final String PNF_VENDOR_NAME = "pnfVendorName";
  46     private static final String PNF_SERIAL_NUMBER = "pnfSerialNumber";
  47
  48     /**
  49      * Extract info from string and create @see {@link org.onap.dcaegen2.collectors.datafile.model.ConsumerDmaapModel}.
  50      *
  51      * @param monoMessage - results from DMaaP
  52      * @return reactive DMaaPModel
  53      */
  54     public Mono<ConsumerDmaapModel> getJsonObject(Mono<String> monoMessage) {
  55         return monoMessage
  56             .flatMap(this::getJsonParserMessage)
  57             .flatMap(this::createJsonConsumerModel);
  58     }
  59
  60     private Mono<JsonElement> getJsonParserMessage(String message) {
  61         return StringUtils.isEmpty(message) ? Mono.error(new DmaapEmptyResponseException())
  62             : Mono.fromSupplier(() -> new JsonParser().parse(message));
  63     }
  64
  65     private Mono<ConsumerDmaapModel> createJsonConsumerModel(JsonElement jsonElement) {
  66         return jsonElement.isJsonObject()
  67             ? create(Mono.fromSupplier(jsonElement::getAsJsonObject))
  68             : getConsumerDmaapModelFromJsonArray(jsonElement);
  69     }
  70
  71     private Mono<ConsumerDmaapModel> getConsumerDmaapModelFromJsonArray(JsonElement jsonElement) {
  72         return create(
  73             Mono.fromCallable(() -> StreamSupport.stream(jsonElement.getAsJsonArray().spliterator(), false).findFirst()
  74                 .flatMap(this::getJsonObjectFromAnArray)
  75                 .orElseThrow(DmaapEmptyResponseException::new)));
  76     }
  77
  78     public Optional<JsonObject> getJsonObjectFromAnArray(JsonElement element) {
  79         return Optional.of(new JsonParser().parse(element.getAsString()).getAsJsonObject());
  80     }
  81
  82     private Mono<ConsumerDmaapModel> create(Mono<JsonObject> jsonObject) {
  83         return jsonObject.flatMap(monoJsonP ->
  84             !containsHeader(monoJsonP) ? Mono.error(new DmaapNotFoundException("Incorrect JsonObject - missing header"))
  85                 : transform(monoJsonP));
  86     }
  87
  88     private Mono<ConsumerDmaapModel> transform(JsonObject monoJsonP) {
  89         monoJsonP = monoJsonP.getAsJsonObject(EVENT).getAsJsonObject(OTHER_FIELDS);
  90         String pnfVendorName = getValueFromJson(monoJsonP, PNF_VENDOR_NAME);
  91         String pnfSerialNumber = getValueFromJson(monoJsonP, PNF_SERIAL_NUMBER);
  92         String pnfOamIpv4Address = getValueFromJson(monoJsonP, PNF_OAM_IPV_4_ADDRESS);
  93         String pnfOamIpv6Address = getValueFromJson(monoJsonP, PNF_OAM_IPV_6_ADDRESS);
  94         return
  95             (!vendorAndSerialNotEmpty(pnfSerialNumber, pnfVendorName) || !ipPropertiesNotEmpty(pnfOamIpv4Address,
  96                 pnfOamIpv6Address))
  97                 ? Mono.error(new DmaapNotFoundException("Incorrect json, consumerDmaapModel can not be created: "
  98                 + printMessage(pnfVendorName, pnfSerialNumber, pnfOamIpv4Address, pnfOamIpv6Address))) :
  99                 Mono.just(ImmutableConsumerDmaapModel.builder()
 100                     .pnfName(pnfVendorName.substring(0, Math.min(pnfVendorName.length(), 3)).toUpperCase()
 101                         .concat(pnfSerialNumber)).ipv4(pnfOamIpv4Address)
 102                     .ipv6(pnfOamIpv6Address).build());
 103     }
 104
 105     private String getValueFromJson(JsonObject jsonObject, String jsonKey) {
 106         return jsonObject.has(jsonKey) ? jsonObject.get(jsonKey).getAsString() : "";
 107     }
 108
 109     private boolean vendorAndSerialNotEmpty(String pnfSerialNumber, String pnfVendorName) {
 110         return (!StringUtils.isEmpty(pnfSerialNumber) && !StringUtils.isEmpty(pnfVendorName));
 111     }
 112
 113     private boolean ipPropertiesNotEmpty(String ipv4, String ipv6) {
 114         return (!StringUtils.isEmpty(ipv4)) || !(StringUtils.isEmpty(ipv6));
 115     }
 116
 117     private boolean containsHeader(JsonObject jsonObject) {
 118         return jsonObject.has(EVENT) && jsonObject.getAsJsonObject(EVENT).has(OTHER_FIELDS);
 119     }
 120
 121     private String printMessage(String pnfVendorName, String pnfSerialNumber, String pnfOamIpv4Address,
 122         String pnfOamIpv6Address) {
 123         return String.format("%n{"
 124             + "\"pnfVendorName\" : \"%s\","
 125             + "\"pnfSerialNumber\": \"%s\","
 126             + "\"pnfOamIpv4Address\": \"%s\","
 127             + "\"pnfOamIpv6Address\": \"%s\""
 128             + "%n}", pnfVendorName, pnfSerialNumber, pnfOamIpv4Address, pnfOamIpv6Address);
 129     }
 130 }