Skip to content

Commit d97918d

Browse files
committed
[Java] Handle discriminator mapping non-ref name
The `mapping` property of the [Discriminator Object] is "An object to hold mappings between payload values and schema names or references." The subsequent examples in the spec have `mapping`s of both types. The `mapping` support introduced in OpenAPITools#536 only supports references. Update the code to support names (identified by lack of `/`) or references and change a test mapping to cover this case. [Discriminator Object]: https://github.com/OAI/OpenAPI-Specification/blob/master/versions/3.0.2.md#discriminatorObject Signed-off-by: Kevin Locke <[email protected]>
1 parent 731b162 commit d97918d

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

modules/openapi-generator/src/main/java/org/openapitools/codegen/DefaultCodegen.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1860,8 +1860,10 @@ private CodegenDiscriminator createDiscriminator(String schemaName, Schema schem
18601860
discriminator.setMapping(schema.getDiscriminator().getMapping());
18611861
if (schema.getDiscriminator().getMapping() != null && !schema.getDiscriminator().getMapping().isEmpty()) {
18621862
for (Entry<String, String> e : schema.getDiscriminator().getMapping().entrySet()) {
1863-
String name = toModelName(ModelUtils.getSimpleRef(e.getValue())); // e.g e.getValue => #/components/schemas/Dog
1864-
discriminator.getMappedModels().add(new MappedModel(e.getKey(), name));
1863+
String nameOrRef = e.getValue();
1864+
String name = nameOrRef.indexOf('/') >= 0 ? ModelUtils.getSimpleRef(nameOrRef) : nameOrRef;
1865+
String modelName = toModelName(name);
1866+
discriminator.getMappedModels().add(new MappedModel(e.getKey(), modelName));
18651867
}
18661868
} else {
18671869
Map<String, Schema> allDefinitions = ModelUtils.getSchemas(this.openAPI);

modules/openapi-generator/src/test/java/org/openapitools/codegen/DefaultCodegenTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -582,7 +582,7 @@ private void verifyPersonDiscriminator(CodegenDiscriminator discriminator) {
582582
test.setPropertyName("DollarUnderscoretype");
583583
test.setMapping(new HashMap<>());
584584
test.getMapping().put("a", "#/components/schemas/Adult");
585-
test.getMapping().put("c", "#/components/schemas/Child");
585+
test.getMapping().put("c", "Child");
586586
test.getMappedModels().add(new CodegenDiscriminator.MappedModel("a", "Adult"));
587587
test.getMappedModels().add(new CodegenDiscriminator.MappedModel("c", "Child"));
588588
Assert.assertEquals(discriminator, test);

modules/openapi-generator/src/test/resources/3_0/allOf.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ components:
3232
propertyName: $_type
3333
mapping:
3434
a: '#/components/schemas/Adult'
35-
c: '#/components/schemas/Child'
35+
c: Child
3636
properties:
3737
$_type:
3838
type: string

0 commit comments

Comments
 (0)