VOL-2620 Optimize igmpProxy app by directly using SADIS
Change-Id: I2c67f6162d446de1fb773fe311a93972f3aa582a
diff --git a/src/main/java/org/opencord/igmpproxy/StateMachine.java b/src/main/java/org/opencord/igmpproxy/StateMachine.java
index 391bdee..1097560 100644
--- a/src/main/java/org/opencord/igmpproxy/StateMachine.java
+++ b/src/main/java/org/opencord/igmpproxy/StateMachine.java
@@ -18,6 +18,8 @@
import com.google.common.collect.Maps;
import org.onlab.packet.Ip4Address;
import org.onosproject.net.DeviceId;
+import org.onosproject.net.PortNumber;
+
import java.util.Map;
import java.util.Set;
@@ -52,11 +54,11 @@
map.remove(getId(devId, groupIp));
}
- public static boolean join(DeviceId devId, Ip4Address groupIp, Ip4Address srcIP) {
+ public static boolean join(DeviceId devId, Ip4Address groupIp, Ip4Address srcIP, PortNumber upLinkPort) {
SingleStateMachine machine = get(devId, groupIp);
if (null == machine) {
- machine = new SingleStateMachine(devId, groupIp, srcIP);
+ machine = new SingleStateMachine(devId, groupIp, srcIP, upLinkPort);
map.put(getId(devId, groupIp), machine);
boolean shouldSendJoin = true;