diff --git a/jOOQ-codegen/src/main/java/org/jooq/codegen/JavaWriter.java b/jOOQ-codegen/src/main/java/org/jooq/codegen/JavaWriter.java index b6b7c31391..eee405bd30 100644 --- a/jOOQ-codegen/src/main/java/org/jooq/codegen/JavaWriter.java +++ b/jOOQ-codegen/src/main/java/org/jooq/codegen/JavaWriter.java @@ -7,6 +7,7 @@ import java.io.File; import java.io.PrintWriter; import java.util.ArrayList; import java.util.Arrays; +import java.util.Comparator; import java.util.List; import java.util.Map; import java.util.Set; @@ -32,7 +33,7 @@ public class JavaWriter extends GeneratorWriter { private final Pattern fullyQualifiedTypes; private final boolean javadoc; - private final Set qualifiedTypes = new TreeSet(); + private final Set qualifiedTypes = new TreeSet(qualifiedTypeComparator()); private final Map unqualifiedTypes = new TreeMap(); private final String className; private final boolean isJava; @@ -161,6 +162,13 @@ public class JavaWriter extends GeneratorWriter { println(IMPORT_STATEMENT); } + /** + * Subclasses may override this to specify their own order of qualified types. + */ + protected Comparator qualifiedTypeComparator() { + return null; + } + @Override protected String beforeClose(String string) { StringBuilder importString = new StringBuilder();