io.agrest.jaxrs3.openapi.modelconverter.AgValueModelConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of agrest-jaxrs3-openapi Show documentation

There is a newer version: 5.0.M19

Show newest version

package io.agrest.jaxrs3.openapi.modelconverter;

import io.agrest.jaxrs3.openapi.TypeWrapper;
import io.swagger.v3.core.converter.AnnotatedType;
import io.swagger.v3.core.converter.ModelConverter;
import io.swagger.v3.core.converter.ModelConverterContext;
import io.swagger.v3.oas.models.media.BinarySchema;
import io.swagger.v3.oas.models.media.Schema;
import io.swagger.v3.oas.models.media.StringSchema;

import java.time.LocalTime;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Supplier;

public class AgValueModelConverter extends AgModelConverter {

    private static final AgValueModelConverter instance = new AgValueModelConverter();

    public static AgValueModelConverter getInstance() {
        return instance;
    }

    private final Map> resolvers;

    protected AgValueModelConverter() {
        this.resolvers = new ConcurrentHashMap<>();
        this.resolvers.put(byte[].class, this::stringByteSchema);
        this.resolvers.put(LocalTime.class, this::stringNullSchema);
    }

    protected Schema stringByteSchema() {
        // Schemas are mutable, so creating a new one every time
        return new BinarySchema().format("byte");
    }

    protected Schema stringNullSchema() {
        // Schemas are mutable, so creating a new one every time
        return new StringSchema();
    }

    @Override
    protected boolean willResolve(AnnotatedType type, ModelConverterContext context, TypeWrapper wrapped) {
        return wrapped != null && resolvers.containsKey(wrapped.getRawClass());
    }

    @Override
    protected Schema doResolve(AnnotatedType type, ModelConverterContext context, Iterator chain, TypeWrapper wrapped) {
        Supplier resolver = resolvers.get(wrapped.getRawClass());
        Schema schema = resolver != null ? resolver.get() : null;
        return schema != null ? schema : delegateResolve(type, context, chain);
    }
}