how to generate non-string enum?

[EdwardJunyan]

JavaBean:

import lombok.Data;
import lombok.Getter;

@Data
public class Test {
    
    private NumberEnum number;
    
    private enum NumberEnum{
        A(1.0),
        B(2.0),
        C(3.0),
        D(4.0);

        @Getter
        private final Double value;

        NumberEnum(Double value) {
            this.value = value;
        }
    }
}

The Json Schema I expect for:

{
  "type" : "object",
  "properties" : {
    "number": {
      "type": "number",
      "default": 1.0,
      "enum": [
        1.0,
        2.0,
        3.0,
        4.0
      ]
    }
  }
}

[CarstenWickner]

Hi @EdwardJunyan,

If you want special handling for a given type, a CustomDefinitionProvider is the way to go.
The below assumes that the NumberEnum is not private but visible from where the schema generator configuration is defined:

configBuilder.forTypesInGeneral().withCustomDefinitionProvider((javaType, context) -> {
    if (!javaType.isInstanceOf(Test.NumberEnum.class)) {
        return null;
    }
    ObjectNode customNode = context.getGeneratorConfig().createObjectNode()
            .put(context.getKeyword(SchemaKeyword.TAG_TYPE), context.getKeyword(SchemaKeyword.TAG_TYPE_NUMBER));
    AttributeCollector collector = new AttributeCollector(context.getGeneratorConfig().getObjectMapper());
    List<Double> enumValues = Stream.of(Test.NumberEnum.values())
            .map(Test.NumberEnum::getValue)
            .collect(Collectors.toList());
    collector.setDefault(customNode, enumValues.get(0), context);
    collector.setEnum(customNode, enumValues, context);
    return new CustomDefinition(customNode);
});

[EdwardJunyan]

Thank you so much for the rapid reply. It works for me.
And I can use an generic interface which can be implemented by different kinds of "number enum class".