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%2FBuilderGenerator.java;h=3bada347ead919c809978659a9b0d2ec5741af4f;hb=d613481dfab5abd014e36fc8f77b291743c57dbf;hp=ab5944b573d8e43a6a255ce703795e9f2ff555c7;hpb=30794cd4999beb5a7a4902d4cf0484692cbd629b;p=mdsal.git

diff --git a/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/BuilderGenerator.java b/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/BuilderGenerator.java
index ab5944b573..3bada347ea 100644
--- a/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/BuilderGenerator.java
+++ b/binding/mdsal-binding-java-api-generator/src/main/java/org/opendaylight/mdsal/binding/java/api/generator/BuilderGenerator.java
@@ -41,6 +41,9 @@ import org.opendaylight.yangtools.yang.binding.Augmentation;
  * class. For generation of the source code is used the template written in XTEND language.
  */
 public final class BuilderGenerator implements CodeGenerator {
+    private static final JavaTypeName AUGMENTABLE = JavaTypeName.create(Augmentable.class);
+    private static final JavaTypeName AUGMENTATION = JavaTypeName.create(Augmentation.class);
+
     private static final Comparator<MethodSignature> METHOD_COMPARATOR = new AlphabeticallyTypeMemberComparator<>();
     private static final Type AUGMENTATION_RET_TYPE;
 
@@ -66,12 +69,10 @@ public final class BuilderGenerator implements CodeGenerator {
         if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
             for (Type t : ((GeneratedType) type).getImplements()) {
                 // "rpc" and "grouping" elements do not implement Augmentable
-                if (t.getFullyQualifiedName().equals(Augmentable.class.getName())) {
-                    return true;
-                } else if (t.getFullyQualifiedName().equals(Augmentation.class.getName())) {
+                final JavaTypeName name = t.getIdentifier();
+                if (name.equals(AUGMENTABLE) || name.equals(AUGMENTATION)) {
                     return true;
                 }
-
             }
         }
         return false;