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