Checkstyle: fix issues and enforce on inmemorydb

[lispflowmapping.git] / mappingservice / inmemorydb / src / main / java / org / opendaylight / lispflowmapping / inmemorydb / HashMapDb.java

diff --git a/mappingservice/inmemorydb/src/main/java/org/opendaylight/lispflowmapping/inmemorydb/HashMapDb.java b/mappingservice/inmemorydb/src/main/java/org/opendaylight/lispflowmapping/inmemorydb/HashMapDb.java
index 3f58480ca2c2c3ee673411435bfa747a249f8bc8..aea004f188463ab9aa8bfb5717cabd5ed96784cb 100644 (file)
--- a/mappingservice/inmemorydb/src/main/java/org/opendaylight/lispflowmapping/inmemorydb/HashMapDb.java
+++ b/mappingservice/inmemorydb/src/main/java/org/opendaylight/lispflowmapping/inmemorydb/HashMapDb.java
@@ -8,26 +8,62 @@
 
 package org.opendaylight.lispflowmapping.inmemorydb;
 
+import java.util.AbstractMap.SimpleImmutableEntry;
 import java.util.Date;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentMap;
 import java.util.concurrent.TimeUnit;
-
+import org.opendaylight.lispflowmapping.inmemorydb.radixtrie.RadixTrie;
 import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
 import org.opendaylight.lispflowmapping.interfaces.dao.IRowVisitor;
 import org.opendaylight.lispflowmapping.interfaces.dao.MappingEntry;
 import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
+import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4PrefixBinary;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv6PrefixBinary;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.lfm.mappingservice.dao.inmemorydb.config.rev151007.InmemorydbConfig;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 public class HashMapDb implements ILispDAO, AutoCloseable {
 
     protected static final Logger LOG = LoggerFactory.getLogger(HashMapDb.class);
-    private ConcurrentMap<Object, ConcurrentMap<String, Object>> data = new ConcurrentHashMap<Object, ConcurrentMap<String, Object>>();
+    private static final Object TABLES = (Object) "tables";
+    private static final int DEFAULT_RECORD_TIMEOUT = 240;   // SB registered mapping entries time out after 4 min
+    private ConcurrentMap<Object, ConcurrentMap<String, Object>> data = new ConcurrentHashMap<>();
     private TimeUnit timeUnit = TimeUnit.SECONDS;
-    private int recordTimeOut = 240;
-    private final Object TABLES = (Object) "tables";
+    private int recordTimeOut;
+
+    public HashMapDb() {
+        this.recordTimeOut = DEFAULT_RECORD_TIMEOUT;
+    }
+
+    public HashMapDb(final InmemorydbConfig inmemoryConfig) {
+        if (inmemoryConfig.getRecordTimeout() <= 0) {
+            this.recordTimeOut = DEFAULT_RECORD_TIMEOUT;
+        } else {
+            this.recordTimeOut = inmemoryConfig.getRecordTimeout();
+        }
+    }
+
+    // IPv4 and IPv6 radix tries used for longest prefix matching
+    private RadixTrie<Object> ip4Trie = new RadixTrie<>(32, true);
+    private RadixTrie<Object> ip6Trie = new RadixTrie<>(128, true);
+
+    public void tryAddToIpTrie(Object key) {
+        if (key instanceof Eid) {
+            Eid eid = (Eid) key;
+            if (eid.getAddress() instanceof Ipv4PrefixBinary) {
+                Ipv4PrefixBinary prefix = (Ipv4PrefixBinary) eid.getAddress();
+                ip4Trie.insert(prefix.getIpv4AddressBinary().getValue(), prefix.getIpv4MaskLength(), key);
+            } else if (eid.getAddress() instanceof Ipv6PrefixBinary) {
+                Ipv6PrefixBinary prefix = (Ipv6PrefixBinary) eid.getAddress();
+                ip6Trie.insert(prefix.getIpv6AddressBinary().getValue(), prefix.getIpv6MaskLength(), key);
+            }
+        }
+    }
 
     @Override
     public void put(Object key, MappingEntry<?>... values) {
@@ -35,6 +71,7 @@ public class HashMapDb implements ILispDAO, AutoCloseable {
             data.put(key, new ConcurrentHashMap<String, Object>());
         }
         for (MappingEntry<?> entry : values) {
+            tryAddToIpTrie(key);
             data.get(key).put(entry.getKey(), entry.getValue());
         }
     }
@@ -53,6 +90,52 @@ public class HashMapDb implements ILispDAO, AutoCloseable {
         return data.get(key);
     }
 
+    @Override
+    public Map<String, Object> getBest(Object key) {
+        if (key instanceof Eid) {
+            Eid eid = (Eid) key;
+            RadixTrie<Object>.TrieNode node = null;
+
+            if (eid.getAddress() instanceof Ipv4PrefixBinary) {
+                Ipv4PrefixBinary prefix = (Ipv4PrefixBinary) eid.getAddress();
+                node = ip4Trie.lookupBest(prefix.getIpv4AddressBinary().getValue(), prefix.getIpv4MaskLength());
+            } else if (eid.getAddress() instanceof Ipv6PrefixBinary) {
+                Ipv6PrefixBinary prefix = (Ipv6PrefixBinary) eid.getAddress();
+                node = ip6Trie.lookupBest(prefix.getIpv6AddressBinary().getValue(), prefix.getIpv6MaskLength());
+            }
+            if (node == null) {
+                return get(key);
+            }
+            return get(node.data());
+        }
+        return null;
+    }
+
+    @Override
+    public SimpleImmutableEntry<Eid, Map<String, ?>> getBestPair(Object key) {
+        Map<String, ?> data = null;
+
+        if (key instanceof Eid) {
+            Eid eid = (Eid) key;
+            RadixTrie<Object>.TrieNode node = null;
+
+            if (eid.getAddress() instanceof Ipv4PrefixBinary) {
+                Ipv4PrefixBinary prefix = (Ipv4PrefixBinary) eid.getAddress();
+                node = ip4Trie.lookupBest(prefix.getIpv4AddressBinary().getValue(), prefix.getIpv4MaskLength());
+            } else if (eid.getAddress() instanceof Ipv6PrefixBinary) {
+                Ipv6PrefixBinary prefix = (Ipv6PrefixBinary) eid.getAddress();
+                node = ip6Trie.lookupBest(prefix.getIpv6AddressBinary().getValue(), prefix.getIpv6MaskLength());
+            }
+            if (node == null) {
+                data = get(key);
+                return (data == null) ? null : new SimpleImmutableEntry<>((Eid)key, data);
+            }
+            data = get(node.data());
+            return (data == null) ? null : new SimpleImmutableEntry<>((Eid)node.data(), data);
+        }
+        return null;
+    }
+
     @Override
     public void getAll(IRowVisitor visitor) {
         for (ConcurrentMap.Entry<Object, ConcurrentMap<String, Object>> keyEntry : data.entrySet()) {
@@ -62,8 +145,42 @@ public class HashMapDb implements ILispDAO, AutoCloseable {
         }
     }
 
+    @Override
+    public Eid getWidestNegativePrefix(Eid key) {
+        RadixTrie<Object>.TrieNode node = null;
+
+        if (key.getAddress() instanceof Ipv4PrefixBinary) {
+            Ipv4PrefixBinary prefix = (Ipv4PrefixBinary) key.getAddress();
+            node = ip4Trie.lookupWidestNegative(prefix.getIpv4AddressBinary().getValue(), prefix.getIpv4MaskLength());
+            if (node != null) {
+                return LispAddressUtil.asIpv4PrefixBinaryEid(key, node.prefix(), (short) node.prefixLength());
+            }
+        } else if (key.getAddress() instanceof Ipv6PrefixBinary) {
+            Ipv6PrefixBinary prefix = (Ipv6PrefixBinary) key.getAddress();
+            node = ip6Trie.lookupWidestNegative(prefix.getIpv6AddressBinary().getValue(), prefix.getIpv6MaskLength());
+            if (node != null) {
+                return LispAddressUtil.asIpv6PrefixBinaryEid(key, node.prefix(), (short) node.prefixLength());
+            }
+        }
+        return null;
+    }
+
+    private void tryRemoveFromTrie(Object key) {
+        if (key instanceof Eid) {
+            Eid eid = (Eid) key;
+            if (eid.getAddress() instanceof Ipv4PrefixBinary) {
+                Ipv4PrefixBinary prefix = (Ipv4PrefixBinary) eid.getAddress();
+                ip4Trie.remove(prefix.getIpv4AddressBinary().getValue(), prefix.getIpv4MaskLength());
+            } else if (eid.getAddress() instanceof Ipv6PrefixBinary) {
+                Ipv6PrefixBinary prefix = (Ipv6PrefixBinary) eid.getAddress();
+                ip6Trie.remove(prefix.getIpv6AddressBinary().getValue(), prefix.getIpv6MaskLength());
+            }
+        }
+    }
+
     @Override
     public void remove(Object key) {
+        tryRemoveFromTrie(key);
         data.remove(key);
     }
 
@@ -76,6 +193,8 @@ public class HashMapDb implements ILispDAO, AutoCloseable {
 
     @Override
     public void removeAll() {
+        ip4Trie.removeAll();
+        ip6Trie.removeAll();
         data.clear();
     }
 
@@ -92,7 +211,8 @@ public class HashMapDb implements ILispDAO, AutoCloseable {
             }
 
             private boolean isExpired(Date date) {
-                return System.currentTimeMillis() - date.getTime() > TimeUnit.MILLISECONDS.convert(recordTimeOut, timeUnit);
+                return System.currentTimeMillis() - date.getTime()
+                        > TimeUnit.MILLISECONDS.convert(recordTimeOut, timeUnit);
             }
         });
     }