yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/leaf/AbstractLeafEffectiveStatement.java

   1 /*
   2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.leaf;
   9
  10 import static com.google.common.base.Verify.verify;
  11 import static java.util.Objects.requireNonNull;
  12
  13 import com.google.common.collect.ImmutableList;
  14 import org.eclipse.jdt.annotation.NonNull;
  15 import org.opendaylight.yangtools.yang.common.QName;
  16 import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
  17 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
  18 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  19 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
  20 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  21 import org.opendaylight.yangtools.yang.model.api.stmt.DefaultEffectiveStatement;
  22 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionEffectiveStatement;
  23 import org.opendaylight.yangtools.yang.model.api.stmt.LeafEffectiveStatement;
  24 import org.opendaylight.yangtools.yang.model.api.stmt.LeafStatement;
  25 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceEffectiveStatement;
  26 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
  27 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
  28 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
  29 import org.opendaylight.yangtools.yang.model.util.type.ConcreteTypeBuilder;
  30 import org.opendaylight.yangtools.yang.model.util.type.ConcreteTypes;
  31 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredEffectiveStatement;
  32 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.DataSchemaNodeMixin;
  33 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.MandatoryMixin;
  34 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.MustConstraintMixin;
  35
  36 abstract class AbstractLeafEffectiveStatement extends AbstractDeclaredEffectiveStatement.Default<QName, LeafStatement>
  37         implements LeafEffectiveStatement, LeafSchemaNode, DerivableSchemaNode,
  38             DataSchemaNodeMixin<QName, LeafStatement>, MandatoryMixin<QName, LeafStatement>,
  39             MustConstraintMixin<QName, LeafStatement> {
  40     // Variable: either a single substatement or an ImmutableList
  41     private final @NonNull Object substatements;
  42     private final @NonNull SchemaPath path;
  43     private final @NonNull TypeDefinition<?> type;
  44     private final int flags;
  45
  46     AbstractLeafEffectiveStatement(final LeafStatement declared, final SchemaPath path, final int flags,
  47             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
  48         super(declared);
  49         this.substatements = substatements.size() == 1 ? substatements.get(0) : substatements;
  50         this.path = requireNonNull(path);
  51         this.flags = flags;
  52         // TODO: lazy instantiation?
  53         this.type = buildType();
  54     }
  55
  56     @Override
  57     @SuppressWarnings("unchecked")
  58     public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
  59         if (substatements instanceof ImmutableList) {
  60             return (ImmutableList<? extends EffectiveStatement<?, ?>>) substatements;
  61         }
  62         verify(substatements instanceof EffectiveStatement, "Unexpected substatement %s", substatements);
  63         return ImmutableList.of((EffectiveStatement<?, ?>) substatements);
  64     }
  65
  66     @Override
  67     public final int flags() {
  68         return flags;
  69     }
  70
  71     @Override
  72     public final @NonNull QName argument() {
  73         return getQName();
  74     }
  75
  76     @Override
  77     public final @NonNull SchemaPath getPath() {
  78         return path;
  79     }
  80
  81     @Override
  82     public final TypeDefinition<?> getType() {
  83         return type;
  84     }
  85
  86     private TypeDefinition<?> buildType() {
  87         final TypeEffectiveStatement<?> typeStmt = findFirstEffectiveSubstatement(TypeEffectiveStatement.class).get();
  88         final ConcreteTypeBuilder<?> builder = ConcreteTypes.concreteTypeBuilder(typeStmt.getTypeDefinition(),
  89             getPath());
  90         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
  91             if (stmt instanceof DefaultEffectiveStatement) {
  92                 builder.setDefaultValue(((DefaultEffectiveStatement)stmt).argument());
  93             } else if (stmt instanceof DescriptionEffectiveStatement) {
  94                 builder.setDescription(((DescriptionEffectiveStatement)stmt).argument());
  95             } else if (stmt instanceof ReferenceEffectiveStatement) {
  96                 builder.setReference(((ReferenceEffectiveStatement)stmt).argument());
  97             } else if (stmt instanceof StatusEffectiveStatement) {
  98                 builder.setStatus(((StatusEffectiveStatement)stmt).argument());
  99             } else if (stmt instanceof UnitsEffectiveStatement) {
 100                 builder.setUnits(((UnitsEffectiveStatement)stmt).argument());
 101             }
 102         }
 103         return builder.build();
 104     }
 105 }