1 files changed, 73 insertions, 0 deletions
diff --git a/components/slice-analysis-ms/src/main/java/org/onap/slice/analysis/ms/service/MLMessageProcessor.java b/components/slice-analysis-ms/src/main/java/org/onap/slice/analysis/ms/service/MLMessageProcessor.java
new file mode 100644
index 00000000..bee7b30e
--- /dev/null
+++ b/components/slice-analysis-ms/src/main/java/org/onap/slice/analysis/ms/service/MLMessageProcessor.java
@@ -0,0 +1,73 @@
+/*******************************************************************************
+ *  ============LICENSE_START=======================================================
+ *  slice-analysis-ms
+ *  ================================================================================
+ *   Copyright (C) 2020 Wipro Limited.
+ *   ==============================================================================
+ *     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.
+ *     ============LICENSE_END=========================================================
+ *
+ *******************************************************************************/
+
+package org.onap.slice.analysis.ms.service;
+
+import java.util.List;
+import java.util.Map;
+
+import javax.annotation.PostConstruct;
+
+import org.onap.slice.analysis.ms.configdb.IConfigDbService;
+import org.onap.slice.analysis.ms.models.CUModel;
+import org.onap.slice.analysis.ms.models.MLOutputModel;
+import org.onap.slice.analysis.ms.models.policy.AdditionalProperties;
+import org.onap.slice.analysis.ms.utils.BeanUtil;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.context.annotation.Scope;
+import org.springframework.stereotype.Component;
+
+/**
+ * Process the message sent by ML service and sends notification to policy
+ */
+@Component
+@Scope("Prototype")
+public class MLMessageProcessor {
+	private static Logger log = LoggerFactory.getLogger(MLMessageProcessor.class);
+
+	private IConfigDbService configDbService;
+	private PolicyService policyService;
+
+
+	@PostConstruct
+	public void init() {
+		configDbService = BeanUtil.getBean(IConfigDbService.class);
+	}
+	
+	public void processMLMsg(MLOutputModel mlOutputMsg) {
+		String snssai = mlOutputMsg.getSnssai();
+		List<CUModel> cuData = mlOutputMsg.getData();
+		Map<String, List<String>>  ricToCellMapping = configDbService.fetchRICsOfSnssai(snssai);
+		log.debug("RIC to cell mapping of S-NSSAI {} is {}",snssai,ricToCellMapping);
+		for(CUModel cuModel: cuData) {
+			String cellId = String.valueOf(cuModel.getCellCUList().get(0).getCellLocalId());
+			ricToCellMapping.forEach((ricId, cells) -> {
+				if(cells.contains(cellId)) {
+					cuModel.setNearRTRICId(ricId);
+				}
+			});
+		}
+		AdditionalProperties<MLOutputModel> addProps = new AdditionalProperties<>();
+		addProps.setResourceConfig(mlOutputMsg);
+		policyService.sendOnsetMessageToPolicy(snssai, addProps, configDbService.fetchServiceDetails(snssai));	
+	}
+}