Cache leaf lookups

[mdsal.git] / binding / mdsal-binding-dom-adapter / src / main / java / org / opendaylight / mdsal / binding / dom / adapter / query / DefaultQueryFactory.java

diff --git a/binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/query/DefaultQueryFactory.java b/binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/query/DefaultQueryFactory.java
index 29e0ad62f39becd19c9f9e42f3e1f48fb33d4c9e..c254625711fcfb99f831e943b464e0024c6d6f51 100644 (file)
--- a/binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/query/DefaultQueryFactory.java
+++ b/binding/mdsal-binding-dom-adapter/src/main/java/org/opendaylight/mdsal/binding/dom/adapter/query/DefaultQueryFactory.java
@@ -10,21 +10,75 @@ package org.opendaylight.mdsal.binding.dom.adapter.query;
 import static java.util.Objects.requireNonNull;
 
 import com.google.common.annotations.Beta;
+import com.google.common.base.Throwables;
+import com.google.common.cache.CacheBuilder;
+import com.google.common.cache.CacheLoader;
+import com.google.common.cache.LoadingCache;
 import java.util.ServiceLoader;
+import java.util.concurrent.ExecutionException;
 import javax.inject.Inject;
 import javax.inject.Singleton;
+import org.eclipse.jdt.annotation.NonNull;
 import org.kohsuke.MetaInfServices;
 import org.opendaylight.mdsal.binding.api.query.DescendantQueryBuilder;
 import org.opendaylight.mdsal.binding.api.query.QueryFactory;
+import org.opendaylight.mdsal.binding.api.query.QueryStructureException;
 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
+import org.opendaylight.mdsal.binding.dom.codec.spi.BindingSchemaMapping;
 import org.opendaylight.yangtools.yang.binding.DataObject;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
+import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 @Beta
 @MetaInfServices
 @Singleton
 public final class DefaultQueryFactory implements QueryFactory {
-    private final BindingCodecTree codec;
+    private static final class MethodId {
+        final DataNodeContainer parent;
+        final String methodName;
+
+        MethodId(final DataNodeContainer parent, final String methodName) {
+            this.parent = requireNonNull(parent);
+            this.methodName = requireNonNull(methodName);
+        }
+
+        @Override
+        public int hashCode() {
+            return parent.hashCode() * 31 + methodName.hashCode();
+        }
+
+        @Override
+        public boolean equals(final Object obj) {
+            if (!(obj instanceof MethodId)) {
+                return false;
+            }
+            final MethodId other = (MethodId) obj;
+            return methodName.equals(other.methodName) && parent.equals(other.parent);
+        }
+    }
+
+    private static final Logger LOG = LoggerFactory.getLogger(DefaultQueryFactory.class);
+
+    private final LoadingCache<MethodId, NodeIdentifier> knownMethods = CacheBuilder.newBuilder().build(
+        new CacheLoader<MethodId, NodeIdentifier>() {
+            @Override
+            public NodeIdentifier load(final MethodId key) {
+                final DataNodeContainer parent = key.parent;
+                final String methodName = key.methodName;
+
+                for (DataSchemaNode child : parent.getChildNodes()) {
+                    if (methodName.equals(BindingSchemaMapping.getGetterMethodName(child))) {
+                        return NodeIdentifier.create(child.getQName());
+                    }
+                }
+                throw new QueryStructureException("Failed to find schema matching " + methodName + " in " + parent);
+            }
+        });
+    private final @NonNull BindingCodecTree codec;
 
     public DefaultQueryFactory() {
         this(ServiceLoader.load(BindingCodecTree.class).findFirst().orElseThrow());
@@ -37,6 +91,21 @@ public final class DefaultQueryFactory implements QueryFactory {
 
     @Override
     public <T extends DataObject> DescendantQueryBuilder<T> querySubtree(final InstanceIdentifier<T> rootPath) {
-        return new DefaultDescendantQueryBuilder<>(codec, rootPath);
+        return new DefaultDescendantQueryBuilder<>(this, rootPath);
+    }
+
+    @NonNull BindingCodecTree codec() {
+        return codec;
+    }
+
+    @NonNull NodeIdentifier findChild(final DataNodeContainer parent, final String methodName) {
+        try {
+            return knownMethods.get(new MethodId(parent, methodName));
+        } catch (ExecutionException e) {
+            LOG.debug("Failed to find method for {}", methodName, e);
+            final Throwable cause = e.getCause();
+            Throwables.throwIfUnchecked(cause);
+            throw new IllegalStateException("Failed to load cache", e);
+        }
     }
 }