Track schema tree generator linkage
[mdsal.git] / binding / mdsal-binding-generator / src / main / java / org / opendaylight / mdsal / binding / generator / impl / reactor / NotificationGenerator.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 org.opendaylight.mdsal.binding.model.api.GeneratedType;
11 import org.opendaylight.mdsal.binding.model.api.Type;
12 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilder;
13 import org.opendaylight.mdsal.binding.model.api.type.builder.GeneratedTypeBuilderBase;
14 import org.opendaylight.mdsal.binding.model.ri.BindingTypes;
15 import org.opendaylight.yangtools.yang.model.api.stmt.NotificationEffectiveStatement;
16 import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
17
18 /**
19  * Generator corresponding to a {@code notification} statement.
20  */
21 final class NotificationGenerator
22         extends CompositeSchemaTreeGenerator<NotificationEffectiveStatement, NotificationGenerator> {
23     NotificationGenerator(final NotificationEffectiveStatement statement, final AbstractCompositeGenerator<?> parent) {
24         super(statement, parent);
25     }
26
27     @Override
28     void pushToInference(final SchemaInferenceStack dataTree) {
29         dataTree.enterSchemaTree(statement().getIdentifier());
30     }
31
32     @Override
33     GeneratedType createTypeImpl(final TypeBuilderFactory builderFactory) {
34         final GeneratedTypeBuilder builder = builderFactory.newGeneratedTypeBuilder(typeName());
35
36         builder.addImplementsType(BindingTypes.DATA_OBJECT);
37         builder.addImplementsType(notificationType(builder, builderFactory));
38
39         addAugmentable(builder);
40         addUsesInterfaces(builder, builderFactory);
41
42         addConcreteInterfaceMethods(builder);
43         addGetterMethods(builder, builderFactory);
44
45         final ModuleGenerator module = currentModule();
46         module.addQNameConstant(builder, localName());
47
48         builderFactory.addCodegenInformation(module, statement(), builder);
49         annotateDeprecatedIfNecessary(builder);
50
51         return builder.build();
52     }
53
54     @Override
55     void addAsGetterMethod(final GeneratedTypeBuilderBase<?> builder, final TypeBuilderFactory builderFactory) {
56         // Notifications are a distinct concept
57     }
58
59     private Type notificationType(final GeneratedTypeBuilder builder, final TypeBuilderFactory builderFactory) {
60         final AbstractCompositeGenerator<?> parent = getParent();
61         if (parent instanceof ModuleGenerator) {
62             return BindingTypes.notification(builder);
63         }
64
65         final Type parentType = Type.of(parent.typeName());
66         if (parent instanceof ListGenerator) {
67             final KeyGenerator keyGen = ((ListGenerator) parent).keyGenerator();
68             if (keyGen != null) {
69                 return BindingTypes.keyedListNotification(builder, parentType, keyGen.getGeneratedType(builderFactory));
70             }
71         }
72         return BindingTypes.instanceNotification(builder, parentType);
73     }
74 }