1 files changed, 74 insertions, 0 deletions
diff --git a/lisp/lisp2vpp/src/main/java/io/fd/honeycomb/lisp/translate/read/trait/MappingReader.java b/lisp/lisp2vpp/src/main/java/io/fd/honeycomb/lisp/translate/read/trait/MappingReader.java
new file mode 100644
index 000000000..2a28d822b
--- /dev/null
+++ b/lisp/lisp2vpp/src/main/java/io/fd/honeycomb/lisp/translate/read/trait/MappingReader.java
@@ -0,0 +1,74 @@
+package io.fd.honeycomb.lisp.translate.read.trait;
+
+import static com.google.common.base.Preconditions.checkNotNull;
+import static io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams.EidType.IPV4;
+import static io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams.EidType.IPV6;
+import static io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams.EidType.MAC;
+
+import io.fd.honeycomb.lisp.translate.read.dump.executor.params.MappingsDumpParams;
+import io.fd.honeycomb.translate.util.read.cache.EntityDumpExecutor;
+import io.fd.honeycomb.translate.vpp.util.JvppReplyConsumer;
+import io.fd.vpp.jvpp.core.dto.LispEidTableDetails;
+import io.fd.vpp.jvpp.core.dto.LispEidTableDetailsReplyDump;
+import io.fd.vpp.jvpp.core.dto.LispEidTableDump;
+import io.fd.vpp.jvpp.core.future.FutureJVppCore;
+import java.util.function.Predicate;
+import javax.annotation.Nonnull;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.local.mappings.LocalMapping;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.dp.subtable.grouping.remote.mappings.RemoteMapping;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.vni.table.BridgeDomainSubtable;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.lisp.rev161214.eid.table.grouping.eid.table.vni.table.VrfSubtable;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+/**
+ * Trait providing predicates to filter mappings to respective subtables
+ */
+public interface MappingReader extends JvppReplyConsumer {
+
+    Predicate<LispEidTableDetails> BRIDGE_DOMAIN_MAPPINGS_ONLY =
+            (LispEidTableDetails detail) -> detail.eidType == MAC.getValue();
+
+    Predicate<LispEidTableDetails> VRF_MAPPINGS_ONLY =
+            (LispEidTableDetails detail) -> detail.eidType == IPV4.getValue() || detail.eidType == IPV6.getValue();
+
+    default Predicate<LispEidTableDetails> subtableFilterForLocalMappings(
+            @Nonnull final InstanceIdentifier<LocalMapping> identifier) {
+
+        if (identifier.firstIdentifierOf(VrfSubtable.class) != null) {
+            return VRF_MAPPINGS_ONLY;
+        } else if (identifier.firstIdentifierOf(BridgeDomainSubtable.class) != null) {
+            return BRIDGE_DOMAIN_MAPPINGS_ONLY;
+        } else {
+            throw new IllegalArgumentException("Cannot determine mappings predicate for " + identifier);
+        }
+    }
+
+    default Predicate<LispEidTableDetails> subtableFilterForRemoteMappings(
+            @Nonnull final InstanceIdentifier<RemoteMapping> identifier) {
+
+        if (identifier.firstIdentifierOf(VrfSubtable.class) != null) {
+            return VRF_MAPPINGS_ONLY;
+        } else if (identifier.firstIdentifierOf(BridgeDomainSubtable.class) != null) {
+            return BRIDGE_DOMAIN_MAPPINGS_ONLY;
+        } else {
+            throw new IllegalArgumentException("Cannot determine mappings predicate for " + identifier);
+        }
+    }
+
+    default EntityDumpExecutor<LispEidTableDetailsReplyDump, MappingsDumpParams> createMappingDumpExecutor(
+            @Nonnull final FutureJVppCore vppApi) {
+        return (identifier, params) -> {
+            checkNotNull(params, "Params for dump request not present");
+
+            LispEidTableDump request = new LispEidTableDump();
+            request.eid = params.getEid();
+            request.eidSet = params.getEidSet();
+            request.eidType = params.getEidType();
+            request.prefixLength = params.getPrefixLength();
+            request.vni = params.getVni();
+            request.filter = params.getFilter();
+
+            return getReplyForRead(vppApi.lispEidTableDump(request).toCompletableFuture(), identifier);
+        };
+    }
+}