Cleanup DocumentedNode

[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / effective / type / EnumSpecificationEffectiveStatementImpl.java

diff --git a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/type/EnumSpecificationEffectiveStatementImpl.java b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/type/EnumSpecificationEffectiveStatementImpl.java
index 1b4efe16faa7ccfda5168298b7a35938f276fa0d..82f918c14c3fe41ac86285bff0be2706d780c9ac 100644 (file)
--- a/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/type/EnumSpecificationEffectiveStatementImpl.java
+++ b/yang/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/stmt/rfc6020/effective/type/EnumSpecificationEffectiveStatementImpl.java
@@ -50,16 +50,17 @@ public final class EnumSpecificationEffectiveStatementImpl extends
                     effectiveValue = enumSubStmt.getDeclaredValue();
                 }
 
-                final EnumPair p = EnumPairBuilder.create(enumSubStmt.getName(), effectiveValue)
-                        .setDescription(enumSubStmt.getDescription()).setReference(enumSubStmt.getReference())
-                        .setStatus(enumSubStmt.getStatus()).setUnknownSchemaNodes(enumSubStmt.getUnknownSchemaNodes())
-                        .build();
+                final EnumPairBuilder pairBuilder = EnumPairBuilder.create(enumSubStmt.getName(), effectiveValue)
+                        .setStatus(enumSubStmt.getStatus()).setUnknownSchemaNodes(enumSubStmt.getUnknownSchemaNodes());
+                enumSubStmt.getDescription().ifPresent(pairBuilder::setDescription);
+                enumSubStmt.getReference().ifPresent(pairBuilder::setReference);
 
-                if (highestValue == null || highestValue < p.getValue()) {
-                    highestValue = p.getValue();
+                final EnumPair pair = pairBuilder.build();
+                if (highestValue == null || highestValue < pair.getValue()) {
+                    highestValue = pair.getValue();
                 }
 
-                builder.addEnum(p);
+                builder.addEnum(pair);
             }
             if (stmt instanceof UnknownEffectiveStatementImpl) {
                 builder.addUnknownSchemaNode((UnknownEffectiveStatementImpl) stmt);