Add a method returns discriminator name

refs https://github.com/wing328/openapi-generator/pull/9#issuecomment-377776109
This commit is contained in:
akihito.nakano 2018-04-01 19:38:19 +09:00
parent 7daa2ec5da
commit 28fcf48f42
2 changed files with 7 additions and 3 deletions

View File

@ -331,7 +331,11 @@ public class CodegenModel {
this.unescapedDescription = unescapedDescription;
}
public String getDiscriminator() {
public Discriminator getDiscriminator() {
return discriminator;
}
public String getDiscriminatorName() {
return discriminator == null ? null : discriminator.getPropertyName();
}

View File

@ -1,6 +1,6 @@
class {{classname}} {{#parentSchema}}extends {{{parent}}} {{/parentSchema}}{{^parentSchema}}implements ModelInterface, ArrayAccess{{/parentSchema}}
{
const DISCRIMINATOR = {{#discriminator}}'{{discriminator}}'{{/discriminator}}{{^discriminator}}null{{/discriminator}};
const DISCRIMINATOR = {{#discriminator}}'{{discriminatorName}}'{{/discriminator}}{{^discriminator}}null{{/discriminator}};
/**
* The original name of the model.
@ -166,7 +166,7 @@ class {{classname}} {{#parentSchema}}extends {{{parent}}} {{/parentSchema}}{{^pa
{{#discriminator}}
// Initialize discriminator property with the model name.
$discriminator = array_search('{{discriminator}}', self::$attributeMap);
$discriminator = array_search('{{discriminatorName}}', self::$attributeMap);
$this->container[$discriminator] = static::$swaggerModelName;
{{/discriminator}}
}