code-generator/binding-java-api-generator/src/main/java/org/opendaylight/yangtools/sal/java/api/generator/BuilderGenerator.java

   1 package org.opendaylight.yangtools.sal.java.api.generator;
   2
   3 import java.io.IOException;
   4 import java.io.StringWriter;
   5 import java.io.Writer;
   6
   7 import org.opendaylight.yangtools.sal.java.api.generator.BuilderTemplate;
   8 import org.opendaylight.yangtools.sal.binding.model.api.CodeGenerator;
   9 import org.opendaylight.yangtools.sal.binding.model.api.GeneratedTransferObject;
  10 import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;
  11 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  12
  13 public final class BuilderGenerator implements CodeGenerator {
  14
  15     public static final String FILE_NAME_SUFFIX = "Builder";
  16
  17     @Override
  18     public Writer generate(Type type) throws IOException {
  19         final Writer writer = new StringWriter();
  20         if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
  21             final GeneratedType genType = (GeneratedType) type;
  22             final BuilderTemplate template = new BuilderTemplate(genType);
  23             writer.write(template.generate().toString());
  24         }
  25         return writer;
  26     }
  27
  28 }