Add null check

microsoft · Oct 30, 2024 · 5eb0010 · 5eb0010
1 parent b806323
commit 5eb0010
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/Microsoft.OpenApi/Models/OpenApiSchema.cs b/src/Microsoft.OpenApi/Models/OpenApiSchema.cs
@@ -476,7 +476,10 @@ public void SerializeInternal(IOpenApiWriter writer, OpenApiSpecVersion version,
             writer.WriteOptionalCollection(OpenApiConstants.Enum, Enum, (nodeWriter, s) => nodeWriter.WriteAny(s));
 
             // type
-            SerializeTypeProperty(Type, writer, version);
+            if (Type is not null)
+            {
+                SerializeTypeProperty(Type, writer, version);
+            }
 
             // allOf
             writer.WriteOptionalCollection(OpenApiConstants.AllOf, AllOf, callback);
@@ -657,7 +660,10 @@ internal void SerializeAsV2(
             writer.WriteStartObject();
 
             // type
-            SerializeTypeProperty(Type, writer, OpenApiSpecVersion.OpenApi2_0);
+            if (Type is not null)
+            {
+                SerializeTypeProperty(Type, writer, OpenApiSpecVersion.OpenApi2_0);
+            }
 
             // description
             writer.WriteProperty(OpenApiConstants.Description, Description);
@@ -836,7 +842,7 @@ private static int CountEnumSetFlags(JsonSchemaType? schemaType)
                 // Check each flag in the enum
                 foreach (JsonSchemaType value in System.Enum.GetValues(typeof(JsonSchemaType)))
                 {
-                    // Ignore the None flag and check if the flag is set
+                    // Check if the flag is set
                     if (schemaType.Value.HasFlag(value))
                     {
                         count++;