From e09a74926d96174b8cbe6b60a5789f0c5d19b650 Mon Sep 17 00:00:00 2001
From: HugoMario <hugo.mario.mercado@gmail.com>
Date: Sat, 8 Feb 2020 23:51:50 -0500
Subject: [PATCH] add schema handle method on codegen config interface.

---
 .../src/main/java/io/swagger/codegen/v3/CodegenConfig.java   | 2 ++
 .../main/java/io/swagger/codegen/v3/DefaultGenerator.java    | 5 ++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/CodegenConfig.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/CodegenConfig.java
index a967a3b6999..7c92de81d13 100644
--- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/CodegenConfig.java
+++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/CodegenConfig.java
@@ -243,4 +243,6 @@ public interface CodegenConfig {
     void setIgnoreImportMapping(boolean ignoreImportMapping);
 
     boolean defaultIgnoreImportMappingOption();
+
+    ISchemaHandler getSchemaHandler();
 }
diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/DefaultGenerator.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/DefaultGenerator.java
index e5e305eadc8..da5e2c5c617 100644
--- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/DefaultGenerator.java
+++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/DefaultGenerator.java
@@ -386,13 +386,12 @@ public int compare(String o1, String o2) {
                 throw new RuntimeException("Could not process model '" + name + "'" + ".Please make sure that your schema is correct!", e);
             }
         }
-        final List<CodegenModel> composedModels = null;
-        /** todo: uncomment once generator repo is updated
+
         final ISchemaHandler schemaHandler = config.getSchemaHandler();
         schemaHandler.readProcessedModels(allProcessedModels);
 
         final List<CodegenModel> composedModels = schemaHandler.getModels();
-         */
+
          if (composedModels != null && !composedModels.isEmpty()) {
             for (CodegenModel composedModel : composedModels) {
                 final Map<String, Object> models = processModel(composedModel, config, schemas);