model/yang-model-ri/src/main/java/org/opendaylight/yangtools/yang/model/ri/stmt/impl/eff/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.model.ri.stmt.impl.eff;
   9
  10 import static java.util.Objects.requireNonNull;
  11
  12 import com.google.common.collect.ImmutableList;
  13 import org.eclipse.jdt.annotation.NonNull;
  14 import org.opendaylight.yangtools.yang.common.QName;
  15 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
  16 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
  17 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  18 import org.opendaylight.yangtools.yang.model.api.stmt.DefaultEffectiveStatement;
  19 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionEffectiveStatement;
  20 import org.opendaylight.yangtools.yang.model.api.stmt.LeafEffectiveStatement;
  21 import org.opendaylight.yangtools.yang.model.api.stmt.LeafStatement;
  22 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceEffectiveStatement;
  23 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
  24 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
  25 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
  26 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypeBuilder;
  27 import org.opendaylight.yangtools.yang.model.ri.type.ConcreteTypes;
  28 import org.opendaylight.yangtools.yang.model.spi.meta.AbstractDeclaredEffectiveStatement;
  29 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.DataSchemaNodeMixin;
  30 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MandatoryMixin;
  31 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.MustConstraintMixin;
  32
  33 public abstract class AbstractLeafEffectiveStatement
  34         extends AbstractDeclaredEffectiveStatement.Default<QName, LeafStatement>
  35         implements LeafEffectiveStatement, LeafSchemaNode, DataSchemaNodeMixin<LeafStatement>,
  36             MandatoryMixin<QName, LeafStatement>, MustConstraintMixin<QName, LeafStatement> {
  37     private final @NonNull Object substatements;
  38     private final @NonNull QName argument;
  39     private final @NonNull TypeDefinition<?> type;
  40     private final int flags;
  41
  42     AbstractLeafEffectiveStatement(final LeafStatement declared, final QName argument, final int flags,
  43             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
  44         super(declared);
  45         this.argument = requireNonNull(argument);
  46         this.substatements = maskList(substatements);
  47         this.flags = flags;
  48         // TODO: lazy instantiation?
  49         this.type = buildType();
  50     }
  51
  52     AbstractLeafEffectiveStatement(final AbstractLeafEffectiveStatement original, final QName argument,
  53             final int flags) {
  54         super(original);
  55         this.argument = requireNonNull(argument);
  56         this.substatements = original.substatements;
  57         this.flags = flags;
  58         // FIXME: share with original?
  59         this.type = buildType();
  60     }
  61
  62     @Override
  63     public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
  64         return unmaskList(substatements);
  65     }
  66
  67     @Override
  68     public final int flags() {
  69         return flags;
  70     }
  71
  72     @Override
  73     public final QName argument() {
  74         return argument;
  75     }
  76
  77     @Override
  78     public final TypeDefinition<?> getType() {
  79         return type;
  80     }
  81
  82     @Override
  83     public final LeafEffectiveStatement asEffectiveStatement() {
  84         return this;
  85     }
  86
  87     private TypeDefinition<?> buildType() {
  88         final TypeEffectiveStatement<?> typeStmt = findFirstEffectiveSubstatement(TypeEffectiveStatement.class).get();
  89         final ConcreteTypeBuilder<?> builder = ConcreteTypes.concreteTypeBuilder(typeStmt.getTypeDefinition(),
  90             getQName());
  91         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
  92             if (stmt instanceof DefaultEffectiveStatement) {
  93                 builder.setDefaultValue(((DefaultEffectiveStatement)stmt).argument());
  94             } else if (stmt instanceof DescriptionEffectiveStatement) {
  95                 builder.setDescription(((DescriptionEffectiveStatement)stmt).argument());
  96             } else if (stmt instanceof ReferenceEffectiveStatement) {
  97                 builder.setReference(((ReferenceEffectiveStatement)stmt).argument());
  98             } else if (stmt instanceof StatusEffectiveStatement) {
  99                 builder.setStatus(((StatusEffectiveStatement)stmt).argument());
 100             } else if (stmt instanceof UnitsEffectiveStatement) {
 101                 builder.setUnits(((UnitsEffectiveStatement)stmt).argument());
 102             }
 103         }
 104         return builder.build();
 105     }
 106 }