SEBA-640 IgmpProxy should use distributed storage infrastructure of ONOS
Change-Id: I4b1c4d326a5501e9c0e046e3ee8d973ca5f73d70
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/AbstractGroupMemberStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/AbstractGroupMemberStore.java
new file mode 100644
index 0000000..9a3af33
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/AbstractGroupMemberStore.java
@@ -0,0 +1,72 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+import org.onosproject.store.AbstractStore;
+import org.opencord.igmpproxy.GroupMemberId;
+
+import java.util.Collection;
+import java.util.Map;
+import java.util.Set;
+
+/**
+ * Abstract implementation of group-member store.
+ */
+public abstract class AbstractGroupMemberStore extends AbstractStore<GroupMemberEvent, GroupMemberStoreDelegate>
+ implements GroupMemberStore {
+
+ protected Map<GroupMemberId, GroupMember> groupMemberMap;
+
+ protected AbstractGroupMemberStore() {
+
+ }
+
+ protected AbstractGroupMemberStore(Map<GroupMemberId, GroupMember> groupMemberMap) {
+ this.groupMemberMap = groupMemberMap;
+ }
+
+ @Override
+ public GroupMember putGroupMember(GroupMember groupMember) {
+ return groupMemberMap.put(groupMember.getGroupMemberId(), groupMember);
+ }
+
+ @Override
+ public GroupMember updateGroupMember(GroupMember groupMember) {
+ return groupMemberMap.replace(groupMember.getGroupMemberId(), groupMember);
+ }
+
+ @Override
+ public GroupMember removeGroupMember(GroupMemberId groupMemberId) {
+ return groupMemberMap.remove(groupMemberId);
+ }
+
+ @Override
+ public GroupMember getGroupMember(GroupMemberId groupMemberId) {
+ return groupMemberMap.getOrDefault(groupMemberId, null);
+ }
+
+ @Override
+ public Set<GroupMemberId> getAllGroupMemberIds() {
+ return ImmutableSet.copyOf(groupMemberMap.keySet());
+ }
+
+ @Override
+ public Collection<GroupMember> getAllGroupMembers() {
+ return ImmutableList.copyOf(groupMemberMap.values());
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/DistributedGroupMemberStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/DistributedGroupMemberStore.java
new file mode 100644
index 0000000..01f9af9
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/DistributedGroupMemberStore.java
@@ -0,0 +1,72 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import org.onlab.util.KryoNamespace;
+import org.onosproject.store.serializers.KryoNamespaces;
+import org.onosproject.store.service.ConsistentMap;
+import org.onosproject.store.service.Serializer;
+import org.onosproject.store.service.StorageService;
+import org.opencord.igmpproxy.GroupMemberId;
+import org.opencord.igmpproxy.GroupMemberIdSerializer;
+import org.osgi.service.component.annotations.Component;
+import org.osgi.service.component.annotations.ReferenceCardinality;
+import org.osgi.service.component.annotations.Reference;
+import org.osgi.service.component.annotations.Activate;
+import org.osgi.service.component.annotations.Deactivate;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Group member store based on distributed storage.
+ */
+@Component(service = GroupMemberStore.class)
+public class DistributedGroupMemberStore extends AbstractGroupMemberStore {
+ private final Logger log = LoggerFactory.getLogger(getClass());
+ private static final String GROUP_MEMBER_MAP_NAME = "onos-igmpproxy-groupmember-table";
+
+ @Reference(cardinality = ReferenceCardinality.MANDATORY)
+ protected StorageService storageService;
+
+ private ConsistentMap<GroupMemberId, GroupMember> consistentMap;
+
+ public DistributedGroupMemberStore() {
+ super();
+ }
+
+ @Activate
+ public void activate() {
+ KryoNamespace groupMemberSerializer = KryoNamespace.newBuilder()
+ .register(KryoNamespaces.API)
+ .register(new GroupMemberIdSerializer(), GroupMemberId.class)
+ .register(GroupMember.class)
+ .build();
+ consistentMap = storageService.<GroupMemberId, GroupMember>consistentMapBuilder()
+ .withName(GROUP_MEMBER_MAP_NAME)
+ .withSerializer(Serializer.using(groupMemberSerializer))
+ .build();
+ groupMemberMap = consistentMap.asJavaMap();
+ log.info("Started.");
+ }
+
+ @Deactivate
+ public void deactivate() {
+ groupMemberMap.clear();
+ groupMemberMap = null;
+ consistentMap.destroy();
+ log.info("Stopped.");
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMember.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMember.java
new file mode 100644
index 0000000..0add46c
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMember.java
@@ -0,0 +1,215 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import org.onlab.packet.IGMPMembership;
+import org.onlab.packet.Ip4Address;
+import org.onlab.packet.VlanId;
+import org.onosproject.net.DeviceId;
+import org.onosproject.net.PortNumber;
+import org.opencord.igmpproxy.GroupMemberId;
+
+import java.util.ArrayList;
+import java.util.Iterator;
+
+/**
+ * Date struct to keep Igmp member infomations.
+ */
+public final class GroupMember {
+
+ private final VlanId vlan;
+ private final GroupMemberId groupMemberId;
+ private final boolean v2;
+ private byte recordType = IGMPMembership.MODE_IS_INCLUDE;
+ private ArrayList<Ip4Address> sourceList = new ArrayList<>();
+ private int keepAliveQueryInterval = 0;
+ private int keepAliveQueryCount = 0;
+ private int lastQueryInterval = 0;
+ private int lastQueryCount = 0;
+ private boolean leave = false;
+
+ public GroupMember(Ip4Address groupIp, VlanId vlan, DeviceId deviceId, PortNumber portNum, boolean isV2) {
+ this.vlan = vlan;
+ this.groupMemberId = GroupMemberId.of(groupIp, deviceId, portNum);
+ v2 = isV2;
+ }
+
+ public String getkey() {
+ return groupMemberId.toString();
+ }
+
+ public GroupMemberId getGroupMemberId() {
+ return groupMemberId;
+ }
+
+ public VlanId getvlan() {
+ return vlan;
+ }
+
+ public DeviceId getDeviceId() {
+ return groupMemberId.getDeviceId();
+ }
+
+ public PortNumber getPortNumber() {
+ return groupMemberId.getPortNum();
+ }
+
+ public Ip4Address getGroupIp() {
+ return groupMemberId.getGroupIp();
+ }
+
+ public byte getRecordType() {
+ return recordType;
+ }
+
+ public boolean getv2() {
+ return v2;
+ }
+
+ public ArrayList<Ip4Address> getSourceList() {
+ return sourceList;
+ }
+
+
+ public void updateList(byte recordType, ArrayList<Ip4Address> newSourceList) {
+ this.recordType = recordType;
+ this.sourceList.clear();
+ this.sourceList.addAll(newSourceList);
+ }
+
+ /*join B to A (A+B)*/
+ private void join(ArrayList<Integer> listA, ArrayList<Integer> listB) {
+ Iterator<Integer> iterA = null;
+ Iterator<Integer> iterB = listB.iterator();
+ boolean exists;
+ while (iterB.hasNext()) {
+ iterA = listA.iterator();
+ exists = false;
+ int ipToAdd = iterB.next();
+ while (iterA.hasNext()) {
+ if (iterA.next().equals(ipToAdd)) {
+ exists = true;
+ break;
+ }
+ }
+ if (!exists) {
+ listA.add(ipToAdd);
+ }
+ }
+ }
+
+ /* include A and B (A*B)*/
+ private void intersection(ArrayList<Integer> listA, ArrayList<Integer> listB) {
+ Iterator<Integer> iterA = listA.iterator();
+ Iterator<Integer> iterB;
+ boolean exists;
+
+ while (iterA.hasNext()) {
+ iterB = listB.iterator();
+ int ipToInclude = iterA.next();
+ exists = false;
+ while (iterB.hasNext()) {
+ if (iterB.next().equals(ipToInclude)) {
+ exists = true;
+ break;
+ }
+ }
+ if (!exists) {
+ iterA.remove();
+ }
+ }
+ }
+
+ /*exclude B from A (A-B)*/
+ private void exclude(ArrayList<Integer> listA, ArrayList<Integer> listB) {
+ Iterator<Integer> iterA = null;
+ Iterator<Integer> iterB = listB.iterator();
+
+ while (iterB.hasNext()) {
+ iterA = listA.iterator();
+ int ipToDel = iterB.next();
+ while (iterA.hasNext()) {
+ if (iterA.next().equals(ipToDel)) {
+ iterA.remove();
+ break;
+ }
+ }
+ }
+ }
+
+ public void setLeave(boolean l) {
+ leave = l;
+ }
+
+ public boolean isLeave() {
+ return leave;
+ }
+
+ public int getKeepAliveQueryInterval() {
+ return keepAliveQueryInterval;
+ }
+
+ public int getKeepAliveQueryCount() {
+ return keepAliveQueryCount;
+ }
+
+ public int getLastQueryInterval() {
+ return lastQueryInterval;
+ }
+
+ public int getLastQueryCount() {
+ return lastQueryCount;
+ }
+
+ public void keepAliveQueryCount(boolean add) {
+ if (add) {
+ keepAliveQueryCount++;
+ } else {
+ keepAliveQueryCount = 0;
+ }
+ }
+
+ public void lastQueryCount(boolean add) {
+ if (add) {
+ lastQueryCount++;
+ } else {
+ lastQueryCount = 0;
+ }
+ }
+
+ public void keepAliveInterval(boolean add) {
+ if (add) {
+ keepAliveQueryInterval++;
+ } else {
+ keepAliveQueryInterval = 0;
+ }
+ }
+
+ public void lastQueryInterval(boolean add) {
+ if (add) {
+ lastQueryInterval++;
+ } else {
+ lastQueryInterval = 0;
+ }
+ }
+
+ public void resetAllTimers() {
+ keepAliveQueryInterval = 0;
+ keepAliveQueryCount = 0;
+ lastQueryInterval = 0;
+ lastQueryCount = 0;
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberEvent.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberEvent.java
new file mode 100644
index 0000000..494b89b
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberEvent.java
@@ -0,0 +1,56 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import org.onosproject.event.AbstractEvent;
+
+/**
+ * Group-member event.
+ */
+public class GroupMemberEvent extends
+ AbstractEvent<GroupMemberEvent.Type, GroupMember> {
+
+ /**
+ * Type of group-member event.
+ */
+ public enum Type {
+ /**
+ * Signifies that group-member added to store.
+ */
+ GROUP_MEMBER_ADDED,
+ /**
+ * Signifies that group-member updated in store.
+ */
+ GROUP_MEMBER_UPDATED,
+ /**
+ * Signifies that group-member has been removed from store.
+ */
+ GROUP_MEMBER_REMOVED
+ }
+
+ /**
+ *
+ * @param type group-member event type.
+ * @param subject group-member.
+ */
+ public GroupMemberEvent(GroupMemberEvent.Type type, GroupMember subject) {
+ super(type, subject);
+ }
+
+ protected GroupMemberEvent(GroupMemberEvent.Type type, GroupMember subject, long time) {
+ super(type, subject, time);
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStore.java
new file mode 100644
index 0000000..74f4ca7
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStore.java
@@ -0,0 +1,74 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import org.opencord.igmpproxy.GroupMemberId;
+
+import java.util.Collection;
+import java.util.Set;
+
+/**
+ * Group Member Store Service.
+ */
+public interface GroupMemberStore {
+
+ /**
+ * Add new group-member to store.
+ *
+ * @param groupMember given group-member.
+ * @return added group-member.
+ */
+ GroupMember putGroupMember(GroupMember groupMember);
+
+ /**
+ * Update group-member in store.
+ *
+ * @param groupMember given group-member.
+ * @return updated group-member.
+ */
+ GroupMember updateGroupMember(GroupMember groupMember);
+
+ /**
+ * removed group-member from store.
+ *
+ * @param groupMemberId given group-member id
+ * @return removed group-member
+ */
+ GroupMember removeGroupMember(GroupMemberId groupMemberId);
+
+ /**
+ * get group-member from store.
+ *
+ * @param groupMemberId given group-member identification
+ * @return group-member or null if not found
+ */
+ GroupMember getGroupMember(GroupMemberId groupMemberId);
+
+ /**
+ * Returns all group member ids.
+ *
+ * @return set of group member ids
+ */
+ Set<GroupMemberId> getAllGroupMemberIds();
+
+ /**
+ * Returns all group members.
+ *
+ * @return all group members
+ */
+ Collection<GroupMember> getAllGroupMembers();
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStoreDelegate.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStoreDelegate.java
new file mode 100644
index 0000000..e9f5509
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/GroupMemberStoreDelegate.java
@@ -0,0 +1,24 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.groupmember;
+
+import org.onosproject.store.StoreDelegate;
+
+/**
+ * Group member store delegate abstraction.
+ */
+public interface GroupMemberStoreDelegate extends StoreDelegate<GroupMemberEvent> {
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/package-info.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/package-info.java
new file mode 100644
index 0000000..7c025a6
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/groupmember/package-info.java
@@ -0,0 +1,20 @@
+/*
+ * Copyright 2017-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.
+ */
+
+/**
+ * Implementation of group-member store.
+ */
+package org.opencord.igmpproxy.impl.store.groupmember;
\ No newline at end of file
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/AbstractStateMachineStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/AbstractStateMachineStore.java
new file mode 100644
index 0000000..0ecbb14
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/AbstractStateMachineStore.java
@@ -0,0 +1,87 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.machine;
+
+import com.google.common.collect.ImmutableList;
+import org.onosproject.store.AbstractStore;
+import org.opencord.igmpproxy.statemachine.StateMachine;
+import org.opencord.igmpproxy.statemachine.StateMachineId;
+
+import java.util.Collection;
+import java.util.Map;
+
+/**
+ * Abstract implementation of state-machine store.
+ */
+public abstract class AbstractStateMachineStore
+ extends AbstractStore<StateMachineEvent, StateMachineStoreDelegate>
+ implements StateMachineStore {
+
+ protected Map<StateMachineId, StateMachine> stateMachineMap;
+
+ protected AbstractStateMachineStore() {
+ }
+
+ protected AbstractStateMachineStore(Map<StateMachineId, StateMachine> stateMachineMap) {
+ this.stateMachineMap = stateMachineMap;
+ }
+
+ @Override
+ public StateMachine putStateMachine(StateMachine stateMachine) {
+ return stateMachineMap.put(stateMachine.getStateMachineId(), stateMachine);
+ }
+
+ @Override
+ public StateMachine updateStateMachine(StateMachine machine) {
+ return stateMachineMap.replace(machine.getStateMachineId(), machine);
+ }
+
+ @Override
+ public StateMachine removeStateMachine(StateMachineId id) {
+ return stateMachineMap.remove(id);
+ }
+
+ @Override
+ public StateMachine getStateMachine(StateMachineId id) {
+ return stateMachineMap.get(id);
+ }
+
+ @Override
+ public Collection<StateMachine> getAllStateMachines() {
+ return ImmutableList.copyOf(stateMachineMap.values());
+ }
+
+ @Override
+ public void clearAllStateMachines() {
+ stateMachineMap.clear();
+ stateMachineMap = null;
+ }
+
+ @Override
+ public void decreaseTimeout(StateMachineId machineId) {
+ StateMachine machine = stateMachineMap.get(machineId);
+ machine.decreaseTimeOut();
+ updateStateMachine(machine);
+ }
+
+ @Override
+ public void increaseTimeout(StateMachineId machineId) {
+ StateMachine machine = stateMachineMap.get(machineId);
+ machine.increaseTimeOut();
+ updateStateMachine(machine);
+ }
+
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/DistributedStateMachineStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/DistributedStateMachineStore.java
new file mode 100644
index 0000000..f337b09
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/DistributedStateMachineStore.java
@@ -0,0 +1,119 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.machine;
+
+import org.onlab.util.KryoNamespace;
+import org.onosproject.store.serializers.KryoNamespaces;
+import org.onosproject.store.service.AtomicCounterMap;
+import org.onosproject.store.service.ConsistentMap;
+import org.onosproject.store.service.Serializer;
+import org.onosproject.store.service.StorageService;
+import org.opencord.igmpproxy.impl.SingleStateMachine;
+import org.opencord.igmpproxy.impl.SingleStateMachineSerializer;
+import org.opencord.igmpproxy.statemachine.StateMachine;
+import org.opencord.igmpproxy.statemachine.StateMachineId;
+import org.opencord.igmpproxy.statemachine.StateMachineIdSerializer;
+import org.osgi.service.component.annotations.Component;
+import org.osgi.service.component.annotations.ReferenceCardinality;
+import org.osgi.service.component.annotations.Reference;
+import org.osgi.service.component.annotations.Activate;
+import org.osgi.service.component.annotations.Deactivate;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * State machine store based on distributed storage.
+ */
+@Component(service = StateMachineStore.class)
+public class DistributedStateMachineStore extends AbstractStateMachineStore {
+ private final Logger log = LoggerFactory.getLogger(getClass());
+ private static final String STATE_MACHINE_COUNTER_STORE = "onos-state-machine-counter-store";
+ private static final String STATE_MACHINE_MAP_NAME = "onos-state-machine-store";
+
+ @Reference(cardinality = ReferenceCardinality.MANDATORY)
+ protected StorageService storageService;
+
+ private AtomicCounterMap<StateMachineId> stateMachineCounters;
+
+ private ConsistentMap<StateMachineId, StateMachine> consistentMap;
+
+ public DistributedStateMachineStore() {
+ super();
+ }
+
+ @Activate
+ public void activate() {
+ KryoNamespace.Builder stateMachineKryoBuilder = KryoNamespace.newBuilder()
+ .register(KryoNamespaces.API)
+ .register(new StateMachineIdSerializer(), StateMachineId.class);
+
+ stateMachineCounters = storageService.<StateMachineId>atomicCounterMapBuilder()
+ .withName(STATE_MACHINE_COUNTER_STORE)
+ .withSerializer(Serializer.using(stateMachineKryoBuilder.build())).build();
+
+ stateMachineKryoBuilder
+ .register(new SingleStateMachineSerializer(), SingleStateMachine.class);
+ this.consistentMap = storageService.<StateMachineId, StateMachine>consistentMapBuilder()
+ .withName(STATE_MACHINE_MAP_NAME)
+ .withSerializer(Serializer.using(stateMachineKryoBuilder.build()))
+ .build();
+ super.stateMachineMap = consistentMap.asJavaMap();
+
+
+ log.info("Started.");
+ }
+
+ @Deactivate
+ public void deactivate() {
+ stateMachineMap.clear();
+ stateMachineMap = null;
+ consistentMap.destroy();
+ stateMachineCounters.clear();
+ stateMachineCounters.destroy();
+ log.info("Stopped.");
+ }
+
+ @Override
+ public long increaseAndGetCounter(StateMachineId stateMachineId) {
+ return stateMachineCounters.incrementAndGet(stateMachineId);
+ }
+
+ @Override
+ public long decreaseAndGetCounter(StateMachineId stateMachineId) {
+ if (stateMachineCounters.get(stateMachineId) > 0) {
+ return stateMachineCounters.decrementAndGet(stateMachineId);
+ } else {
+ return stateMachineCounters.get(stateMachineId);
+ }
+ }
+
+ @Override
+ public long getCounter(StateMachineId stateMachineId) {
+ return stateMachineCounters.get(stateMachineId);
+ }
+
+ @Override
+ public boolean removeCounter(StateMachineId stateMachineId) {
+ stateMachineCounters.remove(stateMachineId);
+ return true;
+ }
+
+ @Override
+ public void clearAllStateMachines() {
+ super.clearAllStateMachines();
+ stateMachineCounters.clear();
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineEvent.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineEvent.java
new file mode 100644
index 0000000..ab9df23
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineEvent.java
@@ -0,0 +1,58 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.machine;
+
+import org.onosproject.event.AbstractEvent;
+import org.opencord.igmpproxy.statemachine.StateMachine;
+
+/**
+ * State machine event.
+ */
+public class StateMachineEvent extends
+ AbstractEvent<StateMachineEvent.Type, StateMachine> {
+ /**
+ * Internal state-machine event type.
+ */
+ public enum Type {
+ /**
+ * Signifies that state-machine added to store.
+ */
+ STATE_MACHINE_ADDED,
+ /**
+ * Signifies that state-machine updated in store.
+ */
+ STATE_MACHINE_UPDATED,
+ /**
+ * Signifies that state-machine removed from store.
+ */
+ STATE_MACHINE_REMOVED
+ }
+
+ /**
+ * Creates new state machine event.
+ *
+ * @param type state-machine event type.
+ * @param subject state machine.
+ */
+ public StateMachineEvent(Type type, StateMachine subject) {
+ super(type, subject);
+ }
+
+ protected StateMachineEvent(Type type, StateMachine subject, long time) {
+ super(type, subject, time);
+ }
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStore.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStore.java
new file mode 100644
index 0000000..ddb566a
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStore.java
@@ -0,0 +1,116 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.machine;
+
+import org.opencord.igmpproxy.statemachine.StateMachine;
+import org.opencord.igmpproxy.statemachine.StateMachineId;
+
+import java.util.Collection;
+
+/**
+ * State Machine Store.
+ */
+public interface StateMachineStore {
+ /**
+ * Increases counter state-machine.
+ *
+ * @param stateMachineId identification of machine
+ * @return counter
+ */
+ long increaseAndGetCounter(StateMachineId stateMachineId);
+
+ /**
+ * Decreases counter of state-machine.
+ *
+ * @param stateMachineId identification of machine
+ * @return if counter greater than zero, decreases counter and return counter
+ */
+ long decreaseAndGetCounter(StateMachineId stateMachineId);
+
+ /**
+ * Removes counter from counter-map.
+ *
+ * @param stateMachineId identification of machine
+ * @return true
+ */
+ boolean removeCounter(StateMachineId stateMachineId);
+
+ /**
+ * Get counter using state-machine id.
+ *
+ * @param stateMachineId identification of machine
+ * @return counter of member that use this id
+ */
+ long getCounter(StateMachineId stateMachineId);
+
+ /**
+ * Add new machine to store.
+ *
+ * @param stateMachine given machine
+ * @return added informations
+ */
+ StateMachine putStateMachine(StateMachine stateMachine);
+
+ /**
+ * Update information in store.
+ *
+ * @param stateMachine given machine
+ * @return updated info
+ */
+ StateMachine updateStateMachine(StateMachine stateMachine);
+
+ /**
+ * Get information from store.
+ *
+ * @param id given identification of machine
+ * @return information of machine
+ */
+ StateMachine getStateMachine(StateMachineId id);
+
+ /**
+ * Remove machine from store.
+ *
+ * @param id given identification of machine
+ * @return removed info
+ */
+ StateMachine removeStateMachine(StateMachineId id);
+
+ /**
+ * Get informations of all machine.
+ *
+ * @return machine informations
+ */
+ Collection<StateMachine> getAllStateMachines();
+
+ /**
+ * clear information map.
+ */
+ void clearAllStateMachines();
+
+ /**
+ * Decreases timeout of timer.
+ *
+ * @param machineId given machine id
+ */
+ void decreaseTimeout(StateMachineId machineId);
+
+ /**
+ * Increases timeout of timer.
+ *
+ * @param machineId given machine id
+ */
+ void increaseTimeout(StateMachineId machineId);
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStoreDelegate.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStoreDelegate.java
new file mode 100644
index 0000000..f8db089
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/StateMachineStoreDelegate.java
@@ -0,0 +1,25 @@
+/*
+ * Copyright 2017-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.igmpproxy.impl.store.machine;
+
+import org.onosproject.store.StoreDelegate;
+
+/**
+ * State machine store delegate abstraction.
+ */
+public interface StateMachineStoreDelegate extends StoreDelegate<StateMachineEvent> {
+}
diff --git a/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/package-info.java b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/package-info.java
new file mode 100644
index 0000000..2b9bc9e
--- /dev/null
+++ b/app/src/main/java/org/opencord/igmpproxy/impl/store/machine/package-info.java
@@ -0,0 +1,20 @@
+/*
+ * Copyright 2017-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.
+ */
+
+/**
+ * Implementation of state-machine store.
+ */
+package org.opencord.igmpproxy.impl.store.machine;
\ No newline at end of file