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