import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
-import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
+import org.opendaylight.yangtools.yang.data.api.schema.builder.DataContainerNodeBuilder;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.valid.DataValidationException;
import org.opendaylight.yangtools.yang.data.impl.schema.nodes.AbstractImmutableDataContainerNode;
@Override
public DataContainerNodeBuilder<AugmentationIdentifier, AugmentationNode> withChild(
- final DataContainerChild<?, ?> child) {
+ final DataContainerChild child) {
// Check nested augments
- DataValidationException.checkLegalData(!(child instanceof AugmentationNode),
- "Unable to add: %s, as a child for: %s, Nested augmentations are not permitted", child.getNodeType(),
- getNodeIdentifier() == null ? this : getNodeIdentifier());
+ if (child instanceof AugmentationNode) {
+ final AugmentationIdentifier myId = getNodeIdentifier();
+ throw new DataValidationException(String.format(
+ "Unable to add: %s, as a child for: %s, Nested augmentations are not permitted", child.getIdentifier(),
+ myId == null ? this : myId));
+ }
return super.withChild(child);
}
}
private static final class ImmutableAugmentationNode
- extends AbstractImmutableDataContainerNode<AugmentationIdentifier> implements AugmentationNode {
+ extends AbstractImmutableDataContainerNode<AugmentationIdentifier, AugmentationNode>
+ implements AugmentationNode {
ImmutableAugmentationNode(final AugmentationIdentifier nodeIdentifier,
final Map<PathArgument, Object> children) {
super(children, nodeIdentifier);
}
+
+ @Override
+ protected Class<AugmentationNode> implementedType() {
+ return AugmentationNode.class;
+ }
}
}