yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/augment/AugmentEffectiveStatementImpl.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 package org.opendaylight.yangtools.yang.parser.rfc7950.stmt.augment;
   9
  10 import static com.google.common.base.Verify.verify;
  11
  12 import com.google.common.collect.Collections2;
  13 import com.google.common.collect.ImmutableSet;
  14 import java.net.URI;
  15 import java.util.Collection;
  16 import java.util.Optional;
  17 import java.util.Set;
  18 import org.eclipse.jdt.annotation.NonNull;
  19 import org.opendaylight.yangtools.yang.common.QNameModule;
  20 import org.opendaylight.yangtools.yang.common.Revision;
  21 import org.opendaylight.yangtools.yang.model.api.ActionDefinition;
  22 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
  23 import org.opendaylight.yangtools.yang.model.api.NamespaceRevisionAware;
  24 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
  25 import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
  26 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  27 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  28 import org.opendaylight.yangtools.yang.model.api.stmt.AugmentEffectiveStatement;
  29 import org.opendaylight.yangtools.yang.model.api.stmt.AugmentStatement;
  30 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
  31 import org.opendaylight.yangtools.yang.model.api.stmt.WhenEffectiveStatement;
  32 import org.opendaylight.yangtools.yang.model.api.stmt.compat.ActionNodeContainerCompat;
  33 import org.opendaylight.yangtools.yang.model.api.stmt.compat.NotificationNodeContainerCompat;
  34 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveDocumentedDataNodeContainer;
  35 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  36 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
  37 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase;
  38
  39 final class AugmentEffectiveStatementImpl
  40         extends AbstractEffectiveDocumentedDataNodeContainer<SchemaNodeIdentifier, AugmentStatement>
  41         implements AugmentEffectiveStatement, AugmentationSchemaNode, NamespaceRevisionAware,
  42             ActionNodeContainerCompat<SchemaNodeIdentifier, AugmentStatement>,
  43             NotificationNodeContainerCompat<SchemaNodeIdentifier, AugmentStatement> {
  44     private final SchemaPath targetPath;
  45     private final QNameModule rootModuleQName;
  46     private final @NonNull ImmutableSet<ActionDefinition> actions;
  47     private final @NonNull ImmutableSet<NotificationDefinition> notifications;
  48     private final RevisionAwareXPath whenCondition;
  49     private final AugmentationSchemaNode copyOf;
  50
  51     AugmentEffectiveStatementImpl(
  52             final StmtContext<SchemaNodeIdentifier, AugmentStatement, AugmentEffectiveStatement> ctx) {
  53         super(ctx);
  54         targetPath = ctx.coerceStatementArgument().asSchemaPath();
  55         rootModuleQName = StmtContextUtils.getRootModuleQName(ctx);
  56         copyOf = (AugmentationSchemaNode) ctx.getOriginalCtx().map(StmtContext::buildEffective).orElse(null);
  57         whenCondition = findFirstEffectiveSubstatementArgument(WhenEffectiveStatement.class).orElse(null);
  58
  59         // initSubstatementCollections
  60         final ImmutableSet.Builder<ActionDefinition> actionsBuilder = ImmutableSet.builder();
  61         final ImmutableSet.Builder<NotificationDefinition> notificationsBuilder = ImmutableSet.builder();
  62         for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
  63             if (effectiveStatement instanceof ActionDefinition) {
  64                 actionsBuilder.add((ActionDefinition) effectiveStatement);
  65             } else if (effectiveStatement instanceof NotificationDefinition) {
  66                 notificationsBuilder.add((NotificationDefinition) effectiveStatement);
  67             }
  68         }
  69
  70         actions = actionsBuilder.build();
  71         notifications = notificationsBuilder.build();
  72     }
  73
  74     @Override
  75     protected Collection<? extends EffectiveStatement<?, ?>> initSubstatements(
  76             final StmtContext<SchemaNodeIdentifier, AugmentStatement, ?> ctx,
  77             final Collection<? extends StmtContext<?, ?, ?>> substatementsInit) {
  78         final StatementContextBase<?, ?, ?> implicitDef = ctx.getFromNamespace(AugmentImplicitHandlingNamespace.class,
  79             ctx);
  80         return implicitDef == null ? super.initSubstatements(ctx, substatementsInit)
  81                 : Collections2.transform(Collections2.filter(substatementsInit,
  82                     StmtContext::isSupportedToBuildEffective),
  83                     subCtx -> {
  84                         verify(subCtx instanceof StatementContextBase);
  85                         return implicitDef.wrapWithImplicit((StatementContextBase<?, ?, ?>) subCtx).buildEffective();
  86                     });
  87     }
  88
  89     @Override
  90     public Optional<AugmentationSchemaNode> getOriginalDefinition() {
  91         return Optional.ofNullable(this.copyOf);
  92     }
  93
  94     @Override
  95     public SchemaPath getTargetPath() {
  96         return targetPath;
  97     }
  98
  99     @Override
 100     public Optional<RevisionAwareXPath> getWhenCondition() {
 101         return Optional.ofNullable(whenCondition);
 102     }
 103
 104     @Override
 105     public URI getNamespace() {
 106         return rootModuleQName.getNamespace();
 107     }
 108
 109     @Override
 110     public Optional<Revision> getRevision() {
 111         return rootModuleQName.getRevision();
 112     }
 113
 114     @Override
 115     public Set<ActionDefinition> getActions() {
 116         return actions;
 117     }
 118
 119     @Override
 120     public Set<NotificationDefinition> getNotifications() {
 121         return notifications;
 122     }
 123
 124     @Override
 125     public String toString() {
 126         return AugmentEffectiveStatementImpl.class.getSimpleName() + "[" + "targetPath=" + targetPath + ", when="
 127                 + whenCondition + "]";
 128     }
 129 }