yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/type/LeafrefTypeEffectiveStatementImpl.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.type;
  10
  11 import javax.annotation.Nonnull;
  12 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
  13 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  14 import org.opendaylight.yangtools.yang.model.api.stmt.RequireInstanceEffectiveStatement;
  15 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
  16 import org.opendaylight.yangtools.yang.model.api.stmt.TypeStatement;
  17 import org.opendaylight.yangtools.yang.model.api.type.LeafrefTypeDefinition;
  18 import org.opendaylight.yangtools.yang.model.util.type.RequireInstanceRestrictedTypeBuilder;
  19 import org.opendaylight.yangtools.yang.model.util.type.RestrictedTypes;
  20 import org.opendaylight.yangtools.yang.parser.rfc6020.util.DeclaredEffectiveStatementBase;
  21 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  22 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.TypeUtils;
  23
  24 public final class LeafrefTypeEffectiveStatementImpl extends DeclaredEffectiveStatementBase<String, TypeStatement>
  25         implements TypeEffectiveStatement<TypeStatement> {
  26
  27     private final LeafrefTypeDefinition typeDefinition;
  28
  29     public LeafrefTypeEffectiveStatementImpl(
  30             final StmtContext<String, TypeStatement, EffectiveStatement<String, TypeStatement>> ctx,
  31             final LeafrefTypeDefinition baseType) {
  32         super(ctx);
  33
  34         final RequireInstanceRestrictedTypeBuilder<LeafrefTypeDefinition> builder =
  35                 RestrictedTypes.newLeafrefBuilder(baseType, TypeUtils.typeEffectiveSchemaPath(ctx));
  36
  37         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
  38             if (stmt instanceof RequireInstanceEffectiveStatement) {
  39                 builder.setRequireInstance(((RequireInstanceEffectiveStatement) stmt).argument());
  40             } else if (stmt instanceof UnknownSchemaNode) {
  41                 builder.addUnknownSchemaNode((UnknownSchemaNode)stmt);
  42             }
  43         }
  44
  45         typeDefinition = builder.build();
  46     }
  47
  48     @Nonnull
  49     @Override
  50     public LeafrefTypeDefinition getTypeDefinition() {
  51         return typeDefinition;
  52     }
  53 }