Refactor Leaf(List)EffectiveStatementImpl
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / leaf_list / AbstractLeafListEffectiveStatement.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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.leaf_list;
9
10 import static com.google.common.base.Verify.verify;
11 import static java.util.Objects.requireNonNull;
12
13 import com.google.common.collect.ImmutableList;
14 import java.util.Objects;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.opendaylight.yangtools.yang.common.QName;
17 import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
18 import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
19 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
20 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
21 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionEffectiveStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.LeafListEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.LeafListStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceEffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
28 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
29 import org.opendaylight.yangtools.yang.model.util.type.ConcreteTypeBuilder;
30 import org.opendaylight.yangtools.yang.model.util.type.ConcreteTypes;
31 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractDeclaredEffectiveStatement;
32 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.DataSchemaNodeMixin;
33 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.MustConstraintMixin;
34 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.UserOrderedMixin;
35
36 abstract class AbstractLeafListEffectiveStatement
37         extends AbstractDeclaredEffectiveStatement.Default<QName, LeafListStatement>
38         implements LeafListEffectiveStatement, LeafListSchemaNode, DerivableSchemaNode,
39             UserOrderedMixin<QName, LeafListStatement>, DataSchemaNodeMixin<QName, LeafListStatement>,
40             MustConstraintMixin<QName, LeafListStatement> {
41     // Variable: either a single substatement or an ImmutableList
42     private final @NonNull Object substatements;
43     private final @NonNull SchemaPath path;
44     private final @NonNull TypeDefinition<?> type;
45     private final int flags;
46
47     AbstractLeafListEffectiveStatement(final LeafListStatement declared, final SchemaPath path, final int flags,
48             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
49         super(declared);
50         this.substatements = substatements.size() == 1 ? substatements.get(0) : substatements;
51         this.path = requireNonNull(path);
52         this.flags = flags;
53         // TODO: lazy instantiation?
54         this.type = buildType();
55     }
56
57     @Override
58     @SuppressWarnings("unchecked")
59     public final ImmutableList<? extends EffectiveStatement<?, ?>> effectiveSubstatements() {
60         if (substatements instanceof ImmutableList) {
61             return (ImmutableList<? extends EffectiveStatement<?, ?>>) substatements;
62         }
63         verify(substatements instanceof EffectiveStatement, "Unexpected substatement %s", substatements);
64         return ImmutableList.of((EffectiveStatement<?, ?>) substatements);
65     }
66
67     @Override
68     public final int flags() {
69         return flags;
70     }
71
72     @Override
73     public final @NonNull QName argument() {
74         return getQName();
75     }
76
77     @Override
78     public final @NonNull SchemaPath getPath() {
79         return path;
80     }
81
82     @Override
83     public final TypeDefinition<?> getType() {
84         return type;
85     }
86
87     @Override
88     public final boolean isUserOrdered() {
89         return userOrdered();
90     }
91
92     @Override
93     public final int hashCode() {
94         final int prime = 31;
95         int result = 1;
96         result = prime * result + Objects.hashCode(getQName());
97         result = prime * result + Objects.hashCode(getPath());
98         return result;
99     }
100
101     @Override
102     public final boolean equals(final Object obj) {
103         if (this == obj) {
104             return true;
105         }
106         if (!(obj instanceof AbstractLeafListEffectiveStatement)) {
107             return false;
108         }
109         final AbstractLeafListEffectiveStatement other = (AbstractLeafListEffectiveStatement) obj;
110         return Objects.equals(getQName(), other.getQName()) && Objects.equals(getPath(), other.getPath());
111     }
112
113     @Override
114     public final String toString() {
115         return getClass().getSimpleName() + "[" + getQName() + "]";
116     }
117
118     private TypeDefinition<?> buildType() {
119         final TypeEffectiveStatement<?> typeStmt = findFirstEffectiveSubstatement(TypeEffectiveStatement.class).get();
120         final ConcreteTypeBuilder<?> builder = ConcreteTypes.concreteTypeBuilder(typeStmt.getTypeDefinition(),
121             getPath());
122         for (final EffectiveStatement<?, ?> stmt : effectiveSubstatements()) {
123             // NOTE: 'default' is ommitted here on purpose
124             if (stmt instanceof DescriptionEffectiveStatement) {
125                 builder.setDescription(((DescriptionEffectiveStatement)stmt).argument());
126             } else if (stmt instanceof ReferenceEffectiveStatement) {
127                 builder.setReference(((ReferenceEffectiveStatement)stmt).argument());
128             } else if (stmt instanceof StatusEffectiveStatement) {
129                 builder.setStatus(((StatusEffectiveStatement)stmt).argument());
130             } else if (stmt instanceof UnitsEffectiveStatement) {
131                 builder.setUnits(((UnitsEffectiveStatement)stmt).argument());
132             }
133         }
134         return builder.build();
135     }
136 }