Add SchemaGeneratorFactory property to OpenApiDocumentGeneratorSettin…

…gs, #4524
i-moonlight · Dec 19, 2023 · b84c169 · b84c169
1 parent 53907f1
commit b84c169
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/src/NSwag.Generation/OpenApiDocumentGenerator.cs b/src/NSwag.Generation/OpenApiDocumentGenerator.cs
@@ -25,7 +25,8 @@ public class OpenApiDocumentGenerator
         /// <param name="schemaResolver">The schema resolver.</param>
         public OpenApiDocumentGenerator(OpenApiDocumentGeneratorSettings settings, JsonSchemaResolver schemaResolver)
         {
-            SchemaGenerator = new OpenApiSchemaGenerator(settings);
+            SchemaGenerator = settings.SchemaGeneratorFactory();
+
             _schemaResolver = schemaResolver;
             _settings = settings;
         }

diff --git a/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs b/src/NSwag.Generation/OpenApiDocumentGeneratorSettings.cs
@@ -23,9 +23,13 @@ public class OpenApiDocumentGeneratorSettings
         /// <summary>Initializes a new instance of the <see cref="OpenApiDocumentGeneratorSettings"/> class.</summary>
         public OpenApiDocumentGeneratorSettings()
         {
+            SchemaGeneratorFactory = () => new OpenApiSchemaGenerator(this);
             DefaultResponseReferenceTypeNullHandling = ReferenceTypeNullHandling.NotNull;
         }
 
+        /// <summary>Gets or sets the JSON Schema generator factory (default: new instance of <see cref="OpenApiSchemaGenerator"/>.</summary>
+        public Func<OpenApiSchemaGenerator> SchemaGeneratorFactory { get; set; }
+
         /// <summary></summary>
         public JsonSchemaGeneratorSettings SchemaSettings { get; set; } = new SystemTextJsonSchemaGeneratorSettings
         {

diff --git a/src/NSwag.Generation/OpenApiSchemaGenerator.cs b/src/NSwag.Generation/OpenApiSchemaGenerator.cs
@@ -21,12 +21,13 @@ public class OpenApiSchemaGenerator : JsonSchemaGenerator
 
         /// <summary>Initializes a new instance of the <see cref="OpenApiSchemaGenerator" /> class.</summary>
         /// <param name="settings">The settings.</param>
-        public OpenApiSchemaGenerator(OpenApiDocumentGeneratorSettings settings) : base(settings.SchemaSettings)
+        public OpenApiSchemaGenerator(OpenApiDocumentGeneratorSettings settings) 
+            : base(settings.SchemaSettings)
         {
         }
 
         /// <summary>Generates the properties for the given type and schema.</summary>
-        /// <param name="typeDescription">The type desription.</param>
+        /// <param name="typeDescription">The type description.</param>
         /// <param name="schema">The properties</param>
         /// <param name="schemaResolver">The schema resolver.</param>
         /// <returns></returns>
@@ -51,7 +52,7 @@ protected override void GenerateObject(JsonSchema schema, JsonTypeDescription ty
             }
         }
 
-        /// <summary>Generetes a schema directly or referenced for the requested schema type; also adds nullability if required.</summary>
+        /// <summary>Generates a schema directly or referenced for the requested schema type; also adds nullability if required.</summary>
         /// <typeparam name="TSchemaType">The resulted schema type which may reference the actual schema.</typeparam>
         /// <param name="contextualType">The type of the schema to generate.</param>
         /// <param name="isNullable">Specifies whether the property, parameter or requested schema type is nullable.</param>