Include vendor extensions from security scheme (#4830)

OpenAPITools · Dec 18, 2019 · fb0479a · fb0479a
1 parent fee7924
commit fb0479a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java b/modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java
@@ -346,7 +346,7 @@ public Map<String, Object> updateAllModels(Map<String, Object> objs) {
     public void setCircularReferences(Map<String, CodegenModel> models) {
         final Map<String, List<CodegenProperty>> dependencyMap = models.entrySet().stream()
             .collect(Collectors.toMap(Entry::getKey, entry -> getModelDependencies(entry.getValue())));
-        
+
         models.keySet().forEach(name -> setCircularReferencesOnProperties(name, dependencyMap));
     }
 
@@ -3529,6 +3529,9 @@ public List<CodegenSecurity> fromSecurity(Map<String, SecurityScheme> securitySc
             cs.isCode = cs.isPassword = cs.isApplication = cs.isImplicit = false;
             cs.isBasicBasic = cs.isBasicBearer = false;
             cs.scheme = securityScheme.getScheme();
+            if (securityScheme.getExtensions() != null) {
+                cs.vendorExtensions.putAll(securityScheme.getExtensions());
+            }
 
             if (SecurityScheme.Type.APIKEY.equals(securityScheme.getType())) {
                 cs.isBasic = cs.isOAuth = false;