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