1 files changed, 73 insertions, 0 deletions
diff --git a/src/main/java/org/onap/a1pesimulator/service/report/RanCellEventCustomizer.java b/src/main/java/org/onap/a1pesimulator/service/report/RanCellEventCustomizer.java
new file mode 100644
index 0000000..f16e29c
--- /dev/null
+++ b/src/main/java/org/onap/a1pesimulator/service/report/RanCellEventCustomizer.java
@@ -0,0 +1,73 @@
+/*
+ * Copyright (C) 2021 Samsung Electronics
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License
+ */
+
+package org.onap.a1pesimulator.service.report;
+
+import java.util.List;
+import java.util.Optional;
+
+import org.onap.a1pesimulator.data.ves.VesEvent;
+import org.onap.a1pesimulator.data.ves.MeasurementFields.AdditionalMeasurement;
+import org.onap.a1pesimulator.service.common.EventCustomizer;
+import org.onap.a1pesimulator.service.ue.RanUeHolder;
+import org.onap.a1pesimulator.util.Constants;
+import org.onap.a1pesimulator.util.JsonUtils;
+import org.onap.a1pesimulator.util.RanVesUtils;
+import org.springframework.stereotype.Service;
+
+@Service
+public class RanCellEventCustomizer implements EventCustomizer {
+
+    private static final String UE_PARAM_TRAFFIC_MODEL_RANGE = "[[20-50]]";
+    private final RanUeHolder ranUeHolder;
+
+    public RanCellEventCustomizer(RanUeHolder ueHolder) {
+        this.ranUeHolder = ueHolder;
+    }
+
+    @Override
+    public VesEvent apply(VesEvent t) {
+        VesEvent event = JsonUtils.INSTANCE.clone(t);
+        return customizeEvent(event);
+    }
+
+    private VesEvent customizeEvent(VesEvent event) {
+        RanVesUtils.updateHeader(event);
+        enrichWithUeData(event);
+        randomizeEvent(event);
+        return event;
+    }
+
+    private void randomizeEvent(VesEvent event) {
+        List<AdditionalMeasurement> additionalMeasurementsToRandomize =
+                event.getMeasurementFields().getAdditionalMeasurements();
+        event.getMeasurementFields().setAdditionalMeasurements(
+                RanVesUtils.randomizeAdditionalMeasurements(additionalMeasurementsToRandomize));
+    }
+
+    private void enrichWithUeData(VesEvent event) {
+
+        Optional<AdditionalMeasurement> identity = event.getMeasurementFields().getAdditionalMeasurements().stream()
+                .filter(msrmnt -> Constants.MEASUREMENT_FIELD_IDENTIFIER
+                        .equalsIgnoreCase(
+                                msrmnt.getName()))
+                .findAny();
+        identity.ifPresent(m -> addTrafficModelMeasurement(event));
+    }
+
+    private void addTrafficModelMeasurement(VesEvent event) {
+        AdditionalMeasurement trafficModelMeasurement =
+                RanVesUtils.buildTrafficModelMeasurement( ranUeHolder, UE_PARAM_TRAFFIC_MODEL_RANGE);
+        event.getMeasurementFields().getAdditionalMeasurements().add(trafficModelMeasurement);
+    }
+}