add "x-enumNames" for schema properties no matter if the type is gene…

…ric or not
unchase · Feb 18, 2020 · 51ccdfa · 51ccdfa
1 parent 14d6fb7
commit 51ccdfa
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Unchase.Swashbuckle.AspNetCore.Extensions/Filters/XEnumNamesSchemaFilter.cs b/src/Unchase.Swashbuckle.AspNetCore.Extensions/Filters/XEnumNamesSchemaFilter.cs
@@ -59,9 +59,11 @@ public void Apply(OpenApiSchema schema, SchemaFilterContext context)
                         schema.Extensions.Add("x-enumDescriptions", enumsDescriptionsArray);
                     }
                 }
+                return;
             }
+
             // add "x-enumNames" for schema with generic types
-            else if (typeInfo.IsGenericType && !schema.Extensions.ContainsKey("x-enumNames"))
+            if (typeInfo.IsGenericType && !schema.Extensions.ContainsKey("x-enumNames"))
             {
                 foreach (var genericArgumentType in typeInfo.GetGenericArguments())
                 {
@@ -98,7 +100,8 @@ public void Apply(OpenApiSchema schema, SchemaFilterContext context)
                     }
                 }
             }
-            else if (schema.Properties?.Count > 0)
+
+            if (schema.Properties?.Count > 0)
             {
                 foreach (var schemaProperty in schema.Properties)
                 {