package org.opendaylight.yangtools.yang.parser.builder.util;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import java.util.Map;
import java.util.Set;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
import org.opendaylight.yangtools.yang.model.api.UsesNode;

public abstract class AbstractDocumentedDataNodeContainer extends AbstractDocumentedNode implements DataNodeContainer {

    private final Map<QName, DataSchemaNode> childNodes;
    private final Set<GroupingDefinition> groupings;
    private final Set<UsesNode> uses;
    private final Set<TypeDefinition<?>> typeDefinitions;

    protected AbstractDocumentedDataNodeContainer(final AbstractDocumentedDataNodeContainerBuilder data) {
        super(data);
        // FIXME : Should be unmodifiable ordered set (ordered by QName... or appearance in YANG file).
        // consider using new TreeSet<>(Comparators.SCHEMA_NODE_COMP);

        childNodes = ImmutableMap.copyOf(data.getChildNodes());
        groupings = ImmutableSet.copyOf(data.getGroupings());
        uses = ImmutableSet.copyOf(data.getUsesNodes());
        typeDefinitions = ImmutableSet.copyOf(data.getTypeDefinitions());
    }

    @Override
    public final Set<TypeDefinition<?>> getTypeDefinitions() {
        return typeDefinitions;
    }

    @Override
    public final Set<DataSchemaNode> getChildNodes() {
        return ImmutableSet.copyOf(childNodes.values());
    }

    @Override
    public final Set<GroupingDefinition> getGroupings() {
        return groupings;
    }

    @Override
    public final DataSchemaNode getDataChildByName(final QName name) {
        for (DataSchemaNode node : childNodes.values()) {
            if (node.getQName().equals(name)) {
                return node;
            }
        }
        return null;
    }

    @Override
    public final DataSchemaNode getDataChildByName(final String name) {
        for (DataSchemaNode node : childNodes.values()) {
            if (node.getQName().getLocalName().equals(name)) {
                return node;
            }
        }
        return null;
    }

    @Override
    public Set<UsesNode> getUses() {
        return uses;
    }

}

@@ -1,8 +1,316 @@
+/*%0A * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.%0A *%0A * This program and the accompanying materials are made available under the%0A * terms of the Eclipse Public License v1.0 which accompanies this distribution,%0A * and is available at http://www.eclipse.org/legal/epl-v10.html%0A */%0A
 package 
@@ -453,24 +453,25 @@
 mutableSet;%0A
+%0A
 import java.
@@ -476,24 +476,24 @@
 a.util.Map;%0A
-
 import java.
@@ -502,16 +502,17 @@
 il.Set;%0A
+%0A
 import o
@@ -2230,178 +2230,122 @@
     
-for (DataSchemaNode node : childNodes.values()) %7B%0A            if (node.getQName().equals(name)) %7B%0A                return node;%0A            %7D%0A        %7D%0A        return null
+// Child nodes are keyed by their container name, so we can do a direct lookup%0A        return childNodes.get(name)
 ;%0A  

Optimize AbstractDocumentedDataNodeContainer.getDataChildByName()