2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.valid;
10 import com.google.common.base.Preconditions;
11 import java.util.HashSet;
13 import org.opendaylight.yangtools.yang.common.QName;
14 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
15 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
16 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
17 import org.opendaylight.yangtools.yang.data.impl.schema.SchemaUtils;
18 import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
19 import org.opendaylight.yangtools.yang.model.api.AugmentationTarget;
20 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
21 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
22 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
25 * General validator for container like statements, e.g. container, list-entry, choice, augment
27 public class DataNodeContainerValidator {
28 private final Set<AugmentationIdentifier> augments = new HashSet<>();
29 private final DataNodeContainer schema;
30 private final Set<QName> childNodes;
32 public DataNodeContainerValidator(final DataNodeContainer schema) {
33 this.schema = Preconditions.checkNotNull(schema, "Schema was null");
35 this.childNodes = getChildNodes(schema);
37 if (schema instanceof AugmentationTarget) {
38 for (AugmentationSchema augmentationSchema : ((AugmentationTarget) schema).getAvailableAugmentations()) {
39 augments.add(SchemaUtils.getNodeIdentifierForAugmentation(augmentationSchema));
44 private boolean isKnownChild(final PathArgument child) {
45 if (child instanceof AugmentationIdentifier) {
46 return augments.contains(child);
49 return childNodes.contains(child.getNodeType());
52 public void validateChild(final PathArgument child) {
53 DataValidationException.checkLegalChild(isKnownChild(child), child, schema, childNodes, augments);
56 public DataContainerChild<?, ?> validateChild(final DataContainerChild<?, ?> child) {
57 validateChild(child.getIdentifier());
62 * Map all direct child nodes. Skip augments since they have no qname. List cases since cases do not exist in
65 private static Set<QName> getChildNodes(final DataNodeContainer nodeContainer) {
66 Set<QName> allChildNodes = new HashSet<>();
68 for (DataSchemaNode childSchema : nodeContainer.getChildNodes()) {
69 if (childSchema instanceof ChoiceCaseNode) {
70 allChildNodes.addAll(getChildNodes((DataNodeContainer) childSchema));
71 } else if (!(childSchema instanceof AugmentationSchema)) {
72 allChildNodes.add(childSchema.getQName());