[SEBA-42] Operational Status Multicast Source.
Change-Id: I28595ef5a0b241f49975ad71f043bc6c9e1e84d6
diff --git a/src/main/java/org/opencord/kafka/integrations/McastKafkaIntegration.java b/src/main/java/org/opencord/kafka/integrations/McastKafkaIntegration.java
new file mode 100644
index 0000000..8712333
--- /dev/null
+++ b/src/main/java/org/opencord/kafka/integrations/McastKafkaIntegration.java
@@ -0,0 +1,119 @@
+/*
+ * Copyright 2018-present Open Networking Foundation
+ *
+ * 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.opencord.kafka.integrations;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import org.opencord.cordmcast.CordMcastStatistics;
+import org.opencord.cordmcast.CordMcastStatisticsEvent;
+import org.opencord.cordmcast.CordMcastStatisticsEventListener;
+import org.opencord.cordmcast.CordMcastStatisticsService;
+import org.opencord.kafka.EventBusService;
+import org.osgi.service.component.annotations.Activate;
+import org.osgi.service.component.annotations.Component;
+import org.osgi.service.component.annotations.Deactivate;
+import org.osgi.service.component.annotations.Reference;
+import org.osgi.service.component.annotations.ReferenceCardinality;
+import org.osgi.service.component.annotations.ReferencePolicy;
+
+import java.time.Instant;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicReference;
+
+/**
+ * Listens to Mcast events and pushes them into kafka bus.
+ */
+@Component(immediate = true)
+public class McastKafkaIntegration extends AbstractKafkaIntegration {
+
+ @Reference(cardinality = ReferenceCardinality.MANDATORY)
+ protected EventBusService eventBusService;
+
+ @Reference(cardinality = ReferenceCardinality.OPTIONAL,
+ policy = ReferencePolicy.DYNAMIC,
+ bind = "bindMcastStatisticsService",
+ unbind = "unbindMcastStatisticsService")
+ protected volatile CordMcastStatisticsService cordMcastStatisticsService;
+ protected final AtomicReference<CordMcastStatisticsService> cordMcastStatisticsServiceRef = new AtomicReference<>();
+
+ private final CordMcastStatisticsEventListener cordMcastStatisticsEventListener =
+ new InternalCorcMcastStatisticsListener();
+
+ private static final String MCAST_OPERATIONAL_STATUS_TOPIC = "mcastOperationalStatus.events";
+
+ //cord mcast stats event params
+ private static final String TIMESTAMP = "timestamp";
+ private static final String GROUP = "Group";
+ private static final String SOURCE = "Source";
+ private static final String VLAN = "Vlan";
+ private static final String MCAST_EVENT_DATA = "McastEventData";
+
+ protected void bindMcastStatisticsService(CordMcastStatisticsService cordMcastStatisticsService) {
+ bindAndAddListener(cordMcastStatisticsService, cordMcastStatisticsServiceRef, cordMcastStatisticsEventListener);
+ }
+
+ protected void unbindMcastStatisticsService(CordMcastStatisticsService cordMcastStatisticsService) {
+ unbindAndRemoveListener(cordMcastStatisticsService,
+ cordMcastStatisticsServiceRef, cordMcastStatisticsEventListener);
+ }
+
+ @Activate
+ public void activate() {
+ log.info("Started McastKafkaIntegration");
+ }
+
+ @Deactivate
+ public void deactivate() {
+ log.info("Stopped McastKafkaIntegration");
+ }
+
+ private void handleMcastStat(CordMcastStatisticsEvent mcastStatEvent) {
+ eventBusService.send(MCAST_OPERATIONAL_STATUS_TOPIC, serializeMcastStat(mcastStatEvent));
+ log.info("CordMcastStatisticsEvent sent successfully");
+ }
+
+ private JsonNode serializeMcastStat(CordMcastStatisticsEvent mcastStatEvent) {
+ log.debug("Serializing AuthenticationStatisticsEvent");
+ ObjectMapper mapper = new ObjectMapper();
+ ObjectNode mcastStat = mapper.createObjectNode();
+ mcastStat.put(TIMESTAMP, Instant.now().toString());
+ ArrayNode mcastArrayNode = mcastStat.putArray(MCAST_EVENT_DATA);
+ List<CordMcastStatistics> cordMcastStatsList = mcastStatEvent.subject();
+ cordMcastStatsList.forEach(stats -> {
+ ObjectNode mcastNode = mapper.createObjectNode();
+ if (stats.getGroupAddress() != null) {
+ mcastNode.put(GROUP, stats.getGroupAddress().toString());
+ }
+ if (stats.getSourceAddress() != null) {
+ mcastNode.put(SOURCE, stats.getSourceAddress().toString());
+ }
+ mcastNode.put(VLAN, stats.getVlanId().toShort());
+ mcastArrayNode.add(mcastNode);
+ });
+ return mcastStat;
+ }
+
+ private class InternalCorcMcastStatisticsListener implements CordMcastStatisticsEventListener {
+
+ @Override
+ public void event(CordMcastStatisticsEvent mcastStatEvent) {
+ handleMcastStat(mcastStatEvent);
+ }
+ }
+}