X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=binding%2Fmdsal-binding-java-api-generator%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fmdsal%2Fbinding%2Fjava%2Fapi%2Fgenerator%2FEnumTemplate.xtend;h=89239186786e3216f6a80289d0ac33c7c0e6edd4;hb=0ceceb790380a4d3098d7c046c20d8d71233e465;hp=1fb24d9d8f1d1091d55600040e9f5225ca57f1f5;hpb=1223c177f889d632fbb2b9c005c077966236365b;p=mdsal.git

diff --git a/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/EnumTemplate.xtend b/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/EnumTemplate.xtend
index 1fb24d9d8f..8923918678 100644
--- a/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/EnumTemplate.xtend
+++ b/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/EnumTemplate.xtend
@@ -7,12 +7,9 @@
  */
 package org.opendaylight.mdsal.binding.java.api.generator
 
-import static extension org.opendaylight.mdsal.binding.model.util.BindingGeneratorUtil.encodeAngleBrackets
-import static org.opendaylight.mdsal.binding.model.util.Types.STRING;
+import static extension org.opendaylight.mdsal.binding.generator.BindingGeneratorUtil.encodeAngleBrackets
+import static org.opendaylight.mdsal.binding.model.ri.Types.STRING;
 
-import com.google.common.collect.ImmutableMap
-import com.google.common.collect.ImmutableMap.Builder
-import java.util.Optional
 import org.opendaylight.mdsal.binding.model.api.Enumeration
 import org.opendaylight.mdsal.binding.model.api.GeneratedType
 
@@ -72,31 +69,16 @@ class EnumTemplate extends BaseTemplate {
         public enum «enums.name» implements «org.opendaylight.yangtools.yang.binding.Enumeration.importedName» {
             «writeEnumeration(enums)»
 
-            private static final «JU_MAP.importedName»<«STRING.importedName», «enums.name»> NAME_MAP;
-            private static final «JU_MAP.importedName»<«Integer.importedName», «enums.name»> VALUE_MAP;
-
-            static {
-                final «Builder.importedName»<«STRING.importedName», «enums.name»> nb = «ImmutableMap.importedName».builder();
-                final «Builder.importedName»<«Integer.importedName», «enums.name»> vb = «ImmutableMap.importedName».builder();
-                for («enums.name» enumItem : «enums.name».values()) {
-                    vb.put(enumItem.value, enumItem);
-                    nb.put(enumItem.name, enumItem);
-                }
-
-                NAME_MAP = nb.build();
-                VALUE_MAP = vb.build();
-            }
-
-            private final «STRING.importedName» name;
+            private final «STRING.importedNonNull» name;
             private final int value;
 
-            private «enums.name»(int value, «STRING.importedName» name) {
+            private «enums.name»(int value, «STRING.importedNonNull» name) {
                 this.value = value;
                 this.name = name;
             }
 
             @«OVERRIDE.importedName»
-            public «STRING.importedName» getName() {
+            public «STRING.importedNonNull» getName() {
                 return name;
             }
 
@@ -106,24 +88,57 @@ class EnumTemplate extends BaseTemplate {
             }
 
             /**
-             * Return the enumeration member whose {@link #getName()} matches specified value.
+             * Return the enumeration member whose {@link #getName()} matches specified assigned name.
+             *
+             * @param name YANG assigned name
+             * @return corresponding «enums.name» item, or {@code null} if no such item exists
+             * @throws NullPointerException if {@code name} is null
+             */
+            public static «enums.importedNullable» forName(«STRING.importedName» name) {
+                return switch (name) {
+                    «FOR v : enums.values»
+                    case "«v.name»" -> «v.mappedName»;
+                    «ENDFOR»
+                    default -> null;
+                };
+            }
+
+            /**
+             * Return the enumeration member whose {@link #getIntValue()} matches specified value.
+             *
+             * @param intValue integer value
+             * @return corresponding «enums.name» item, or {@code null} if no such item exists
+             */
+            public static «enums.importedNullable» forValue(int intValue) {
+                return switch (intValue) {
+                    «FOR v : enums.values»
+                    case «v.value» -> «v.mappedName»;
+                    «ENDFOR»
+                    default -> null;
+                };
+            }
+
+            /**
+             * Return the enumeration member whose {@link #getName()} matches specified assigned name.
              *
              * @param name YANG assigned name
-             * @return corresponding «enums.name» item, if present
-             * @throws NullPointerException if name is null
+             * @return corresponding «enums.name» item
+             * @throws NullPointerException if {@code name} is null
+             * @throws IllegalArgumentException if {@code name} does not match any item
              */
-            public static «Optional.importedName»<«enums.name»> forName(«STRING.importedName» name) {
-                return «Optional.importedName».ofNullable(NAME_MAP.get(«JU_OBJECTS.importedName».requireNonNull(name)));
+            public static «enums.importedNonNull» ofName(«STRING.importedName» name) {
+                return «CODEHELPERS.importedName».checkEnum(forName(name), name);
             }
 
             /**
              * Return the enumeration member whose {@link #getIntValue()} matches specified value.
              *
              * @param intValue integer value
-             * @return corresponding «enums.name» item, or null if no such item exists
+             * @return corresponding «enums.name» item
+             * @throws IllegalArgumentException if {@code intValue} does not match any item
              */
-            public static «enums.name» forValue(int intValue) {
-                return VALUE_MAP.get(intValue);
+            public static «enums.importedNonNull» ofValue(int intValue) {
+                return «CODEHELPERS.importedName».checkEnum(forValue(intValue), intValue);
             }
         }
     '''