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.ImmutableList;
  14 import com.google.common.collect.ImmutableSet;
  15 import java.net.URI;
  16 import java.util.Collection;
  17 import java.util.Objects;
  18 import java.util.Optional;
  19 import java.util.Set;
  20 import org.eclipse.jdt.annotation.NonNull;
  21 import org.opendaylight.yangtools.yang.common.QNameModule;
  22 import org.opendaylight.yangtools.yang.common.Revision;
  23 import org.opendaylight.yangtools.yang.model.api.ActionDefinition;
  24 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
  25 import org.opendaylight.yangtools.yang.model.api.NamespaceRevisionAware;
  26 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
  27 import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
  28 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  29 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
  30 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
  31 import org.opendaylight.yangtools.yang.model.api.stmt.AugmentEffectiveStatement;
  32 import org.opendaylight.yangtools.yang.model.api.stmt.AugmentStatement;
  33 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
  34 import org.opendaylight.yangtools.yang.model.api.stmt.WhenEffectiveStatement;
  35 import org.opendaylight.yangtools.yang.model.api.stmt.compat.ActionNodeContainerCompat;
  36 import org.opendaylight.yangtools.yang.model.api.stmt.compat.NotificationNodeContainerCompat;
  37 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveDocumentedDataNodeContainer;
  38 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  39 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
  40 import org.opendaylight.yangtools.yang.parser.stmt.reactor.StatementContextBase;
  41
  42 final class AugmentEffectiveStatementImpl
  43         extends AbstractEffectiveDocumentedDataNodeContainer<SchemaNodeIdentifier, AugmentStatement>
  44         implements AugmentEffectiveStatement, AugmentationSchemaNode, NamespaceRevisionAware,
  45             ActionNodeContainerCompat<SchemaNodeIdentifier, AugmentStatement>,
  46             NotificationNodeContainerCompat<SchemaNodeIdentifier, AugmentStatement> {
  47     private final SchemaPath targetPath;
  48     private final URI namespace;
  49     private final Revision revision;
  50     private final @NonNull ImmutableSet<ActionDefinition> actions;
  51     private final @NonNull ImmutableSet<NotificationDefinition> notifications;
  52     private final RevisionAwareXPath whenCondition;
  53     private final AugmentationSchemaNode copyOf;
  54
  55     AugmentEffectiveStatementImpl(final StmtContext<SchemaNodeIdentifier, AugmentStatement,
  56             EffectiveStatement<SchemaNodeIdentifier, AugmentStatement>> ctx) {
  57         super(ctx);
  58
  59         this.targetPath = ctx.coerceStatementArgument().asSchemaPath();
  60
  61         final QNameModule rootModuleQName = StmtContextUtils.getRootModuleQName(ctx);
  62         this.namespace = rootModuleQName.getNamespace();
  63         this.revision = rootModuleQName.getRevision().orElse(null);
  64
  65         this.copyOf = (AugmentationSchemaNode) ctx.getOriginalCtx().map(StmtContext::buildEffective).orElse(null);
  66         whenCondition = findFirstEffectiveSubstatementArgument(WhenEffectiveStatement.class).orElse(null);
  67
  68         // initSubstatementCollections
  69         final ImmutableSet.Builder<ActionDefinition> actionsBuilder = ImmutableSet.builder();
  70         final ImmutableSet.Builder<NotificationDefinition> notificationsBuilder = ImmutableSet.builder();
  71         final ImmutableList.Builder<UnknownSchemaNode> listBuilder = new ImmutableList.Builder<>();
  72         for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
  73             if (effectiveStatement instanceof ActionDefinition) {
  74                 actionsBuilder.add((ActionDefinition) effectiveStatement);
  75             } else if (effectiveStatement instanceof NotificationDefinition) {
  76                 notificationsBuilder.add((NotificationDefinition) effectiveStatement);
  77             }
  78         }
  79
  80         this.actions = actionsBuilder.build();
  81         this.notifications = notificationsBuilder.build();
  82     }
  83
  84     @Override
  85     protected Collection<? extends EffectiveStatement<?, ?>> initSubstatements(
  86             final StmtContext<SchemaNodeIdentifier, AugmentStatement, ?> ctx,
  87             final Collection<? extends StmtContext<?, ?, ?>> substatementsInit) {
  88         final StatementContextBase<?, ?, ?> implicitDef = ctx.getFromNamespace(AugmentImplicitHandlingNamespace.class,
  89             ctx);
  90         return implicitDef == null ? super.initSubstatements(ctx, substatementsInit)
  91                 : Collections2.transform(Collections2.filter(substatementsInit,
  92                     StmtContext::isSupportedToBuildEffective),
  93                     subCtx -> {
  94                         verify(subCtx instanceof StatementContextBase);
  95                         return implicitDef.wrapWithImplicit((StatementContextBase<?, ?, ?>) subCtx).buildEffective();
  96                     });
  97     }
  98
  99     @Override
 100     public Optional<AugmentationSchemaNode> getOriginalDefinition() {
 101         return Optional.ofNullable(this.copyOf);
 102     }
 103
 104     @Override
 105     public SchemaPath getTargetPath() {
 106         return targetPath;
 107     }
 108
 109     @Override
 110     public Optional<RevisionAwareXPath> getWhenCondition() {
 111         return Optional.ofNullable(whenCondition);
 112     }
 113
 114     @Override
 115     public URI getNamespace() {
 116         return namespace;
 117     }
 118
 119     @Override
 120     public Set<ActionDefinition> getActions() {
 121         return actions;
 122     }
 123
 124     @Override
 125     public Set<NotificationDefinition> getNotifications() {
 126         return notifications;
 127     }
 128
 129     @Override
 130     public Optional<Revision> getRevision() {
 131         return Optional.ofNullable(revision);
 132     }
 133
 134     @Override
 135     public int hashCode() {
 136         final int prime = 17;
 137         int result = 1;
 138         result = prime * result + Objects.hashCode(targetPath);
 139         result = prime * result + Objects.hashCode(whenCondition);
 140         result = prime * result + getChildNodes().hashCode();
 141         return result;
 142     }
 143
 144     @Override
 145     public boolean equals(final Object obj) {
 146         if (this == obj) {
 147             return true;
 148         }
 149         if (!(obj instanceof AugmentEffectiveStatementImpl)) {
 150             return false;
 151         }
 152         final AugmentEffectiveStatementImpl other = (AugmentEffectiveStatementImpl) obj;
 153         return Objects.equals(targetPath, other.targetPath) && Objects.equals(whenCondition, other.whenCondition)
 154                 && getChildNodes().equals(other.getChildNodes());
 155     }
 156
 157     @Override
 158     public String toString() {
 159         return AugmentEffectiveStatementImpl.class.getSimpleName() + "[" + "targetPath=" + targetPath + ", when="
 160                 + whenCondition + "]";
 161     }
 162 }