yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/stmt/identity/AbstractIdentityEffectiveStatement.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.identity;
   9
  10 import static com.google.common.base.Preconditions.checkState;
  11
  12 import com.google.common.base.MoreObjects;
  13 import com.google.common.collect.ImmutableSet;
  14 import java.util.Collections;
  15 import java.util.HashSet;
  16 import java.util.List;
  17 import java.util.Objects;
  18 import java.util.Set;
  19 import org.eclipse.jdt.annotation.NonNull;
  20 import org.opendaylight.yangtools.yang.common.QName;
  21 import org.opendaylight.yangtools.yang.model.api.IdentitySchemaNode;
  22 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  23 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityEffectiveStatement;
  24 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
  25 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredEffectiveStatement.DefaultArgument;
  26 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.SchemaNodeMixin;
  27 import org.opendaylight.yangtools.yang.parser.spi.meta.DerivedIdentitiesNamespace;
  28 import org.opendaylight.yangtools.yang.parser.spi.meta.MutableStatement;
  29 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
  30
  31 abstract class AbstractIdentityEffectiveStatement extends DefaultArgument<QName, IdentityStatement>
  32         implements IdentityEffectiveStatement, IdentitySchemaNode, MutableStatement,
  33                    SchemaNodeMixin<QName, IdentityStatement> {
  34     private final @NonNull SchemaPath path;
  35     private final ImmutableSet<IdentitySchemaNode> derivedIdentities;
  36
  37     private @NonNull Set<IdentitySchemaNode> baseIdentities;
  38     private boolean sealed;
  39
  40     AbstractIdentityEffectiveStatement(final IdentityStatement declared,
  41             final StmtContext<QName, IdentityStatement, IdentityEffectiveStatement> ctx) {
  42         super(declared);
  43         this.path = ctx.getSchemaPath().get();
  44
  45         this.baseIdentities = new HashSet<>();
  46         ((StmtContext.Mutable<?, ?, ?>) ctx).addMutableStmtToSeal(this);
  47
  48         // initDerivedIdentities
  49         final Set<IdentitySchemaNode> derivedIdentitiesInit = new HashSet<>();
  50         final List<StmtContext<?, ?, ?>> derivedIdentitiesCtxList = ctx.getFromNamespace(
  51                 DerivedIdentitiesNamespace.class, ctx.coerceStatementArgument());
  52         if (derivedIdentitiesCtxList == null) {
  53             this.derivedIdentities = ImmutableSet.of();
  54             return;
  55         }
  56         for (final StmtContext<?, ?, ?> derivedIdentityCtx : derivedIdentitiesCtxList) {
  57             final AbstractIdentityEffectiveStatement derivedIdentity =
  58                     (AbstractIdentityEffectiveStatement) derivedIdentityCtx.buildEffective();
  59             derivedIdentity.addBaseIdentity(this);
  60             derivedIdentitiesInit.add(derivedIdentity);
  61         }
  62         this.derivedIdentities = ImmutableSet.copyOf(derivedIdentitiesInit);
  63     }
  64
  65     private void addBaseIdentity(final AbstractIdentityEffectiveStatement baseIdentity) {
  66         checkState(!sealed, "Attempt to modify sealed identity effective statement %s", getQName());
  67         this.baseIdentities.add(baseIdentity);
  68     }
  69
  70     @Override
  71     public final SchemaPath getPath() {
  72         return path;
  73     }
  74
  75     @Override
  76     public final Set<IdentitySchemaNode> getBaseIdentities() {
  77         checkState(sealed, "Attempt to get base identities from unsealed identity effective statement %s", getQName());
  78         return baseIdentities;
  79     }
  80
  81     @Override
  82     public final Set<IdentitySchemaNode> getDerivedIdentities() {
  83         return Collections.unmodifiableSet(derivedIdentities);
  84     }
  85
  86     @Override
  87     public final int hashCode() {
  88         final int prime = 31;
  89         int result = 1;
  90         result = prime * result + Objects.hashCode(getQName());
  91         result = prime * result + Objects.hashCode(getPath());
  92         return result;
  93     }
  94
  95     @Override
  96     public final boolean equals(final Object obj) {
  97         if (this == obj) {
  98             return true;
  99         }
 100         if (!(obj instanceof AbstractIdentityEffectiveStatement)) {
 101             return false;
 102         }
 103         final AbstractIdentityEffectiveStatement other = (AbstractIdentityEffectiveStatement) obj;
 104         return Objects.equals(getQName(), other.getQName()) && Objects.equals(getPath(), other.getPath());
 105     }
 106
 107     @Override
 108     public final String toString() {
 109         return MoreObjects.toStringHelper(this).add("qname", getQName()).add("path", getPath()).toString();
 110     }
 111
 112     @Override
 113     public final void seal() {
 114         if (!sealed) {
 115             baseIdentities = ImmutableSet.copyOf(baseIdentities);
 116             sealed = true;
 117         }
 118     }
 119 }