From ca45aa6befeccab7b7cd2675bc740a4e7ff1fd20 Mon Sep 17 00:00:00 2001
From: Jan-Willem Gmelig Meyling <jan-willem@youngmediaexperts.nl>
Date: Sat, 4 Jan 2020 02:37:51 +0100
Subject: [PATCH] Register APT extensions by service loader

---
 .../querydsl/apt/DefaultConfiguration.java    |  6 ++++
 .../main/java/com/querydsl/apt/Extension.java | 33 +++++++++++++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 querydsl-apt/src/main/java/com/querydsl/apt/Extension.java

diff --git a/querydsl-apt/src/main/java/com/querydsl/apt/DefaultConfiguration.java b/querydsl-apt/src/main/java/com/querydsl/apt/DefaultConfiguration.java
index d49128003..17d391457 100644
--- a/querydsl-apt/src/main/java/com/querydsl/apt/DefaultConfiguration.java
+++ b/querydsl-apt/src/main/java/com/querydsl/apt/DefaultConfiguration.java
@@ -213,6 +213,12 @@ public class DefaultConfiguration implements Configuration {
             // do nothing
         }
 
+        ServiceLoader<Extension> loader = ServiceLoader.load(Extension.class, Extension.class.getClassLoader());
+
+        for (Extension extension : loader) {
+            extension.addSupport(module);
+        }
+
         defaultSerializerConfig = new SimpleSerializerConfig(entityAccessors, listAccessors,
                 mapAccessors, createDefaultVariable, "");
 
diff --git a/querydsl-apt/src/main/java/com/querydsl/apt/Extension.java b/querydsl-apt/src/main/java/com/querydsl/apt/Extension.java
new file mode 100644
index 000000000..00b39dfe4
--- /dev/null
+++ b/querydsl-apt/src/main/java/com/querydsl/apt/Extension.java
@@ -0,0 +1,33 @@
+/*
+ * Copyright 2016, The Querydsl Team (http://www.querydsl.com/team)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.querydsl.apt;
+
+import com.querydsl.codegen.AbstractModule;
+
+/**
+ * {@code Extension} allows for custom code generation extensions to be registered as service provider
+ *
+ * @author Jan-Willem Gmelig Meyling
+ *
+ */
+public interface Extension {
+
+    /**
+     * Register custom types to the given codegen module
+     *
+     * @param module module to be customized
+     */
+    void addSupport(AbstractModule module);
+
+}