- public NodeCodecContext yangPathArgumentChild(final YangInstanceIdentifier.PathArgument arg) {
- final NodeContextSupplier childSupplier;
- if (arg instanceof NodeIdentifierWithPredicates) {
- childSupplier = byYang.get(new NodeIdentifier(arg.getNodeType()));
- } else if (arg instanceof AugmentationIdentifier) {
- childSupplier = yangAugmentationChild((AugmentationIdentifier) arg);
- } else {
- childSupplier = byYang.get(arg);
- }
-
- return childNonNull(childSupplier, arg, "Argument %s is not valid child of %s", arg, getSchema()).get();
- }
-
- protected final ValueNodeCodecContext getLeafChild(final String name) {
- final ValueNodeCodecContext value = leafChild.get(name);
- if (value == null) {
- throw IncorrectNestingException.create("Leaf %s is not valid for %s", name, getBindingClass());
- }
- return value;
- }
-
- private DataContainerCodecPrototype<?> loadChildPrototype(final Class<?> childClass) {
- final DataSchemaNode origDef = factory().getRuntimeContext().getSchemaDefinition(childClass);
- // Direct instantiation or use in same module in which grouping
- // was defined.
- DataSchemaNode sameName;
- try {
- sameName = getSchema().getDataChildByName(origDef.getQName());
- } catch (final IllegalArgumentException e) {
- sameName = null;
- }
- final DataSchemaNode childSchema;
- if (sameName != null) {
- // Exactly same schema node
- if (origDef.equals(sameName)) {
- childSchema = sameName;
- // We check if instantiated node was added via uses
- // statement and is instantiation of same grouping
- } else if (origDef.equals(SchemaNodeUtils.getRootOriginalIfPossible(sameName))) {
- childSchema = sameName;
- } else {
- // Node has same name, but clearly is different
- childSchema = null;
+ final CodecContextSupplier yangChildSupplier(final NodeIdentifier arg) {
+ final var child = super.yangChildSupplier(arg);
+ if (child == null) {
+ final var augClass = yangToAugmentClass.get(arg);
+ if (augClass != null) {
+ return augmentToPrototype.get(augClass);