7199f341b2819a9e90050ed42f06705cb9b6eee2
[genius.git] / mdsalutil / mdsalutil-api / src / main / java / org / opendaylight / genius / mdsalutil / ericmatches / EricMatchInfoHelper.java
1 /*
2  * Copyright (c) 2019 Alten Calsoft Labs India Pvt Ltd. 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.genius.mdsalutil.ericmatches;
9
10 import java.util.HashMap;
11 import java.util.Map;
12 import org.opendaylight.genius.mdsalutil.MatchInfo;
13 import org.opendaylight.genius.utils.SuperTypeUtil;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.EricAugMatchNodesNodeTableFlow;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.eric.match.rev180730.EricAugMatchNodesNodeTableFlowBuilder;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.ExtensionKey;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlow;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.GeneralAugMatchNodesNodeTableFlowBuilder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.grouping.ExtensionBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.list.grouping.ExtensionList;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.list.grouping.ExtensionListBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflowplugin.extension.general.rev140714.general.extension.list.grouping.ExtensionListKey;
24 import org.opendaylight.yangtools.concepts.Builder;
25 import org.opendaylight.yangtools.yang.binding.DataObject;
26
27
28 public abstract class EricMatchInfoHelper<T extends DataObject, B extends Builder<T>> extends MatchInfo {
29     private final Class<B> builderClass;
30     // The key class can't be a type parameter, it varies in some subclasses
31     private final Class<? extends ExtensionKey> keyClass;
32
33     protected EricMatchInfoHelper(Class<? extends ExtensionKey> keyClass) {
34         this.keyClass = keyClass;
35         builderClass = SuperTypeUtil.getTypeParameter(getClass(), 1);
36     }
37
38     @Override
39     public void createInnerMatchBuilder(Map<Class<?>, Object> mapMatchBuilder) {
40         populateBuilder((B) mapMatchBuilder.computeIfAbsent(builderClass, key -> {
41             try {
42                 return builderClass.newInstance();
43             } catch (InstantiationException | IllegalAccessException e) {
44                 throw new IllegalStateException("Unable to create an instance of " + builderClass, e);
45             }
46         }));
47     }
48
49     @Override
50     public void setMatch(MatchBuilder matchBuilder, Map<Class<?>, Object> mapMatchBuilder) {
51         B builder = (B) mapMatchBuilder.remove(builderClass);
52
53         if (builder != null) {
54
55             EricAugMatchNodesNodeTableFlowBuilder ericAugMatchBuilder = new EricAugMatchNodesNodeTableFlowBuilder();
56             applyValue(ericAugMatchBuilder, builder.build());
57             EricAugMatchNodesNodeTableFlow ericAugMatch = ericAugMatchBuilder.build();
58             GeneralAugMatchNodesNodeTableFlow existingAugmentations = matchBuilder
59                     .augmentation(GeneralAugMatchNodesNodeTableFlow.class);
60             GeneralAugMatchNodesNodeTableFlow genAugMatch = generalAugMatchBuilder(existingAugmentations,
61                     ericAugMatch, keyClass);
62             matchBuilder.addAugmentation(GeneralAugMatchNodesNodeTableFlow.class, genAugMatch);
63         }
64     }
65
66     private GeneralAugMatchNodesNodeTableFlow generalAugMatchBuilder(
67             GeneralAugMatchNodesNodeTableFlow existingAugmentations, EricAugMatchNodesNodeTableFlow ericAugMatch,
68             Class<? extends ExtensionKey> extentionKey) {
69         Map<ExtensionListKey, ExtensionList> extensions = null;
70         if (existingAugmentations != null) {
71             extensions = existingAugmentations.getExtensionList();
72         }
73         if (extensions == null) {
74             extensions = new HashMap<>();
75         }
76         ExtensionList extensionList = new ExtensionListBuilder().setExtensionKey(extentionKey)
77                 .setExtension(
78                         new ExtensionBuilder().addAugmentation(EricAugMatchNodesNodeTableFlow.class, ericAugMatch)
79                                 .build())
80                 .build();
81         extensions.put(extensionList.key(),extensionList);
82         return new GeneralAugMatchNodesNodeTableFlowBuilder().setExtensionList(extensions).build();
83     }
84
85     protected abstract void applyValue(EricAugMatchNodesNodeTableFlowBuilder matchBuilder, T value);
86
87     protected abstract void populateBuilder(B builder);
88
89     @Override
90     public boolean equals(Object other) {
91         if (this == other) {
92             return true;
93         }
94         if (other == null || getClass() != other.getClass()) {
95             return false;
96         }
97
98         EricMatchInfoHelper<?, ?> that = (EricMatchInfoHelper<?, ?>) other;
99
100         if (builderClass != null ? !builderClass.equals(that.builderClass) : that.builderClass != null) {
101             return false;
102         }
103         return keyClass != null ? keyClass.equals(that.keyClass) : that.keyClass == null;
104     }
105
106     @Override
107     public int hashCode() {
108         int result = builderClass != null ? builderClass.hashCode() : 0;
109         result = 31 * result + (keyClass != null ? keyClass.hashCode() : 0);
110         return result;
111     }
112 }