From 12a5a26fd2a65d799e76f2ec3393ffc1320ecaa5 Mon Sep 17 00:00:00 2001
From: lukaseder <lukas.eder@gmail.com>
Date: Wed, 28 Nov 2018 12:25:41 +0100
Subject: [PATCH] [#7680] Allow for overriding the order of generated import
 statements

---
 .../src/main/java/org/jooq/codegen/JavaWriter.java     | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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<JavaWriter> {
 
     private final Pattern             fullyQualifiedTypes;
     private final boolean             javadoc;
-    private final Set<String>         qualifiedTypes   = new TreeSet<String>();
+    private final Set<String>         qualifiedTypes   = new TreeSet<String>(qualifiedTypeComparator());
     private final Map<String, String> unqualifiedTypes = new TreeMap<String, String>();
     private final String              className;
     private final boolean             isJava;
@@ -161,6 +162,13 @@ public class JavaWriter extends GeneratorWriter<JavaWriter> {
         println(IMPORT_STATEMENT);
     }
 
+    /**
+     * Subclasses may override this to specify their own order of qualified types.
+     */
+    protected Comparator<String> qualifiedTypeComparator() {
+        return null;
+    }
+
     @Override
     protected String beforeClose(String string) {
         StringBuilder importString = new StringBuilder();