a45c1dcb35c8d74f920b9135e2af0271a4c02363
[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / reactor / KeyGenerator.java
1 /*
2  * Copyright (c) 2021 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.mdsal.binding.generator.impl.reactor;
9
10 import static com.google.common.base.Verify.verify;
11 import static java.util.Objects.requireNonNull;
12
13 import java.util.Set;
14 import org.opendaylight.mdsal.binding.generator.impl.reactor.CollisionDomain.Member;
15 import org.opendaylight.mdsal.binding.generator.impl.rt.DefaultKeyRuntimeType;
16 import org.opendaylight.mdsal.binding.model.api.GeneratedTransferObject;
17 import org.opendaylight.mdsal.binding.model.api.GeneratedType;
18 import org.opendaylight.mdsal.binding.model.api.Type;
19 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedPropertyBuilder;
20 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTOBuilder;
21 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilderBase;
22 import org.opendaylight.mdsal.binding.model.ri.BindingTypes;
23 import org.opendaylight.mdsal.binding.runtime.api.KeyRuntimeType;
24 import org.opendaylight.mdsal.binding.spec.naming.BindingMapping;
25 import org.opendaylight.yangtools.yang.common.QName;
26 import org.opendaylight.yangtools.yang.model.api.stmt.KeyEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
28
29 final class KeyGenerator extends AbstractExplicitGenerator<KeyEffectiveStatement, KeyRuntimeType> {
30     private final ListGenerator listGen;
31
32     KeyGenerator(final KeyEffectiveStatement statement, final AbstractCompositeGenerator<?, ?> parent,
33             final ListGenerator listGen) {
34         super(statement, parent);
35         this.listGen = requireNonNull(listGen);
36     }
37
38     @Override
39     void pushToInference(final SchemaInferenceStack inferenceStack) {
40         throw new UnsupportedOperationException();
41     }
42
43     @Override
44     Member createMember(final CollisionDomain domain) {
45         return domain.addSecondary(this, listGen.getMember(), BindingMapping.KEY_SUFFIX);
46     }
47
48     @Override
49     GeneratedTransferObject createTypeImpl(final TypeBuilderFactory builderFactory) {
50         final GeneratedTOBuilder builder = builderFactory.newGeneratedTOBuilder(typeName());
51
52         builder.addImplementsType(BindingTypes.identifier(Type.of(listGen.typeName())));
53
54         final Set<QName> leafNames = statement().argument();
55         for (Generator listChild : listGen) {
56             if (listChild instanceof LeafGenerator) {
57                 final LeafGenerator leafGen = (LeafGenerator) listChild;
58                 final QName qname = leafGen.statement().argument();
59                 if (leafNames.contains(qname)) {
60                     final GeneratedPropertyBuilder prop = builder
61                         .addProperty(BindingMapping.getPropertyName(qname.getLocalName()))
62                         .setReturnType(leafGen.methodReturnType(builderFactory))
63                         .setReadOnly(true);
64
65 //                    addComment(propBuilder, leaf);
66
67                     builder.addEqualsIdentity(prop);
68                     builder.addHashIdentity(prop);
69                     builder.addToStringProperty(prop);
70                 }
71             }
72         }
73
74         // serialVersionUID
75         addSerialVersionUID(builder);
76
77         return builder.build();
78     }
79
80     @Override
81     GeneratedType runtimeJavaType() {
82         return generatedType().orElse(null);
83     }
84
85     @Override
86     KeyRuntimeType createExternalRuntimeType(final Type type) {
87         verify(type instanceof GeneratedTransferObject, "Unexpected type %s", type);
88         return new DefaultKeyRuntimeType((GeneratedTransferObject) type, statement());
89     }
90
91     @Override
92     KeyRuntimeType createInternalRuntimeType(final ChildLookup lookup,final KeyEffectiveStatement statement,
93             final Type type) {
94         // The only reference to this runtime type is from ListGenerator which is always referencing the external type
95         throw new UnsupportedOperationException("Should never be called");
96     }
97
98     @Override
99     void addAsGetterMethod(final GeneratedTypeBuilderBase<?> builder, final TypeBuilderFactory builderFactory) {
100         // Keys are explicitly handled by their corresponding list
101     }
102 }