net.morimekta.providence.jdbi.v3.EnumValueArgumentFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of providence-jdbi-v3 Show documentation

Utilities for handling providence messages using jdbi v3.

There is a newer version: 2.7.0

package net.morimekta.providence.jdbi.v3;

import net.morimekta.providence.PEnumValue;
import org.jdbi.v3.core.argument.Argument;
import org.jdbi.v3.core.argument.ArgumentFactory;
import org.jdbi.v3.core.argument.NullArgument;
import org.jdbi.v3.core.config.ConfigRegistry;

import java.lang.reflect.Type;
import java.sql.Types;
import java.util.Optional;

public class EnumValueArgumentFactory implements ArgumentFactory {
    private final Class enumClass;

    @SuppressWarnings("unused")
    public EnumValueArgumentFactory() {
        this(PEnumValue.class);
    }

    public EnumValueArgumentFactory(Class enumClass) {
        this.enumClass = enumClass;
    }

    @Override
    public Optional build(Type type, Object value, ConfigRegistry config) {
        if (type instanceof Class) {
            Class klass = (Class) type;
            if (enumClass.isAssignableFrom(klass)) {
                if (value == null) {
                    return Optional.of(new NullArgument(Types.INTEGER));
                }
                return Optional.of(new EnumValueArgument((PEnumValue) value));
            }
        }
        return Optional.empty();
    }

}